백준 17952번 과제는 끝나지 않아! C++

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

17952번: 과제는 끝나지 않아!

 

최근에 했던 과제를 꺼내서 하기에는 스택이 적합한 것 같아서 스택을 사용해서 풀었다.

 

#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; 

struct Task {
    int score;
    int time;
};

int n;

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

    cin >> n;
    int ans = 0;
    Task* currTask = nullptr;
    stack<Task*> jobStack;
    for (int i = 0; i < n; i++) {
        int val;
        cin >> val;
        if (val == 1) {
            int a, t;
            cin >> a >> t;
            if (currTask == nullptr) {
                currTask = new Task({ a,t });
                if (--currTask->time == 0) {
                    ans += currTask->score;
                    currTask = nullptr;
                }
            }
            else {
                jobStack.push(currTask);
                currTask = new Task({ a,t });
                if (--currTask->time == 0) {
                    ans += currTask->score;
                    currTask = nullptr;
                }
            }
        }
        else {
            if (currTask == nullptr) {
                if (jobStack.empty()) continue;
                currTask = jobStack.top();
                jobStack.pop();
            }
            if (--currTask->time == 0) {
                ans += currTask->score;
                currTask = nullptr;
            }
        }
    }
    cout << ans;
}