본문 바로가기
Computer Science/Problem Solving(Algorithm)

< BOJ > 10828 - 스택

by 진뚱 2019. 6. 2.
728x90

문제

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

 

10828번: 스택

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

www.acmicpc.net

10828번: 스택

10828번 제출 맞은 사람 숏코딩 풀이 풀이 작성 풀이 요청 재채점/수정 채점 현황 강의 스택 풀이 시간 제한 메모리 제한 제출 정답 맞은 사람 정답 비율 1 초 256 MB 19648 7604 5933 40.235% 문제 정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오. 명령은 총 다섯 가지이다. push X: 정수 X를 스택에 넣는 연산이다. pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한

www.acmicpc.net

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 다섯 가지이다.

push X: 정수 X를 스택에 넣는 연산이다.

pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

size: 스택에 들어있는 정수의 개수를 출력한다.

empty: 스택이 비어있으면 1, 아니면 0을 출력한다.

top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

 

입력

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

 

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

 

해결방법

스택을 문자열 처리를 통해 사용할 수 있도록 바꿔 주었다.

문자열을 비교하여 해당하는 구문에 접근 할 수 있도록 하면 되는 간단한 문제이다.

 

코드

#include <stdio.h>
#include <stack>
#include<string.h>
#include <iostream>


using namespace std;

int main(){
    
    int TestCase;
    cin >> TestCase;
    
    stack<int> st;
    string s;
    
    for(int i=0;i<TestCase;i++){
        
        cin>> s;
        if(s=="push"){
            int val;
            cin >> val;
            st.push(val);
        }
        else if (s=="pop"){
            if(st.empty()) {cout<<-1<<endl;}
            else{
                cout << st.top() << endl;
                st.pop();
            }
        }
        else if (s=="size"){
            cout << st.size() << endl;
        }
        else if (s=="empty"){
            if(st.empty()) {cout << 1 << endl;}
            else { cout << 0 << endl;}
            
        }
        else if (s=="top"){
            if(st.empty()) {cout << -1 << endl;}
            else {cout << st.top() << endl;}
        }
    }
    return 0;
}

Stack STL 을 사용하지 않고 Vector만을 이용해 구현

#include<stdio.h>
#include<string.h>
#include<vector>

using namespace std;


int main(){
    int n;
    scanf("%d",&n);
    char str[6];
    vector<int> stack;
    for(int i=0;i<n;i++){
        scanf("%s",str);
        int temp;
        if(!strcmp(str,"push")){
            scanf("%d",&temp);
            stack.push_back(temp);
        }
        else if(!strcmp(str,"pop")){
            if(stack.empty()) printf("-1\n");
            else{temp = stack.back();
            printf("%d\n",temp);
            stack.pop_back();
            }
        }
        else if(!strcmp(str,"size")){
            printf("%d\n",stack.size());
        }
        else if(!strcmp(str,"empty")){
            if(stack.empty())   printf("1\n");
            else printf("0\n");
        }
        else if(!strcmp(str,"top")){
            if(stack.empty()) printf("-1\n");
            else printf("%d\n",stack.back());
                }
        else{
            printf("unknown command!\n");
        }
    }
    return 0;
}
728x90

'Computer Science > Problem Solving(Algorithm)' 카테고리의 다른 글

< BOJ > 2571 - 색종이-3  (0) 2019.06.04
< BOJ > 1912 - 연속합  (0) 2019.06.02
< BOJ > 1406 - 에디터  (0) 2019.06.02
< BOJ > 11057 - 오르막 수  (0) 2019.06.02
< BOJ > 9012 - 괄호  (0) 2019.06.02

댓글