COJ1241(数字序列)

题目链接

并查集的题，食物链模型。

题目大意：已知关于n个数的m条信息，信息有两类，一类是已知Ai比Aj大K，另一类是已知Ai为K，问是否能从这m条信息中唯一确定这n个数，或者能推出矛盾。

#include <stdio.h>
#define N 10001
#define M 20001
int p[N],c[N],v[N];
int a[M],b[M],cnt;
char vis[N];
int n,m;
void make_set()
{
    int i;
    for(i=0;i<n;i++)    p[i]=i,c[i]=0,vis[i]=0;
}
int find_set(int i)
{
    int pi=p[i];
    if(i!=pi)   p[i]=find_set(p[i]);
    if(pi!=p[i])    c[i]+=c[pi];
    return p[i];
}
void union_set(int i,int j,int d)
{
    int pi=find_set(i),pj=find_set(j);
    p[pi]=pj;
    c[pi]=-c[i]+d+c[j];
}
int main()
{
    int i,j,k,d;
    char ch,RE,WA;
    while(~scanf("%d%d",&n,&m))
    {
        make_set();
        RE=WA=0;
        for(k=cnt=0;k<m;)
        {
            scanf("%c",&ch);
            if(ch=='B')
            {
                k++;
                scanf("%d%d%d",&i,&j,&d);
                i--,j--;
                if(find_set(i)!=find_set(j))    union_set(i,j,d);
                else if(c[i]-c[j]!=d)   RE=1;
            }
            else if(ch=='S')
            {
                k++;
                scanf("%d%d",&j,&d),j--;
                a[cnt]=j,b[cnt++]=d;
            }
        }
        for(i=0;i<n;i++)    find_set(i);
        for(i=0;!RE&&i<cnt;i++)
        {
            k=a[i];
            d=b[i];;
            j=find_set(k);
            if(!vis[j])
            {
                vis[j]=1;
                v[j]=d-c[k];
            }
            else if(v[j]!=d-c[k])   RE=1;
        }
        for(i=0;!RE&&!WA&&i<n;i++)  if(!vis[find_set(i)])   WA=1;
        if(RE)  puts("RE");
        else if(WA) puts("WA");
        else    puts("AC");
    }
    return 0;
}