10773번: 제로

mint* 2023. 3. 1. 19:44

728x90

https://www.acmicpc.net/problem/10773

10773번: 제로

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

www.acmicpc.net

벡터로 구현

#include <bits/stdc++.h>
using namespace std;

int k, a,sum;

int main(){
    ios_base::sync_with_stdio(0); cin.tie(NULL); cout.tie(NULL);
    //1. 입력받기
    cin >> k;
    vector<int> v; //저장할 자료구조(stack 사용해도 됨)
    //2. 0이 아니면 수 저장, 0이면 맨 뒤 원소 삭제(pop_back)
    for(int i=0;i<k;i++){
        cin >> a;
        if(a==0) v.pop_back();
        else v.push_back(a);
    }
    //3. 합계 출력
    for(int e:v) sum+=e;
    cout << sum;
}

스택으로 구현

#include <bits/stdc++.h>
using namespace std;

int k, a,sum;

int main(){
    ios_base::sync_with_stdio(0); cin.tie(NULL); cout.tie(NULL);
    //1. 입력받기
    cin >> k;
    stack<int> s; //저장할 자료구조
    //2. 0이 아니면 수 저장, 0이면 맨 뒤 원소 삭제(pop)
    for(int i=0;i<k;i++){
        cin >> a;
        if(a==0) s.pop();
        else s.push(a);
    }
    //3. 합계 구하여 출력하기
    while(!s.empty()){
        sum += s.top();
        s.pop();
    }
    cout << sum;
}

728x90