스택(1)

https://www.youtube.com/watch?v=0DsyCXIN7Wg

스택 강의

 

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

10828번: 스택

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

int main(){
  ios::sync_with_stdio(0);
  cin.tie(0);
  int n;
  stack<int> st;
  cin>>n;
  while(n--){
  	string str;
  	cin>>str;
  	if(str=="push"){
  		int a;
		cin>>a;
		st.push(a);
	}
  	if(str=="pop"){
  		if(st.empty()) cout<<-1<<"\n";
  		else{
		  cout<<st.top()<<"\n";
  		  st.pop();
  		}
	}
  	if(str=="size") cout<<st.size()<<"\n";
  	if(str=="empty") cout<<st.empty()<<"\n"; 
  	if(str=="top"){
  		if(st.empty()) cout<<-1<<"\n";
  		else cout<<st.top()<<"\n";
	  }
  }
  return 0;
}

---

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

10773번: 제로

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

int main(){
  ios::sync_with_stdio(0);
  cin.tie(0);
  int n,input,sum=0;
  stack<int> ST;
  cin>>n;
  for(int i=0;i<n;i++){
  	cin>>input;
  	if(input==0) ST.pop();
  	else ST.push(input);
  }
  while(!ST.empty()){
  	sum+=ST.top();
  	ST.pop();
  }
  cout<<sum;
  return 0;
}

 

처음에 문제를 보고 굳이 스택이 필요한가? 란 생각이 들어서 입력과 동시에 sum을 계산했는데 0이 연속으로 나올 경우 문제가 생겨서 스택을 써야 함을 알 수 있었다.

---