[BOJ] 10828번 - 스택 (Swift, C++)

문제

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

명령은 총 다섯 가지이다.

• push X: 정수 X를 스택에 넣는 연산이다.
• pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 스택에 들어있는 정수의 개수를 출력한다.
• empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
• top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

<문제 링크>

10828번: 스택

---

입력

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

---

출력

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

---

문제 풀이

단순히 스택을 구현해보는 문제입니다.스택은 FIFO 방식을 따르는 자료구조로 프링글스를 생각하시면 됩니다!(헷갈리신다면 제가 설명한 글 참고해주세요~)

 

Swift에서는 자료구조를 라이브러리로 지원해주지 않지만 C++에서는 지원을 해주기 때문에Swift와 C++, 두 개의 언어로 구현을 해보았습니다.

 

Stack은 배열의 마지막 데이터를 주로 다루는 구조이기 때문에 각 메서드의 시간복잡도는 O(1) 상수시간을 갖습니다.

풀이 순서

1. Swift의 경우 Stack을 구조체로 만들어주고, C++의 경우 라이브러리에서 지원해주는 Stack을 정의해줍니다.

2. 문제에서 요구하는 값을 조건에 맞게 콘솔에 출력해줍니다.

---

풀이 코드

< Swift>

// Stack 구현
struct Stack {
    var stack =  [Int]()
    
    mutating func push(n: Int) {
        stack.append(n)
    }
    
    mutating func pop() -> Int {
        return stack.popLast() ?? -1
    }
    
    func getSize() -> Int {
        stack.count
    }
    
    func isEmpty() -> Int {
        stack.isEmpty ? 1 : 0
    }
    
    func top() -> Int {
        stack.last ?? -1
    }
}

// 문제 구현부
var list = Stack()

let n = Int(readLine()!)!

for _ in 1...n {
    let cmd = readLine()!
    
    switch cmd {
    case "pop":
        print(list.pop())
    case "size":
        print(list.getSize())
    case "empty":
        print(list.isEmpty())
    case "top":
        print(list.top())
    default:
        let arr = cmd.split(separator: " ")
        let num = Int(String(arr.last!))!
        list.push(n: num)
    }
}

---

< C++>

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

int main(void) {
    ios::sync_with_stdio(0);
    cin.tie(0);
    
    stack<int> S;  // stack 생성
    
    int n;
    cin >> n;
    
    while(n--){  // n번 반복
        string c;
        cin >> c;
        
        if(c=="push"){
            int n;
            cin >> n;
            S.push(n);
        }
        else if(c=="pop"){
            if(S.empty()) cout << -1 << '\n';
            else{
                cout << S.top() << '\n';
                S.pop();
            }
        }
        else if(c=="size") {
              cout << S.size() << '\n';
        }
        else if(c=="empty"){
            cout << int(S.empty()) << '\n';
        }
        else{
            if(S.empty()) cout << -1 << '\n';
            else{
                cout << S.top() << '\n';
            }
        }
    }
    
}

---

마무리

stack을 자료구조 측면으로만 공부해보았고, 이번에 처음으로 문제를 풀어보았습니다. 

처음이라 그런지 stack을 완전히 구조체로 작성하고 풀이에 이용했는데, 앞으로는 이렇게 할 필요가 없을 것 같습니다. 

 

코드 작성에 시간을 단축해야 하는 테스트의 경우, 완전한 구조를 만들지 않고 개념만을 고려하여 코드를 간단하게 작성할 수 있다는 것을 깨달았습니다. 다른 사람의 풀이를 보니 대부분이 스택을 완전히 구현하지 않고 개념만을 이용하여 문제를 해결했다는 것을 알게 되었습니다.

 

또한, 앞으로 C++ 언어도 함께 공부하고자 합니다.

자료구조를 라이브러리로 지원해주는 것뿐만 아니라 언어적 측면에서도 공부를 해보면 도움을 받을 수 있을 것 같습니다.