[2016-03-17][HDU][1272][小希的迷宫]

• 时间:2016-03-17 15:01:10 星期四

• 题目编号:[2016-03-17][HDU][1272][小希的迷宫]

• 题目大意:给定若干条边,问这些边是否组成树

• 分析:

  • 只有一条通路—>树

• 方法:

  • 并查集合并边,如果遇到不能合并(祖宗节点一样),就不是树
  • 注意空树的情况
  • 给出的边号不一定按1~n的顺序,需要vis 做标记,刚开始vis数组忘记初始化

#define _WORK_
#ifdef _WORK_
#include <cstdio>
#include<cstring>
using namespace std;
typedef long long LL;
#define CLR(x,y) memset((x),(y),sizeof((x)))
#define FOR(x,y,z) for(int (x)=(y);(x)<(z);++(x))
const int maxn = 100000 + 100;
int fa[maxn],mrank[maxn],vis[maxn];
void ini(int n){
        CLR(mrank,0);
        CLR(vis,0);
        FOR(i,0,n + 1){
                fa[i] = i;
        }
}
int fnd(int x){
        return x == fa[x]?x:fa[x] = fnd(fa[x]);
}
int uni(int x,int y){
        x = fnd(x);
        y = fnd(y);
        if(x == y)      return 0;
        if(mrank[x] > mrank[y]){
                fa[y] = fa[x];
        }else {
                fa[x] = fa[y];
                if(mrank[x] == mrank[y])        ++mrank[y];
        }
        return 1;
}
int main(){
        //freopen("in.txt","r",stdin);
        //freopen("out.txt","w",stdout);
        int u,v,flg,cntcase = 0;
        while(~scanf("%d%d",&u,&v) && ~u && ~v){
                //空树
                if(!(u + v)){
                        puts("Yes");
                        continue;
                }
                ini(maxn - 1);
                flg = 1;
                if(flg){
                        if(uni(u,v)){
                                vis[v] = vis[u] = 1;
                        }
                        else flg = 0;                       
                }
                while(~scanf("%d%d",&u,&v) && u && v){
                        if(flg){
                                if(uni(u,v)){
                                        vis[v] = vis[u] = 1;
                                }
                                else flg = 0;                       
                        }
                }
                //判断不是森林
                int cnt = 0;
                if(flg){
                        FOR(i,0,maxn){
                                if(vis[i] && fa[i] == i){
                                        ++cnt;
                                }
                                if(cnt > 1){
                                        flg = 0;break;
                                }
                        }
                }
                if(flg) puts("Yes");
                else puts("No");
        }
        return 0;
}
#endif

来自为知笔记(Wiz)