CodeForces - 109C Lucky Tree

最新推荐文章于 2025-07-30 14:22:39 发布

sjyOvO

最新推荐文章于 2025-07-30 14:22:39 发布

阅读量304

点赞数

CC 4.0 BY-SA版权

文章标签：算法并查集 CodeForces

本文链接：https://blog.csdn.net/sjyisdog/article/details/120177425

该博客介绍了一个编程问题，涉及幸运边和非幸运边的概念。在给定的图中，幸运边连接的数字不能包含4或7，而其他边是非幸运边。通过并查集实现非幸运边的联通，并计算每组联通块内部节点数量的组合数，从而求得答案。代码实现了这一逻辑，对输入的边进行处理，并输出最终的答案。

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

原题链接：

https://vjudge.net/problem/CodeForces-109C

AC代码：

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <list>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <stdio.h>
#include <stdlib.h>

using namespace std;

typedef long long ll;
const int64_t mod = 1e9 + 7;
const int maxn = 1e6 + 10;
const int INF = 0x3f3f3f3f;
int n, fa[maxn], num[maxn];

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

void uni(int u, int v) {
    int xx = find(u);
    int yy = find(v);
    if (xx == yy) {
        return;
    }
    fa[yy] = xx;
    num[xx] += num[yy];
}

int check(int w) {
    while (w) {
        if (w % 10 != 4 && w % 10 != 7)return 0;
        w /= 10;
    }
    return 1;
}

int main() {
    cin >> n;
    for (int i = 1; i <= n; i++) {
        fa[i] = i;
        num[i] = 1;
    }
    int u, v, w;
    for (int i = 1; i < n; i++) {
        cin >> u >> v >> w;
        if (!check(w)) {
            uni(u, v);
        }
    }
    ll ans = 0;
    for (int i = 1; i <= n; i++) {
        int t = n - num[find(i)];
        ans += (ll) t * (t - 1);
    }
    printf("%I64d\n", ans);
    return 0;
}