HDU 4750 Count The Pairs kruskal

http://acm.hdu.edu.cn/showproblem.php?pid=4750

题意：

　　给出一些边，一个点到另一个点的一条路径，这条路径上的权值尽可能小，然后他给你一个t 值，

然你计算比 t 大的这些路径有多少条。

 

坑爹：

　　看了下边和点的数量，暴力是不行的，但也还是试了一下，因为当时实在想不到怎么做，就每次找

比 t 大的边然后向这条边上的两个点外拓展，最后只用计算一下两边点个数相乘再乘以二，最后发现是

不行的，因为我这次找到比 t 大所对应的边然后拓展出去所经过的边也有可能大于 t ， 但是下次拓展到

这个边的时候就会有重复，一下子就不知道怎么弄了。

 

解法：

　　开始的时候也想过要做一些处理，因为那么大的数据暴力一定超时，但却忘记了kruskal加边的特性，

就是每次加最小的边，这样我们加边的时候每次以这条边上的两个点拓展的时候所经过的边一定小于原来

这个边。

 

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

const int maxn = 500000 + 10;

struct Edge {
    int v;
    int u;
    int w;
}edge[maxn];
int n;
int m;
int father[maxn];
int sum[maxn];
int total[maxn];
vector<int> V;
int cnt;

void init()
{
    V.clear();
    int i;
    for(i=0; i<maxn; i++)
    {
        father[i] = i;
        sum[i] = 1;
        total[i] = 0;
    }
}

int cmp(Edge e1,Edge e2)
{
    return e1.w < e2.w;
}

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

void kruskal()
{
    int i;
    cnt = 1;
    for(i=0; i<m; i++)
    {
        Edge e = edge[i];
        int root_A = find(e.u);
        int root_B = find(e.v);
        if(root_A != root_B)
        {
            V.push_back(e.w);
            total[cnt++] = sum[root_A] * sum[root_B] * 2;
            sum[root_B] += sum[root_A];
            father[root_A] = root_B;
        }
    }
    for(i=1; i<cnt; i++)
    {
        total[i] += total[i-1];
    }
}

int main()
{
    while(scanf("%d%d",&n,&m) != EOF)
    {
        init();
        int i;
        for(i=0; i<m; i++)
        {
            int u;
            int v;
            int w;
            scanf("%d%d%d",&u,&v,&w);
            edge[i].u = u;
            edge[i].v = v;
            edge[i].w = w;
        }
        sort(edge,edge+m,cmp);
        kruskal();

        int p;
        scanf("%d",&p);
        while(p--)
        {
            int t;
            scanf("%d",&t);
            int temp = lower_bound(V.begin(), V.end(), t) - V.begin();
            printf("%d\n",total[cnt-1] - total[temp]);
        }
    }
    return 0;
}