본문 바로가기

ps(알고리즘)

문제집

먼저 푸는 것이 없는 문제를 풀 수 있고, 쉬운 문제부터 풀어야 하기 때문에 먼저 푸는 것이 없는 문제들을 

우선 순위 큐에 넣고 최소값을 top으로 구해서 출력하면 된다.

그리고 문제를 썼으면 pop을 하면서 해당 문제로 인해 먼저 푸는 것이 없어지는 문제들을 큐에 넣어주면 되는 간단한 문제이다.

#include<stdio.h>
#include<vector>
#include<algorithm>
#include<queue>
using namespace std;
vector<int> prob[40004];
vector<int> rev[40004];
priority_queue<int>last;
int N,M;
int main()
{
    scanf("%d %d",&N,&M);
    for(int i=1;i<=M;i++)
    {
        int a,b;
        scanf("%d %d",&a,&b);
        prob[b].push_back(a);
        rev[a].push_back(b);
    }
    for(int i=1;i<=N;i++)
    {
        if(prob[i].empty()){
            last.push(-i);
        }
    }
    while(!last.empty()){
        int x = last.top();
        x *= -1;
        printf("%d ",x);
        last.pop();
        for(int i:rev[x]){
            prob[i].erase(remove(prob[i].begin(),prob[i].end(),x),prob[i].end());
            if(prob[i].empty()){
                last.push(-i);
            }
        }
    }
}

해결 코드이다.

'ps(알고리즘)' 카테고리의 다른 글

친구 네트워크  (0) 2024.09.02
피보나치 수 6  (0) 2024.07.17
보석 도둑  (0) 2024.07.13
후위 표기식  (0) 2024.07.13
가운데를 말해요  (0) 2024.06.26