P1351 联合权值 题解

博客园同步

原题链接

简要题意：

给定一棵树，每两个距离为 $2$ 的点 $u,v$ 会产生 $w_u\times w_v$ 的“联合权值”。求 “联合权值” 的和，以及所有联合权值中的最大值。

其实这题作为 $\text{NOIP 2014tg Day1T2}$，并不难。

首先考虑：距离为 $2$ 的点只有两种情况：

1. 爷爷和孙子的关系。

2. 弟弟和哥哥的关系。

具体到树上就是，$u=f{a}_{f{a}_v}$，或者 $f{a}_u=f{a}_v$，都会产生 $w_u\times w_v$ 的贡献。

对于求最大值，我们可以遍历每个节点 $i$，统计 $i$ 的爷爷和自己的贡献，以及所以 $f{a}_u=f{a}_v=i$ 的最大值，取出 $i$ 儿子中的最大值和次大值即可。

对于求和，我们发现，如果 $u$ 的第 $v$ 个儿子为 $su{b}_{u,v}$，且共 $k$ 个儿子，则贡献为：

$$\sum _{i=1}^k\sum _{j=i+1}^k{w}_{su{b}_{u,i}}\times w_{su{b}_{u,j}}$$

$$=\frac{{\sum }_{i=1}^k{\sum }_{j=1}^k{w}_{su{b}_{u,i}}\times w_{su{b}_{u,j}}-{\sum }_{i=1}^k{w}_{su{b}_{u,i}}\times w_{su{b}_{u,i}}}{2}$$

（两两的乘积，先总的重复计算，减去自己乘自己的，交换 $÷2$）

$$=\frac{({\sum }_{i=1}^k{w}_{su{b}_{u,i}}{)}^2-{\sum }_{i=1}^k(w_{su{b}_{u,i}}{)}^2}{2}$$

太精妙了！！！

所以统计 平方和 和 和的平方 即可。平方和和和的平方

时间复杂度：$O(n)$.

实际得分：$100pts$.

细节：总和要取模，最大值不用。

#pragma GCC optimize(2)
#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
const ll MOD=1e4+7;
const int N=2e5+1;

inline int read(){char ch=getchar();int f=1; while(!isdigit(ch)) {if(ch=='-') f=-f; ch=getchar();}
	   int x=0;while(isdigit(ch)) x=x*10+ch-'0',ch=getchar(); return x*f;}

ll n,w[N];
vector<int> G[N];
vector<int> son[N];
ll ans1,ans2;

inline void dfs(ll dep,ll fa,ll gr) {
//正在处理 dep,父亲节点为 fa,祖父(爷爷) 节点为 gr
	ll zd=0,cd=0,sum1=0,sum2=0;
	for(ll i=0;i<G[dep].size();i++) {
		ll v=G[dep][i];
		if(v!=fa) {
			sum1=(sum1+w[v])%MOD;
			sum2=(sum2+w[v]*w[v]%MOD)%MOD; //两边一起统计
			if(w[v]>zd) cd=zd,zd=w[v];
			else if(w[v]>cd) cd=w[v]; //最大值，次大值
			dfs(v,dep,fa); //往下搜索
		}
	} sum1=(sum1*sum1)%MOD; //和的平方
	ans1=max(ans1,max(cd*zd,w[dep]*w[gr])); //统计最大值
	ans2=(ans2+(sum1+MOD-sum2)%MOD+w[dep]*w[gr]*2%MOD)%MOD; //统计和
}

int main(){
	n=read();
	for(ll i=1;i<n;i++) {
		ll u=read(),v=read();
		G[u].push_back(v);
		G[v].push_back(u);
	} for(ll i=1;i<=n;i++) w[i]=read();
	dfs(1,0,0);
	printf("%lld %lld\n",ans1,ans2);
	return 0;
}