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

bzoj1202--带权并查集+前缀和

时间:2016-07-18 13:39:46      阅读:176      评论:0      收藏:0      [点我收藏+]

标签:

http://www.lydsy.com/JudgeOnline/problem.php?id=1202

      记s[i]=a[1]+a[2]+...+a[i],即s[i]为前缀和。再令v[i]=s[f[i]]-s[i],其中f[i]为i的父亲。对于每个读入的x,y,k,将x,y视为结点,如果x与y的根结点相同,因为v[y]-v[x]=s[f[y]]-s[y]-s[f[x]]-s[x]且f[y]=f[x],所以v[y]-v[x]就是区间[x,y]的和,所以只需要判断v[y]-v[x]是否等于k就可以了。如果x与y的根结点不相同,合并两个节点并更新信息。

#include<cstdio>
#include<cstring>
using namespace std;
int n,i,j,m,f[101],w,x,y,k,v[101];
bool flag;
int find(int x){
    if(x==f[x])return x;
    int tmp=f[x];                     //记录原来的父亲
    f[x]=find(f[x]);
    v[x]+=v[tmp];                     //更新结点信息
    return f[x];
}
bool union1(int x,int y,int k){
    int fx=find(x),fy=find(y);
    if(fx==fy){
        if(v[y]-v[x]!=k)return 0;
    }else{
        f[fy]=fx;
        v[fy]=k-v[y]+v[x];             //更新结点信息
    }
    return 1;
}
int main()
{
    scanf("%d",&w);
    while(w--){
        memset(v,0,sizeof(v));
        flag=0;
        scanf("%d%d",&n,&m);
        for(i=0;i<=n;i++)f[i]=i;
        for(i=1;i<=m;i++){
            scanf("%d%d%d",&x,&y,&k);
            if(!flag&&!union1(x-1,y,k))flag=1;
        }
        if(flag)printf("false\n");else printf("true\n");
    }
    return 0;
}

 

bzoj1202--带权并查集+前缀和

标签:

原文地址:http://www.cnblogs.com/gjghfd/p/5680075.html

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