CS 300 X Distance 思维+并查集

CS 300
题意:n点m条带权边的图,定义一条路径的代价为:该路径中边最大的权值.
n,m<=3e5,w[i],X<=1e9.问有多少对(i,j)满足i-j路径代价最小为X.


权值大于x的边不管,反正不可能经过这些边.
枚举权值为x的一条边.通过该边可以扩展出一个联通分量.


若去掉权值为x的边后,新图还是联通,则新图上的任意两点都可以不通过x互相到达. 该边不可能作为最小代价的最大权值边.
否则(u,v)该边贡献就为 u左右和v右边的点个数相乘. 若有多个x边？如何考虑? 


用并查集把权值小于x边合并 每次加入权值为x的边 若连通分量个数减小,则乘上合并的贡献即可.

官方解:计算最小代价最多为X的有多少对,把大于x边删除 计算每个联通分量size. 然后在把大于x-1边删除 在重复做一遍即可.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=3e5+5,mod=1e9+7;
int n,m,x,fa[N],sum[N];
struct node{
    int u,v,val;
}a[N];
bool cmp(node a,node b)
{
    return a.val<b.val;
}
int find(int x)
{
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
int main()
{
    cin>>n>>m>>x;
    for(int i=1;i<=n;i++)
        fa[i]=i,sum[i]=1;
    for(int i=0;i<m;i++)
        scanf("%d%d%d",&a[i].u,&a[i].v,&a[i].val);
    sort(a,a+m,cmp);
    ll ans=0;
    for(int i=0;i<m;i++)
    {
        if(a[i].val>x)
            break;
        int fx=find(a[i].u),fy=find(a[i].v);
        if(a[i].val<x)
        {
            if(fx!=fy)
                fa[fx]=fy,sum[fy]+=sum[fx];
        }
        else
        {
            if(fx!=fy)
            {
                ans+=sum[fy]*sum[fx];
                fa[fx]=fy,sum[fy]+=sum[fx];
            }
        }
    }
    cout<<ans<<endl;
    return 0;
}