HDU3938 Portal 并查集

最新推荐文章于 2021-07-27 11:41:07 发布

转载最新推荐文章于 2021-07-27 11:41:07 发布 · 689 阅读

并查集专栏收录该内容

17 篇文章

订阅专栏

本文介绍了一种使用并查集解决图论问题的方法，通过处理边的排序及查询长度来找出任意两点间的最长边不超过给定长度的所有可能组合。

10 10 10
7 2 1
6 8 3
4 5 8
5 8 2
2 8 9
6 4 5
2 1 5
8 10 5
7 3 7
7 8 8
10 //查询长度
6
1
5
9
1
8
2
7
6

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int M = 10005;

/*
本题题意有必要说一下， 给出一系列查询长度， 要求你找出任意两点间的最长边小于等于给出的查询
长度， 做法是对给出的边排序，具体看代码， 我也是看了别人写的才明白题意的。
*/
struct node {

    int x;
    int y;
    int d;

}edge[M*5];

struct Node {

    int len;
    int ans;
    int id;

}query[M];

int p[M];
int sum[M];
int n, m, q;

bool cmp(node p1, node p2) {

    return p1.d < p2.d;
}

bool cmp1(Node p1, Node p2) {

    return p1.len < p2.len;
}

bool cmp2(Node p1, Node p2) {

    return p1.id < p2.id;
}

void init() {

    for(int i = 1; i <= n; i++) {

        p[i] = i;
        sum[i] = 1;
    }
}

int Find(int a) {

    return a == p[a] ? a : p[a] = Find(p[a]);
}

int Union(int a, int b) {

    a = Find(a);
    b = Find(b);
    if(a == b)
        return 0;
    else {

        p[a] = b;
        int temp = sum[a]*sum[b];
        sum[b] += sum[a];
        return temp;
    }
}

int main()
{
    while(scanf("%d%d%d", &n, &m, &q) != EOF) {

        init();
        for(int i = 0; i < m; i++) {

            scanf("%d%d%d", &edge[i].x, &edge[i].y, &edge[i].d);
        }
        for(int i = 0; i < q; i++) {

            scanf("%d", &query[i].len);
            query[i].id = i;
            query[i].ans = 0;
        }
        sort(edge, edge+ m, cmp);
        sort(query, query+q, cmp1);
        int cnt = 0;
        for(int i = 0; i < q; i++) {

            while(edge[cnt].d <= query[i].len && cnt < m) {

                query[i].ans += Union(edge[cnt].x, edge[cnt].y);
                    cnt++;
            }
            if(i > 0)
                query[i].ans += query[i-1].ans;
        }
        sort(query, query+q, cmp2);

        for(int i = 0; i < q; i++) {

            printf("%d\n", query[i].ans);
        }
    }
    return 0;
}