BOJ 1501 多彩瓜 并查集

Description
    dalong的舅舅最近到外地旅游时给dalong带回来一种神奇的水果：多彩瓜。这种瓜太漂亮了，它有很多层，每一层都是一种颜色。由于多彩瓜放久了会变坏，dalong决定每天吃一层，可是多彩瓜有好多层，每天吃一层的话还没吃完多彩瓜就坏了，因为多彩瓜颜色很多，dalong把一些很相近的颜色认为是同一种颜色，这样合并颜色后多彩瓜就没有原来这么多层了，dalong就可以在多彩瓜变坏之前吃完它喽~oh，yeah。现在dalong告诉你多彩瓜最初每一层的颜色，颜色之间的关系，你能帮助他算出吃完这些瓜要几天么？注意，如果dalong认为颜色a和颜色b是同一种颜色，颜色b和颜色c是同一种颜色，那么颜色a和颜色c也是同一种颜色。


Input
    多组数据测试
    对于每组数据
    输入的第一行是一个正整数n(1 <= n <= 50000)，m(1 <= m <= 100)表示多彩瓜最初有n层,一共有m种不同的颜色。
    第二行有n个数，第i个数colori表示多彩瓜第i层的颜色(1 <= colori <= m).
    第三行是一个正整数k(0 <= 3000 <= k)，表示有k组颜色存在相似认为相同的关系。
    后面k行，每行两个正整数a,b(1 <= a, b <= m 且 a != b),表示dalong认为a和b是同一种颜色。
    输入的最后是两个0，表示输入结束，这组数据不用处理


Output
对于每组数据，输出dalong需要几天吃完多彩瓜


Sample Input

5 3
1 1 2 3 1
1
1 2
0 0


Sample Output

3


Hint
    因为1和2是同一种颜色，那dalong第一天可以吃掉前3层(1,1,2),第二天可以吃掉第四层(3),第三天吃掉最后一层(1)。因此吃完多彩瓜需要3天。

思路：很显然是点的合并，用并查集。

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
const int N=50005,M=105;
int n,m;
int in[N];
int father[N];
int find(int u)
{
    while(father[u]>0)
    {
        u=father[u];
    }
    return u;
}
void Union(int a,int b)
{
    if(father[a]<father[b])
    {
        father[a]+=father[b];

        father[b]=a;
        
    }
    else
    {
        father[b]+=father[a];
        father[a]=b;
    }

}
int main()
{
    while(scanf("%d%d",&n,&m),n!=0||m!=0)
    {
        memset(father,-1,sizeof(father));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&in[i]);
        }
        int tmp;
        scanf("%d",&tmp);
        for(int i=1;i<=tmp;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            if((u=find(u))!=(v=find(v)))
            {
                Union(u,v);
            }
        }
        int ans=0;
        int now=0;
        for(int i=1;i<=n;i++)
        {
            int tmp=find(in[i]);
            if(tmp!=now)
            {
                now=tmp;
                ans++;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}