树结构练习——判断给定森林中有多少棵树(并查集）

树结构练习——判断给定森林中有多少棵树

Time Limit: 1000ms   Memory limit: 65536K  

题目描述

 众人皆知，在编程领域中，C++是一门非常重要的语言，不仅仅因为其强大的功能，还因为它是很多其他面向对象语言的祖先和典范。不过这世上几乎没什么东西是完美的，C++也不例外，多继承结构在带来强大功能的同时也给软件设计和维护带来了很多困难。为此，在java语言中，只允许单继承结构，并采用接口来模拟多继承。KK最近获得了一份java编写的迷你游戏的源代码，他对这份代码非常感兴趣。这份java代码是由n个类组成的（本题不考虑接口），现在，他想要知道这份代码中有多少个类没有直接基类。n个类分别用数字1..n表示。

 

输入

 输入数据包含多组，每组数据格式如下。

第一行包含两个整数n,m，表示该份代码中的n个类和m个单继承关系。

后面m行，每行两个整数a b，表示a是b的直接基类。

输出

 对于每组输入，输出该组数据中有多少个类没有直接基类。每组输出占一行。

 

示例输入

2 1
1 2
2 0

示例输出

1
2

提示

这道题考察的是并查集的知识点。

首先来一个投机的解法：因为要求解没有直接基类的个数，可以转化为求解有基类的个数。

#include<iostream>
#include<cstring>
using namespace std;
int main()
{
    bool a[1000];
    int n,m;
    while(cin>>n>>m)
    {
        memset(a,0,sizeof(a));
        while(m--)
        {
            int x,y;
            cin>>x>>y;
            if(!a[y])           //只会被记录一次
            {
                a[y]=true;      //y有直接基类
                n--;
            }
        }
        cout<<n<<endl;
    }
    return 0;
}

这是采用并查集的方法：就是有“合并集合”和“查找集合”两种操作的关于数据结构的一种算法。

用集合中的某个元素来代表这个集合，该元素称为集合的代表元。一个集合内的所有元素组织成以代表元为根的树形结构。对于每一个元素 parent[x]指向x在树形结构上的父亲节点。如果x是根节点，则令pre[x] = x。对于查找操作，假设需要确定x所在的的集合，也就是确定集合的代表元。可以沿着pre[x]不断在树形结构中向上移动直到到达根节点。判断两个元素是否属于同一集合，只需要看他们的代表元是否相同即可。

#include<iostream>
using namespace std;
int pre[1000];
int n;
int Find(int x)             //查找根节点
{
    int r=x;
    while(r!=pre[r])
        r=pre[r];
//
//    int i=x,j;
//    while(i!=r)             //路径压缩
//    {
//        j=pre[i];
//        pre[i]=r;
//        i=j;
//    }
    return r;
}
void join(int x,int y)      //将x和y所在的子树合并
{
    x=Find(x);
    y=Find(y);
    if(x!=y)                // //如果已经连通，就不用管了 ；如果不连通，就把它们所在的连通分支合并起,
        {
            pre[x]=y;
            n--;
        }
}
int main()
{
    int m;
    while(cin>>n>>m)
    {
        int x=n;
        int a,b;
        for(int i=1;i<=n;i++)
            pre[i]=i;
        for(int i=0;i<m;i++)
            {
                cin>>a>>b;
                join(a,b);
            }
        cout<<n<<endl;
    }
    return 0;
}