【SSL 2373】【luogu P1351】【DG特长生2020 T4】星座图 / 联合权值

SSL_TJH

于 2021-05-05 17:02:48 发布

阅读量102

点赞数

分类专栏： # 图论文章标签：图论

要转发的话记得附上链接哦~（不过也不会有人转发的啦）

本文链接：https://blog.csdn.net/weixin_43346722/article/details/116426857

版权

图论专栏收录该内容

121 篇文章 2 订阅

订阅专栏

这篇博客介绍了如何解决一个关于树形结构的问题，其中的任务是找到所有两点间距离为2的点对，并计算它们点权的积之和与最大值。博主给出了利用乘法分配律优化的算法，通过枚举节点和维护最大第二大点权，有效地计算了最大值和总和，避免了超时的情况。

摘要由CSDN通过智能技术生成

星座图 / 联合权值

题目链接：SSL 2373 / luogu P1351

题目大意

给你一个树，一对点满足条件，要它们的距离为 2。（边的距离为 1）
然后有点权，求满足的点对中，每对点两个点点权的积的和以及最大值。
和要对 10007 取模。

思路

你看到这个距离为 $2$ 的很特别，那你考虑怎样就会距离为 $2$ 。

那你会发现，从一个点可以连向到若干个点，这若干个点两两之间距离都是 $2$ 。
那最大值就很好求了，我们直接枚举每个点，求与它相邻的点中点权最大的那两个，然后乘起来。
然后每个点都这么求一遍，取个最大值就是答案。
（最大值不用取模）

接着就是求和。
那你求每个点相邻的那一堆，然后再把每个加起来。
那每一堆怎么求呢？
两两配对都可以，但是我们枚举两个点就会超时。

那我们考虑乘法分配律一下，对于每个点，它都可以和别的点配对。
设点权和是 $s u m$ ，这个点点权是 $x$ ，这个点和别的配对的贡献就是 $x (s u m - x)$ 。
（ $xx_1+xx_2+...xx_n=x(x_1+x_2+...+x_n)=x(sum-x)$ ，这里的 $x_i$ 是不包含 $x$ 的其它点，所以和是 $s u m - x$ ）

然后因为 $a, b$ 的组合和 $b, a$ 的组合是不同的，所以我们并不用除二。

代码

#include<cstdio>
#include<iostream>
#define mo 10007
#define ll long long

using namespace std;

struct node {
	int to, nxt;
}e[600001];
int n, le[200001], KK;
int x, y;
ll ans, maxn, qz[200001];

void add(int x, int y) {
	e[++KK] = (node){y, le[x]}; le[x] = KK;
	e[++KK] = (node){x, le[y]}; le[y] = KK; 
}

int main() {
//	freopen("link.in", "r", stdin);
//	freopen("link.out", "w", stdout);
	
	scanf("%d", &n);
	for (int i = 1; i < n; i++) {
		scanf("%d %d", &x, &y);
		add(x, y);
	}
	for (int i = 1; i <= n; i++) scanf("%lld", &qz[i]);
	
	for (int i = 1; i <= n; i++) {
		ll first = 0, second = 0, sum = 0;
		for (int j = le[i]; j; j = e[j].nxt) {
			if (qz[e[j].to] > first) {//维护最大和第二大
				second = first;
				first = qz[e[j].to];
			}
			else if (qz[e[j].to] > second) {
				second = qz[e[j].to];
			}
			sum = (sum + qz[e[j].to]) % mo;
		}
		maxn = max(maxn, first * second);
		
		for (int j = le[i]; j; j = e[j].nxt) {//用每个点乘剩余点的和，每个点都这么乘一次，就可以搞出两两乘的和结果
			sum = ((sum - qz[e[j].to]) % mo + mo) % mo;
			ans = (ans + (sum * qz[e[j].to]) % mo) % mo;
			sum = (sum + qz[e[j].to]) % mo;
		}
	}
	
	printf("%lld %lld", maxn, ans);
	
	fclose(stdin);
	fclose(stdout);
	
	return 0;
}