백준 10773: 제로(C++)

문제 링크 : https://www.acmicpc.net/problem/10773

10773번: 제로

 

 

문제

 

나코더 기장 재민이는 동아리 회식을 준비하기 위해서 장부를 관리하는 중이다.

재현이는 재민이를 도와서 돈을 관리하는 중인데, 애석하게도 항상 정신없는 재현이는 돈을 실수로 잘못 부르는 사고를 치기 일쑤였다.

재현이는 잘못된 수를 부를 때마다 0을 외쳐서, 가장 최근에 재민이가 쓴 수를 지우게 시킨다.

재민이는 이렇게 모든 수를 받아 적은 후 그 수의 합을 알고 싶어 한다. 재민이를 도와주자!

입력

첫 번째 줄에 정수 K가 주어진다. (1 ≤ K ≤ 100,000)

이후 K개의 줄에 정수가 1개씩 주어진다. 정수는 0에서 1,000,000 사이의 값을 가지며, 정수가 "0" 일 경우에는 가장 최근에 쓴 수를 지우고, 아닐 경우 해당 수를 쓴다.

정수가 "0"일 경우에 지울 수 있는 수가 있음을 보장할 수 있다.

출력

재민이가 최종적으로 적어 낸 수의 합을 출력한다. 최종적으로 적어낸 수의 합은 231-1보다 작거나 같은 정수이다.

 

 

해설

"가장 최근에 재민이가 쓴 수를 지우게 시킨다."
이 부분을 보면 스택을 사용해야 한다는 것을 알 수 있다.
스택은 LIFO(Last In First Out)특성을 가지고 있다. 즉 마지막(최근)이 먼저 나오므로
문제풀기에 부합하다.

구현해보자면 0이 아닌 숫자가 나올 시에는 스택에 숫자를 넣어주면 되고
0이나오면 스택의 가장 윗 부분(top)을 지워주면 된다.
그렇게하면 최근에 쓴 수가 지워진다.

 

참고로 스택은 STL을 사용하였다.

STL을 사용하면 스택을 구현없이 바로 가져와서 쓸 수 있기 때문에 유용하다.

사용법을 모르겠다면 한번 찾아보자.

 

★추후에 STL관련 글을 쓸 예정이다.

 

추가로 스택 없이도 문제를 풀 수 있다. 이는 아래에서 보여주겠다.

코드(스택 사용)

#include<iostream>
#include<stack>
using namespace std;
int main()
{
    int n,ans=0;
    cin>>n;
    stack<int> v;
    for(int i=0; i<n; i++)
    {
        int c;
        cin>>c;
        if(c==0)
            v.pop();
        else
            v.push(c);
    }
    while(!v.empty())
    {
        ans+=v.top();
        v.pop();
    }
    cout<<ans;
}
//스택을 사용하여 풀었다.

 

 

코드(배열 사용)

#include<iostream>
using namespace std;
int main()
{
    int n,ans=0;
    cin>>n;
    int a[n],aa=0;
    for(int i=0; i<n; i++)
    {
        int c;
        cin>>c;
        if(c==0)
            aa--;
        else
            a[aa++]=c;
    }
    for(int i=0; i<aa; i++)
        ans+=a[i];
    cout<<ans;
}
//스택 없이 배열로 풀었다.

스택으로 푼 코드와 구조는 거의 같다.

 

aa는 사용하는 배열의 크기를 의미한다.

입력받은수가 0이 아니라면 배열a에 숫자를 넣어주고(aa인덱스에) aa값을 1 늘려준다.

0이라면 aa값을 1 줄여준다. 이러는 이유는 최근에 쓴 수(마지막에 쓴 수)는 배열의 가장 뒤에 있다.(a[aa])

이 때 aa값을 1 줄여주면 가장 뒤에 있던 수는 사라지는 것이나 마찬가지이다.

나중에 모든 수를 다 합하는데 0~aa-1까지의 숫자를 합한다. 이 때 만약 aa인덱스에 숫자가 있었다면

이 숫자는 더해지지 않는다. 범위안에 없기 때문이다. 지워진 취급을 받기 때문이다.