【题解】洛谷P1351 联合权值（dfs、LCA）

最新推荐文章于 2025-03-25 19:12:09 发布

Rem_Inory

最新推荐文章于 2025-03-25 19:12:09 发布

阅读量377

点赞数

CC 4.0 BY-SA版权

分类专栏：题解 dfs 最小公共祖先文章标签：题解 dfs LCA 洛谷 NOIP

本文链接：https://blog.csdn.net/Rem_Inory/article/details/81485782

题解同时被 3 个专栏收录

140 篇文章

订阅专栏

dfs

10 篇文章

订阅专栏

最小公共祖先

4 篇文章

订阅专栏

一道关于洛谷P1351的题目，初始尝试使用最短路径算法导致得分不高。意识到问题在于图实际上是一棵树，可以利用DFS对每个节点进行扩展，但仍然只得到60分。进一步分析发现，题目关注的是两点间距离为2的情况。通过枚举每个节点作为中心，计算其子节点两两之间的联合权值，采用数学技巧简化计算过程，避免繁琐枚举。最终，计算所有节点为中心时的最大联合权值，并输出最大值和总和，同时记录子节点的最大和最小权值以优化算法。

这道题一开始啥也没想就用最短路写，才40分，然后发现自己对寻找最大值取模了，改了之后60分。。然后又发现n个点，n-1条边，其实这个图就是一棵树，每一个点到其余点的最短路有且只有一条，完全可以用dfs对每个点进行扩展，扩展两层找到点然后进行操作。。虽然看起来更简便了，但还是60分，所以我们得想更好的方法。

为啥它让你找距离为2的点而不是距离为3、4……的点呢？我们可以发现，对于一个中心结点，它的所有儿子相互之间的距离都是2。这个性质很重要，我们就可以利用它，枚举每一个点为中心结点，求出它所有儿子两两之间的联合权值，但不免有些繁琐。所以我们可以把它所有儿子的权值和相加，然后平方，最后减去每一个权值的平方，就能得到以该点为中心结点情况下所有儿子之间联合权值之和。出现加法或乘法就取模，注意减法取模可能为负数，在取模后要加上模数再取模。然后还要存下每个中心结点情况下的子节点权值最大值和最小值，相乘得到该情况下最大联合权值，最后将所有点为中心结点的最大联合权值取最大值就是输出的第一个结果，第二个结果就是每一个中心结点情况下联合权值之和。

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define mod 10007	
#define ll long long
using namespace std;
const int maxn=200010;
int head[maxn*2],nnext[maxn*2],to[maxn*2];
ll w[maxn];
int n,tot;
ll sum=0,ans=-1e9;
void add(int x,int y)
{
	tot++;
	nnext[tot]=head[x];
	head[x]=tot;
	to[tot]=y;
}
void dfs(int x)
{
	ll ans1=-1e9,ans2=-1e9;
	ll sum1=0;
	int p1;
	for(int i=head[x];i;i=nnext[i])
	{
		int y=to[i];
		sum1=(sum1+w[y])%mod;
		if(w[y]>ans1)
		{
			ans1=w[y];
			p1=y;
		}
	}
	sum1=sum1*sum1%mod;
	for(int i=head[x];i;i=nnext[i])
	{
		int y=to[i];
		sum1=((sum1-w[y]*w[y]%mod)%mod+mod)%mod;
		if(w[y]>ans2&&y!=p1)
		{
			ans2=w[y];
		}
	}
	sum=(sum+sum1)%mod;
//	cout<<ans1<<' '<<ans2<<' '<<sum%mod<<' '<<endl;
	ans=max(ans,ans1*ans2);
	return ;
}

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n-1;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);
	}
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&w[i]);
	}
	for(int i=1;i<=n;i++)
	{
		dfs(i);
		
	}
	printf("%lld %lld",ans,sum%mod);
	return 0;
}