A simple problem(并查集判环)

http://acm.sdut.edu.cn/sdutoj/problem.php?action=showproblem&problemid=2497

 题意：给定一些点和边的关系，判断S点是否在所构成无向图的所有环里。

 思路：用并查集将所有（除去S及与 S有关的点）有关系的点放在一个集合里，若此时图中还存在环，那么一定不包含S。

#include <stdio.h>
#include <string.h>
const int maxn = 10002;

int f[maxn],n;
int find(int x)
{
    if (x!=f[x])
        f[x] = find(f[x]);
    return f[x];
}
void merge(int x,int y)
{
    x = find(x);
    y = find(y);
    if (x!=y)
        f[x] = y;
}
void init()
{
    for (int i = 0; i <= n; i ++)
        f[i] = i;
}
int main()
{
    int m,s;
    while(~scanf("%d%d%d",&n,&m,&s))
    {
        init();
        int u,v,flag = 1;
        for (int i = 0; i < m; i ++)
        {
            scanf("%d%d",&u,&v);
            if (u!=s && v!=s)
            {
                if (find(u)==find(v))//判断环
                    flag = 0;
                merge(u,v);//合并集合
            }
        }
        if (flag)
            printf("YES
");
        else
            printf("NO
");
    }
    return 0;

}