CF109C Lucky Tree

最新推荐文章于 2024-10-04 23:00:36 发布

m0_73035684

最新推荐文章于 2024-10-04 23:00:36 发布

阅读量56

点赞数

分类专栏： CF刷题记录文章标签：图论算法

本文链接：https://blog.csdn.net/m0_73035684/article/details/128502586

版权

CF刷题记录专栏收录该内容

25 篇文章 1 订阅

订阅专栏

该博客介绍了一种利用并查集解决特定图论问题的方法。文章讨论了如何通过并查集来处理路径中至少包含一条幸运边的情况，从而找出满足条件的点对组合。在解决过程中，当遇到非幸运边时，将两个节点合并到同一连通块，最终统计不同连通块间的点对数量作为答案。代码示例展示了具体的实现过程。

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

知识点：思维，并查集

难度：5

这个题想到结论的话，那么不需要什么树形DP，用很简单的数据结构就可以过了，经过思考发现，题目说的是路径上至少有一条是幸运边，那么两点之间是符合答案的，那么如果两点之间的路径一条幸运边都没有呢，此时想到遇到不幸运的边，就把这两个结点放到一个连通块里面，具体操作也就是用并查集合并，这样处理完输入的边之后，得到就是若干个连通块（子树），每个连通块里面的点如果找答案的话（以三元组第一个数为基准找答案），是不能找和它在同一个连通块里面的点的，因为它到同一个连通块里面的点是一条幸运边都不会经过的，但是到其它连通块的点就可以至少通过一条边，这样我们就可以遍历点统计答案了，每个点能选的就是除了自己所在连通块的点，用组合公式累加就行了

#include <bits/stdc++.h>

using namespace std;

const int N = 1e5 + 5;

int fa[N], d[N];

int get(int x) {
	if (x == fa[x]) return fa[x];
	return fa[x] = get(fa[x]);
}

void merge(int x, int y) {
	int fx = get(x), fy = get(y);
	fa[fx] = fy;
	if (fx != fy) d[fy] += d[fx];
}

int main() {
	for (int i = 0; i < N; i++) {
		fa[i] = i;
		d[i] = 1;
	}
	int n;
	cin >> n;
	for (int i = 1; i < n; i++) {
		int x, y, z;
		scanf("%d%d%d", &x, &y, &z);
		int ok = 1;
		while (z) {
			int tmp = z % 10;
			if (tmp != 7 && tmp != 4) { ok = 0; break; }
			z /= 10;
		}
		if (!ok) merge(x, y);
	}
	long long ans = 0;
	for (int i = 1; i <= n; i++) {
		int t = n - d[get(i)];
		ans += (long long) t * (t - 1);
	}
	cout << ans;
	return 0;
}