CCF认证 2017-12 商路

首先说明，这道题我只拿了60分，超时，还没有想到怎么优化或者更好的算法

如果有优化方法，其他算法或者建议，欢迎在评论区留言

接下来说我的方法：

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

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

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

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

最后把所有的加起来，具体看代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
const ll MOD=1e18;
const int N=1e5+10;
ll v
,f
,u
,s
;
ll ans
;
int vis
;
void init()
{
memset(ans,0,sizeof(ans));
memset(vis,0,sizeof(vis));
}
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数组保存这个点的直接下级数量
}
int q
,q0=0,q1=0;
for(int i=1;i<=n;i++) //这里类似拓扑排序，直接下级数量为零入队
if(!vis[i]) q[q1++]=i;
while(q0<q1)
{
int k=q[q0++];
int t=u[k],d=s[k];
vis[t]--;
if(!vis[t]) q[q1++]=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;
}