백준 12789번 도키도키 간식드리미 C++

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

12789번: 도키도키 간식드리미

 

처음 줄을 큐로, 다른 대기하는 공간을 스택으로 해서 풀었다.

 

반복문 안의 if문들

1. 스택, 큐 둘다 비어있으면 다 간식을 받았기 때문에 반복 종료

2. 큐의 맨 앞이 현재 번호와 같다면 간식을 받도록 하고 번호 1 증가

3. 스택의 맨 위가 현재 번호와 같다면 간식을 받도록 하고 번호 1 증가

4. 큐의 맨 앞이 현재 번호와 같지 않다면 맨 앞 번호를 스택으로 옮김

5. 스택의 맨 앞이 현재 번호와 같지 않다면 (여기까지 왔으면 큐가 비어있는 상태) 순서대로 받을 수 있는 방법이 없기 때문에 답을 false로 하고 반복 종료

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    int n;
    cin >> n;
    queue<int> q;
    stack<int> st;
    for (int i = 0; i < n; i++) {
        int num;
        cin >> num;
        q.push(num);
    }
    int target = 1;
    bool ans = true;
    while (true) {
        if (q.empty() && st.empty()) {
            break;
        }
        if (q.empty() == false && q.front() == target) {
            q.pop();
            target++;
            continue;
        }
        if (st.empty() == false && st.top() == target) {
            st.pop();
            target++;
            continue;
        }
        if (q.empty() == false && q.front() != target) {
            int num = q.front();
            q.pop();
            st.push(num);
            continue;
        }
        if (st.empty() == false && st.top() != target) {
            ans = false;
            break;
        }
    }
    if (ans) cout << "Nice";
    else cout << "Sad";
}