PAT A 1118 Birds in Forest (25分)

一、题目概述

依据给定信息，求解：（1）树的数量；（2）鸟的数量；（3）判断任两只鸟是否在同一个树上。
信息受以下规范约束：
1、每张照片上的鸟都在同一棵树上（可能有多张照片描述同一棵树）；
2、鸟编号连续，从1开始，且所有鸟都在照片中出现。

二、思路一： 图DFS搜索

1、输入的同时统计鸟的数量。
2、鸟和照片作为结点构造无向图，则每个连通分量就是一棵树，树的数量即连通分量数。
3、为判断鸟是否属于同一棵树，对连通分量编号并记录。

代码实现：

#include <cstdio>
#include <vector>
using namespace std;
vector<int> G[20001], visited(20001, 0), tree(20001, 0);
void dfs( int v, int tag )
{
    visited[v] = 1;
    tree[v] = tag;
    for( int i = 0; i < G[v].size(); ++i )
        if( !visited[ G[v][i] ] )
            dfs( G[v][i], tag );
}
int main()
{
    int N, Q, n_bird = 0, n_tree = 0;
    scanf("%d", &N);
    for( int i = 1, K; i <= N; ++i )
    {
        scanf("%d", &K);
        for( int j = 0, bird; j < K; ++j )
        {
            scanf("%d", &bird);
            if( n_bird < bird )
                n_bird = bird;
            G[10000 + i].push_back(bird);
            G[bird].push_back(10000 + i);
        }
    }
    for( int i = 1; i <= n_bird; ++i )
        if( !visited[i] )
            dfs( i, ++n_tree );
    printf("%d %d\n", n_tree, n_bird);
    scanf("%d", &Q);
    for( int i = 0, b1, b2; i < Q; ++i )
    {
        scanf("%d %d", &b1, &b2);
        printf("%s\n", tree[b1] == tree[b2] ? "Yes":"No");
    }
}

三、思路二：路径压缩并查集

1、路径压缩
目的： 未使用路径压缩方法时，当并查集的树形结构深度较大时，find操作最坏可达O(n)的时间复杂度。
思路： 路径压缩主要思想为，将垂直多层的父子关系树压缩至两层，则结构稳定后，find操作复杂度降至O(1)。
实现： 每次find操作，当找到根后，使根结点成为 被查找结点到根节点路径上所有结点的直接父节点。

vector<int> ufs(20001, 0);
int root(int x)
{
    int root = x, temp;
    while(root != ufs[root])
        root = ufs[root];
    while(x != ufs[x])
    {
        temp = ufs[x];
        ufs[x] = root;
        x = temp;
    }
    return root;
}

2、构造并查集表示树形结构，可高效的找到鸟所在的树及树的个数。
指定照片下标范围为10001~10000+N

代码实现：

#include <cstdio>
#include <vector>
using namespace std;
int getroot( int x, vector<int> &ufs )
{
	int root = x, temp;
	while( root != ufs[root] )
		root = ufs[root];
	while( x != ufs[x] )
	{
		temp = ufs[x];
		ufs[x] = root;
		x = temp;
	}
	return root;
}
int main()
{
	int N, Q, n_tree = 0, n_bird = 0;
	scanf("%d", &N);
	vector<int> ufs(10001 + N, 0);
	for( int i = 1, K; i <= N; ++i )
	{
		ufs[10000 + i] = 10000 + i;
		scanf("%d", &K);
		for( int j = 0, bird; j < K; ++j )
		{
			scanf("%d", &bird);
			if( !ufs[bird] )
			{
				++n_bird;
				ufs[bird] = 10000 + i;
			}
			ufs[getroot(bird, ufs)] = 10000 + i;
		}
	}
	for( int i = 1; i <= N; ++i )
		if( ufs[10000 + i] == 10000 + i )
			++n_tree;
	printf("%d %d\n", n_tree, n_bird);
	scanf("%d", &Q);
	for( int i = 0, b1, b2; i < Q; ++i )
	{
		scanf("%d %d", &b1, &b2);
		printf("%s\n", getroot(b1, ufs) == getroot(b2, ufs) ? "Yes":"No");
	}
}