码迷,mamicode.com
首页 > 其他好文 > 详细

hdu 5441 Travel 长春网赛

时间:2015-09-13 21:29:37      阅读:167      评论:0      收藏:0      [点我收藏+]

标签:

想到最小生成树的sort+并查集算法,于是可以顺便用并查集维护点所在的连通分量的点的数目(不知道学名是不是这么说),记为poi[v];

然后当边权限制为f[i].w时,其答案为ww[i]=ww[i-1]-(poi[u]-1)*poi[u]-(poi[v]-1)*poi[v]+(poi[u]+poi[v])*(poi[u]+poi[v]-1);

之后连通u,v,并将新的poi[u]更新至并查集的树根。这样就相当于用最小生成树+并查集打了个表了,对于询问二分即可

自己YY的方法,很多地方丑

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int INF=0x7f7f7f7f;
const int maxm=100008;
const int maxn=20008;
struct fuck{
    int u,v,w;
    bool operator<(const fuck &a)    const
    {
        return w<a.w;
    }
}f[maxm<<1];
struct shit{
    int u,w;
}ww[maxm];
int poi[maxn],uni[maxn];
int find(int x)
{
    if(uni[x]!=x)
    {
        uni[x]=find(uni[x]);
        poi[x]=max(poi[x],poi[uni[x]]);
        return uni[x];
    }
    return x;
}
int bs(int w,int idx)
{
    ww[idx].u=INF;
    int left=1,right=idx;
    while(left<right)
    {
        int mid=(right+left)>>1;
        if(ww[mid].u>w)
            right=mid;
        else
            left=mid+1;
    }
    return left-1;
}
void update(int x)
{
    if(uni[x]!=x)
    {
        poi[uni[x]]=max(poi[x],poi[uni[x]]);
        find(uni[x]);
    }
}
int main()
{
    int t,i,j,n,m,q,u,v,w;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d",&n,&m,&q);
        for(i=1;i<=m;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            f[i].u=u;f[i].v=v;f[i].w=w;
        }
        sort(f+1,f+1+m);
        for(i=1;i<=n;i++)
        {
            uni[i]=i;
            poi[i]=1;
        }
        int idx=1;
        ww[0].w=0;
        for(i=1;i<=m;i++)
        {
            if(find(f[i].u)!=find(f[i].v))
            {    
                ww[idx].w=ww[idx-1].w;
                poi[f[i].u]=max(poi[f[i].u],poi[uni[f[i].u]]);
                poi[f[i].v]=max(poi[f[i].v],poi[uni[f[i].v]]);
                uni[uni[f[i].u]]=uni[f[i].v];
                if(poi[f[i].u]>1)    ww[idx].w-=poi[f[i].u]*(poi[f[i].u]-1);
                if(poi[f[i].v]>1)    ww[idx].w-=poi[f[i].v]*(poi[f[i].v]-1);
                poi[f[i].u]+=poi[f[i].v];
                poi[f[i].v]=poi[f[i].u];
                update(f[i].u);
                update(f[i].v);
                ww[idx].u=f[i].w;
                ww[idx++].w+=poi[f[i].u]*(poi[f[i].u]-1);
            }
        }
        for(i=1;i<=q;i++)
        {
            scanf("%d",&w);
            int asd=bs(w,idx);
            printf("%d\n",ww[asd].w);
        }
    }
    return 0;
}

 

hdu 5441 Travel 长春网赛

标签:

原文地址:http://www.cnblogs.com/bitch1319453/p/4805503.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!