Baekjoon #1005

Baekjoon #1005

1개의 풀이 · C++

문제 원문 보기 ↗

SOLUTION INFO

C++ · main.cpp

main.cpp
#include<bits/stdc++.h>

using namespace std;

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);

    int T; cin >> T;
    while(T--) {
        int N, M, W; cin >> N >> M;
        vector<vector<int>> graph(N);
        vector<int> V(N), DP(N), indegree(N);
        for(int i=0;i<N;i++) cin >> V[i];
        for(int i=0;i<M;i++) {
            int a, b; cin >> a >> b;
            graph[--a].emplace_back(--b);
            indegree[b]++;
        }
        cin >> W;

        queue<int> Q;
        for(int i=0;i<N;i++){
            if(!indegree[i]) {
                Q.push(i);
                DP[i] = V[i];
            }
        } 

        while(!Q.empty()) {
            int cur = Q.front(); Q.pop();
            for(auto &nxt: graph[cur]) {
                if(--indegree[nxt] == 0) Q.push(nxt);
                DP[nxt] = max(DP[nxt], DP[cur] + V[nxt]);
            }
        }

        cout << DP[--W] << '\n';
    }

    return 0;
}

SOLUTION DESCRIPTION

풀이 설명

등록된 풀이 설명이 없습니다.