联合权值
枚举每一个点,取其任意两个点,然后进行组合,然后两两相乘,得到最大值与他们的和。将所有值统计一下,然后注意因为和是组合出来的,所以再乘2,输出就可以了。
这里先申明一个坑点:题目只说了对和取模,没有说对最大值取模,所以要注意,如果对最大值取模就错了。
然后就是对于每一个点的问题了。如果直接组合,则n^2复杂度会TLE。我们可以使用乘法结合律和前缀和的思想。对于一个点,因为用邻接表存储,所以线性扫过所有与他相连的点,然后动态更新目前的和与目前这些点中的最大值,然后再到下一个点时,将下一个点的权值与这两个值相乘,然后做相应的处理就行了。类似于DP。
再注意一下细节,此题便做完了。
#include<cstdio> #include<cmath> using namespace std; struct Edge { int t,nexty; Edge(){t=nexty=0;} }edge[1000000]; int head[300000]={0},cnt=0; void add(int a,int b)//加边 { cnt++; edge[cnt].t=b,edge[cnt].nexty=head[a]; head[a]=cnt; } long long w[300000]={0}; int main() { int n; scanf("%d",&n); int a,b; for(int i=0;i<n-1;i++) { scanf("%d%d",&a,&b); add(a,b),add(b,a); } for(int i=1;i<=n;i++)scanf("%lld",&w[i]); long long sum=0,maxn=0; long long he,rmax; int node; for(int i=1;i<=n;i++) { node=head[i]; he=(rmax=w[edge[node].t])%10007; node=edge[node].nexty; for(;node!=0;node=edge[node].nexty) { sum=(sum+he*w[edge[node].t])%10007; maxn=max(maxn,rmax*w[edge[node].t]); he=(he+w[edge[node].t])%10007; rmax=max(rmax,w[edge[node].t]); } } printf("%lld %lld",maxn,(sum*2)%10007); return 0; }