Codevs 3728 联合权值 题解

3728 联合权值

 时间限制: 1 s

 空间限制: 128000 KB

 题目等级 : 黄金 Gold

题
解

题目描述 Description

输入描述 Input Description

输出描述 Output Description

样例输入 Sample Input

样例输出 Sample Output

数据范围及提示 Data Size & Hint

#include<iostream>#include<cstdio>#include<vector>using namespace std;vector<int>t[200010];int a[200010],u,v,sum=0,pow=0,m1=0,m2=0,n,k,ans,sm;int main(){cin>>n;for(int i=1;i<=n-1;i++){cin>>u>>v;t[u].push_back(v);t[v].push_back(u);}for(int i=1;i<=n;i++){cin>>a[i];}for(int i=1;i<=n;i++){    m1=m2=0;sum=pow=0;for(int j=0;j<t[i].size();j++){k=a[t[i][j]];sum+=k;sum%=10007;pow+=(k*k);pow%=10007;if(k>m1){m2=m1;m1=k;}else if(k>m2)m2=k;}//cout<<i<<' '<<m1*m2<<' '<<sum*sum-pow<<endl;;sum=sum%10007;pow=pow%10007;if(sm<(m1*m2))sm=m1*m2;ans+=(sum*sum-pow);ans%=10007;}//for(int i=0;i<n;i++)for(int j=0;j<t[i].size();j++){//k=a[t[i][j]];//cout<<i<<j<<k<<endl;//}cout<<sm<<' '<<ans;return 0;}

时间复杂度 O(n)

权值为2→同一个点相邻的两个点

用（m1+m2+m3+……）^2-m1^2-m2^2-……累加求和

注意最大值和次大值 注意最大值和次大值 注意最大值和次大值