洛谷 P6145 [USACO20FEB] Timeline G

#include<bits/stdc++.h>
#define int long long
#define PII pair<int,int>
#define se second
#define fi first
#define pb push_back
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
int dp[200000];
int va[200000];
int in[200000];
vector <PII> v[200000];
int n,m,k;
void topsort()
{
	for(int i=1;i<=n;i++)
	{
		dp[i]=va[i];//第i次挤奶时间最小的值 
	}
	queue<int>q;
	for(int i=1;i<=n;i++)
	{
		if(in[i]==0)
		q.push(i);
	}
	while(q.size())
	{
		auto now=q.front();
		q.pop();
		for(auto t:v[now])
		{
			int spot=t.se;
			int w=t.fi;
			dp[spot]=max(dp[spot],dp[now]+w);//dp[i]为第i次挤奶的时间最早为多少天
			if(--in[spot]==0)q.push(spot); 
		}
	}
}
signed main()
{
	IOS;
	cin>>n>>m>>k;
	for(int i=1;i<=n;i++)cin>>va[i];//第i次挤奶的时间不早于va[i] 
	for(int i=1;i<=k;i++)
	{
		int a,b,c;
		cin>>a>>b>>c;
		v[a].pb({c,b});//第a次挤奶一定早于第b次至少c天 
		in[b]++;
	}
	topsort();
	for(int i=1;i<=n;i++)
	{
		cout<<dp[i]<<"\n";
	}
	return 0;
}