一、题目概述
依据给定信息,求解:(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");
}
}