bzoj 1202 [HNOI2005]狡猾的商人——带偏移量的并查集

题目:https://www.lydsy.com/JudgeOnline/problem.php?id=1202

带偏移量的并查集。

注意先 find() 再调用 dis !!!

自己的对拍太水了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=105;
int T,n,m,fa[N],dis[N];
bool flag;
int rdn()
{
  int ret=0;bool fx=1;char ch=getchar();
  while(ch>'9'||ch<'0'){if(ch=='-')fx=0;ch=getchar();}
  while(ch>='0'&&ch<='9') ret=(ret<<3)+(ret<<1)+ch-'0',ch=getchar();
  return fx?ret:-ret;
}
int find(int a)
{
  if(fa[a]==a)return a;
  int x=find(fa[a]);
  dis[a]+=dis[fa[a]]; return fa[a]=x;
}
int main()
{
  T=rdn();
  while(T--)
    {
      n=rdn(); m=rdn(); flag=0;
      for(int i=0;i<=n;i++)fa[i]=i,dis[i]=0;
      for(int i=1,x,y,s,u,v;i<=m;i++)
    {
      x=rdn()-1; y=rdn();
      u=find(x); v=find(y);
      s=rdn()-dis[x]+dis[y];
      if(u!=v)
        {
          dis[u]=s; fa[u]=v;
        }
      else if(s!=0)flag=1;
    }
      puts(flag?"false":"true");
    }
  return 0;
}

 

posted on 2018-10-29 22:27  Narh  阅读(87)  评论(0编辑  收藏  举报

导航