Jzoj3931【NOIP2014day1官方数据】联合权值

最新推荐文章于 2021-01-18 20:18:11 发布

原创最新推荐文章于 2021-01-18 20:18:11 发布 · 525 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#noip #树

OI 同时被 3 个专栏收录

321 篇文章

订阅专栏

Jzoj

227 篇文章

订阅专栏

数据结构

134 篇文章

订阅专栏

本文解析了一道图论算法竞赛题目，通过计算每个节点与其相连节点权值和的贡献来找出最大值，并通过排序优化求解过程。采用C++实现，展示了具体的算法流程和代码细节。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意略过，口胡正解

对于每个节点i，我们求出每个与i相连的节点的权值和s，那么i的贡献就是∑d[j]*(s-d[j])

求最大就排个序咯

#include<stdio.h>
#include<vector>
#include<algorithm>
#define M 10007
using namespace std;
int n,d[200010],mx=0,sum=0;
vector<int> G[200010];
inline bool c1(int a,int b){ return d[a]>d[b]; }
int main(){
	scanf("%d",&n);
	for(int x,y,i=1;i<n;++i){
		scanf("%d%d",&x,&y);
		G[x].push_back(y);
		G[y].push_back(x);
	}
	for(int i=1;i<=n;++i) scanf("%d",d+i);
	for(int S,i=1;i<=n;++i){
		if(G[i].size()<2) continue;
		sort(G[i].begin(),G[i].end(),c1);
		mx=max(mx,d[G[i][0]]*d[G[i][1]]); S=0;
		for(int j=0,z=G[i].size();j<z;++j) S+=d[G[i][j]];
		for(int j=0,z=G[i].size();j<z;++j) sum=((1ll*(S-d[G[i][j]]+M)*d[G[i][j]])%M+sum)%M;
	}
	printf("%d %d\n",mx,sum);
}