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

 

- 문제 사이트: https://www.acmicpc.net/problem/12789

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

 

줄 서 있는 학생들은 queue 형태로 이동하고, 임시 공간에 서는 학생은 stack 형태로 이동한다.

이에 맞게 아래의 로직을 구현해주면 된다.

 

1) 비교할 수 있도록 현재 순서를 저장 (order)

2) 임시 공간 (stack) 가장 위의 학생이 현재 순서와 일치하는지 확인

3-1) 줄 서 있는 (queue) 가장 앞의 학생이 현재 순서와 일치한다면, 다음 학생 확인: order++

3-2) 줄 서 있는 (queue) 가장 앞의 학생이 현재 순서와 일치하지 않는다면, 임시 공간에 학생 이동

4) 비교 가능한 학생이 없는 경우, 종료

 

#include <iostream>
#include <queue>
#include <stack>
using namespace std;

int main()
{
    int N, n;
    queue<int> student;
    stack<int> waiting;
    
    cin >> N;
    for (int i = 0; i < N; i++)
    {
        cin >> n;
        student.push(n);
    }
    
    int order = 1;
    while (true)
    {
        if (!waiting.empty() && waiting.top() == order)
        {
            waiting.pop();
            order++;
        }
        else if (!student.empty())
        {
            if (student.front() == order)
            {
                student.pop();
                order++;
            }
            else
            {
                n = student.front();
                student.pop();
                waiting.push(n);
            }
        }
        else
        {
            break;
        }
    }
    
    if (student.empty() && waiting.empty())
        cout << "Nice";
    else
        cout << "Sad";
    return 0;
}

- 메모리: 2024 KB

- 시간: 0 ms

- 코드 길이: 933 B