hdu 5441 Travel 2015长春网络赛 图论 并查集

题目

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5441

题目来源：2015 长春网络赛

简要题意：给定一个带权无向图，权为路上时间，一个人只能忍受在路上$x$时间，询问$q$次当人能忍受的时间为$x_i$时能在多少组点之间走。

数据范围：$n\leqslant 20000;\quad m\leqslant 100000;\quad q\leqslant 5000\\edge\ (a,b)\quad d\leqslant 10^5$

题解

开始的时候题意没读明白，后来读明白了之后就会发现这是个像Kruskal一样的东西。

对边进行排序，query进行排序，在query的$x_i$变大之后并查集合并每条新能走的边的两端。

要获取结果的话只要知道每个连通分支中有多少个点就行了，有$v$个点则这个分支中含有点对$\frac{v(v-1)}{2}$对。

维护一下每个连通分支的点数，离线处理出答案再按照顺序输出即可。

实现

实际上每次query的时候遍历一下 n <script type="math/tex" id="MathJax-Element-521">n</script>个节点是会TLE的。

对于这个问题其实很好解决，只要动态进行更新就行了。

合并两个节点的时候只会增加两个节点之间的边。

更新的方式我的方法是直接先把两个分支内的结果去掉，再加上合并后的分支的结果即可（点之间各连一条边）。

还有一种更加好的写法应该是直接加上两分支规模的乘积。

在代码中我将两种更新方法都列出了。

代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <stack>
#include <queue>
#include <string>
#include <vector>
#include <set>
#include <map>

#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define sz(x) ((int)(x).size())
#define fi first
#define se second
using namespace std;
typedef long long LL;
typedef vector<int> VI;
typedef pair<int,int> PII;
LL powmod(LL a,LL b, LL MOD) {LL res=1;a%=MOD;for(;b;b>>=1){if(b&1)res=res*a%MOD;a=a*a%MOD;}return res;}
// head
struct Node {
    int from, to, v;
    Node(int from, int to, int v) : from(from), to(to), v(v) {};
    Node() {};
    bool operator<(const Node &o) const {
        return v < o.v;
    }
};

const int N = 20005;
const int M = 1e5+5;
const int Q = 5e3+5;

LL ans = 0;
int r[N];
int fa[N];
Node a[M];
PII q[Q];
LL res[Q];

int find(int key) {
    return (key == fa[key]) ? key : (fa[key] = find(fa[key]));
}

void init(int n) {
    for (int i = 1; i <= n; i++) {
        r[i] = 1;
        fa[i] = i;
    }
}

int get(int x) {
    return x*(x-1)/2;
}

void joint(int u, int v) {
    int a = find(u), b = find(v);
    if (a != b) {
        if (r[a] < r[b]) {
            swap(a, b);
        }
        fa[b] = a;
        ans -= get(r[a])+get(r[b]);
        r[a] += r[b];
        ans += get(r[a]);
//      fa[b] = a;
//      ans += r[a]*r[b];
//      r[a] += r[b];
    }
}

bool same(int u, int v) {
    return find(u) == find(v);
}

int main()
{
    int t, n, m, qq;
    scanf("%d", &t);
    while (t--) {
        scanf("%d%d%d", &n, &m, &qq);
        init(n);
        for (int i = 0; i < m; i++) {
            scanf("%d%d%d", &a[i].from, &a[i].to, &a[i].v);
        }
        for (int i = 0; i < qq; i++) {
            scanf("%d", &q[i].fi);
            q[i].se = i+1;
        }

        sort(a, a+m);
        sort(q, q+qq);
        int x = 0;
        ans = 0;
        for (int i = 0; i < qq; i++) {
            for (; a[x].v <= q[i].fi && x <= m; x++) {
                joint(a[x].from, a[x].to);
            }
            res[q[i].se] = ans*2;
        }

        for (int i = 1; i <= qq; i++) {
            printf("%I64d\n", res[i]);
        }
    }
    return 0;
}