POJ 1308 Is It A Tree?

并查集水题,刚开始做并查集,错误次数有点多...交20+发,大多RE。以下为代码,注意的地方已标出

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define INF 1000000000
#define max_N 100005
int par[max_N+5];//为初始化不越界故+5
int rank[max_N+5];
void init(int n)
{
    int i;
    for(i=0;i<=n;i++)
    {
        par[i]=i;
        rank[i]=0;
    }
}
int find(int x)
{
    if(par[x]==x)return x;
    else return par[x]=find(par[x]);
}
void uni(int x,int y)
{
    x=find(x);
    y=find(y);
    if(x==y)return;
    if(rank[x]<rank[y])par[x]=y;
    else
    {
        par[y]=x;
        if(rank[x]==rank[y])rank[x]++;
    }
}
int jud(int x,int y)
{
    return find(x)==find(y);
}
int v[max_N+5];
int main()
{
    int num=1,flag,x,i,k,flag2;
    int n,m;
    while(1)
    {
        k=0;
        init(max_N+1);//初始化注意不要越界
        flag=0;flag2=0;//一开始将x初始化为INF了,直接导致RE。0 0数据会导致x=INF,导致RE
        while(1)
        {
            scanf("%d%d",&n,&m);
            if(n==0&&m==0)break;//此处break不能后置,否则v[0]也被记录导致后面查询根节点时,误以为0为另外一颗树
            if(n==-1&&m==-1)break;
            v[k]=n;v[k+1]=m;k=k+2;
            if(n==m)flag=1;
            if(find(n)==m)flag=1;
            if(find(m)==m){uni(n,m);if(flag2==0)x=n;flag2=1;}//本来以为uni(n,m);可以用par[m]=n;代替,但是超内存...par[m]=find(n);正确。不明白为什么超内存...先过去吧
            else {flag=1;}
        }
        if(n==-1&&m==-1)break;
        x=find(x);
        for(i=0;i<k;i++)
        {
            if(find(v[i])!=x){flag=1;break;}
        }
        if(flag)printf("Case %d is not a tree.\n",num++);
        else printf("Case %d is a tree.\n",num++);
    }

    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值