[BOJ] 줄서기 17178번 C++

문제

아이즈원의 팬인 시온이는 드디어 티켓팅에 성공하여 콘서트를 갔다. 콘서트장에 일찍 도착한 시온이는 기대하며 입장을 위해 줄을 섰다. 하지만 아이즈원의 인기대로 시온이를 포함한 많은 팬이 줄을 서고 있다. 콘서트의 입장이 시작되었고 입장은 티켓 번호 순서대로 이루어졌다. 하지만 입구에 너무 많은 팬이 몰려 아무도 이동할 수 없는 상황이 되었고, 결국 주최 측에서 인원을 정렬시켜 다음과 같이 간신히 사람 한 줄이 설 수 있는 대기 공간을 만들었다.

 

주최 측은 번호표 순서대로만 통과할 수 있는 입구를 만들어 두었지만, 줄에서는 마구잡이로 사람들이 기다리고 있다. 대기 공간을 이용하여 입장이 원활히 이루어지도록 하려고 한다. 콘서트장에 사람들이 제대로 들어갈 수 있는지 확인해보자.

사람들은 현재 5명씩 N 줄을 서 있고, 첫 번째 줄 맨 앞사람만 이동이 가능하다. 이 사람은 콘서트장으로 입장할 수도 있고 대기 공간에서 다시 기다릴 수도 있다. 한 줄의 사람이 다 이동했다면 그다음 줄의 사람들이 이동한다. 대기 공간에는 한 줄로만 설 수 있는 공간이 있으며, 마지막에 들어온 사람부터 나갈 수 있다. 나갈 경우 바로 입장해야 하고, 다시 줄로 돌아갈 수는 없다. 티켓은 A-123과 같이 한 개의 대문자 알파벳과 '-', 1000 미만의 자연수의 조합으로 이뤄어져 있다. 만약 수가 7이라면 A-7과 같이 주어진다. 티켓의 순서는 알파벳이 빠른 티켓이 빠르며, 동일하다면 더 적은 수가 더 빠르다. 티켓 번호는 중복되게 주어지지 않는다.

위의 예시를 예로 들면 다음과 같이 모든 사람들이 입장할 수 있다. 그림과는 달리 대기 공간에는 무한히 많은 사람들이 들어올 수 있다는 것에 주의하여야 한다.

 

입력

첫째 줄에는 줄에서 기다리고 있는 사람들의 줄 수 N이 주어진다. (1 ≤ N ≤ 100)

둘째 줄부터 N 개의 줄에는 한 줄에 서 있는 5명의 티켓 번호가 주어진다.

사람들이 서 있는 순서대로 입력이 주어진다.

출력

모든 사람이 무사히 콘서트장에 입장할 수 있다면 “GOOD”을 출력하고 그렇지 않다면 “BAD”를 출력한다.

 

접근 방식

 

풀이는 생각보다 간단했지만 어려웠다..!

 

먼저, 문자열을 비교해서 정렬된 입장 순서를 만들어준다

이때, 문자열 비교 함수를 따로 만들어줘야하는데 그 이유가 단순히 sort 함수를 사용하면

A_10이 A_2보다 순서가 앞서기 때문이다

 

이후, 실제 줄을 순회하면서 입장 순서와 같은지 확인한다

만약 입장 순서와 현재 줄이 같다면 바로 입장하고, 다르면 스택에 넣는다.

 

줄을 모두 순회한 다음은 스택을 확인하면서

만약 스택의 가장 top이 현재 입장 순서와 같으면 pop해주면 된다

 

만약 idx 가 total 값과 같지 않다면 순서대로 입장할 수 없다는 의미이므로 BAD를 출력하고, 같다면 GOOD을 출력하면 된다

 

코드

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

bool cmp(const string &a, const string &b) {
    char ca = a[0], cb = b[0];
    int na = stoi(a.substr(2));
    int nb = stoi(b.substr(2));

    if(ca != cb) return ca < cb;
    return na < nb;
}

int main() {
    int N; cin>>N;
    int total = N*5;

    vector<string> init(total);
    for(int i=0; i<total; i++) cin>>init[i];

    vector<string> order(init);
    sort(order.begin(), order.end(), cmp);

    stack<string> st;
    int idx = 0;

    for(int i=0; i<total; i++) {
        while(!st.empty() && st.top() == order[idx]) {
            st.pop();
            idx++;
        }
        if(init[i] == order[idx]) idx++;
        else st.push(init[i]);
    }

    while(!st.empty() && st.top() == order[idx]) {
        st.pop();
        idx++;
    }

    if(idx == total) cout<<"GOOD";
    else cout<<"BAD";
}