CCF认证 2017-12 商路(自底向上的拓扑序)

因为要计算一个城市接下来通向哪里可以取得最大值，必须知道所有下级城市的商路能取得的最大值（我是这么认为的）

最下级城市的价值一定为0（已知），已知最大价值后就可以计算所有上级直接通向这里可以得到的价值，取最大的那个（保存在一个数组ans里）

如果一个城市的所有直接下级都处理过了，那么它的最大价值就已知了，就可以计算它的上级直接通向它能取得的价值

所以要从下级向上级计算，再用类似拓扑排序的方法，查看是否计算了所有直接下级
 

#include<iostream>
#include<algorithm>
#include<bits/stdc++.h>
#include<queue>
using namespace std;
typedef long long ll;
const ll MOD=1e18;
const int N=1e5+10;
ll v[N],f[N],u[N],s[N];
ll ans[N];
int vis[N];
void init()
{
	memset(ans,0,sizeof(ans));
	memset(vis,0,sizeof(vis));
}
queue<int>q;
int main()
{
	int T,n;
	scanf("%d",&T);
	while(T--)
	{
		init(); //初始化 
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
		{
			scanf("%d%d%d%d",&u[i],&s[i],&v[i],&f[i]);
			vis[u[i]]++; //vis数组保存这个点的直接下级数量 
		}
		
		for(int i=1;i<=n;i++) //这里类似拓扑排序，直接下级数量为零入队 
		if(!vis[i]) q.push(i);
		while(!q.empty())
		{
			int k=q.front();q.pop();
			int t=u[k],d=s[k];
			vis[t]--;
			if(!vis[t]) q.push(t);//直接下级全都处理过了，入队 
			while(t) //寻找所有上级，计算上级城市到这里的获利，取最大值 
			{
				ans[t]=max(ans[t],(ans[k]+v[t]-(f[t]-d)*(f[t]-d))); //ans里保存了以这个城市为开头的商路最大价值 
				d+=s[t];
				t=u[t];
			}
		}
		ll sum=0;
		for(int i=1;i<=n;i++) sum=(sum+ans[i])%MOD; //计算总价值 
		printf("%lld\n",sum);
	}
	return 0;
}