A Simple problem

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

题意：给出顶点数，边数及节点s,判断s是否包含在所有的环中；

思路：并查集的应用，在除s节点以外的节点中，取出有公共源点的节点放到一个集合中，若还能形成环说明s不能经过所有的环，否则s能经过所有的环；

#include<stdio.h>
#include<string.h>

int set[10010],n,m,s;

int find(int x)
{
    if(set[x] != x)
        set[x] = find(set[x]);
    return set[x];
}

int main()
{
    int u,v;
    while(~scanf("%d %d %d",&n,&m,&s))
    {
        int flag = 1;
        for(int i = 1; i <= n; i++)
            set[i] = i;

        for(int i = 1; i <= m; i++)
        {
            scanf("%d %d",&u,&v);
            if(u != s && v != s)
            {
                int tu = find(u);
                int tv = find(v);
                if(tu == tv)
                    flag = 0;
                set[tu] = tv;
            }
        }
        if(flag == 1)
            printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}