愚蠢的宠物图解(并查集父思想节点表示法)

题目描述

背景

大家都知道,sheep有两只可爱的宠物(一只叫神牛,一只叫神菜)。有一天,sheep带着两只宠物到狗狗家时,这两只可爱的宠物竟然迷路了……

描述

狗狗的家因为常常遭到猫猫的攻击,所以不得不把家里前院的路修得非常复杂。狗狗家前院有N个连通的分叉结点,且只有N-1条路连接这N个节点,节点的编号是1-N(1为根节点)。sheep的宠物非常笨,他们只会向前走,不会退后(只向双亲节点走),sheep想知道他们最早什么时候会相遇(即步数最少)。

N的范围《=1000000

输入格式

第1行:一个正整数N,表示节点个数。

第2~N行:两个非负整数A和B,表示A是B的双亲。(保证A,B<=n)

第N+1行:两个非负整数A和B,表示两只宠物所在节点的位置。(保证A,B<=n)

输出格式

输出他们最早相遇的节点号。

样例输入

10

1 2

1 3

1 4

2 5

2 6

3 7

4 8

4 9

4 10

3 6

样例输出

1

 

 

解题思路:

利用并查集中的合并的思想,父节点表示法表示每一个元素。然后任选一个宠物,从该宠物所在的节点开始,搜索它的父节点,并且对搜索的路径进行标记。然后选择另一个宠物,同样从所在结点开始搜索,搜索它的父节点,如果在搜索路径中碰到了被标记的某个节点,那么该节点就是第一个相遇节点。

 

AC代码:

#include<iostream>

#include<string.h>

using namespace std;

 

int set[1000001];

int N;

bool vis[1000001];

 

int Find(int k)

{

   if(vis[k])

              returnk;

       vis[k]=true;

       if(k==set[k])

              returnk;

       returnset[k]=Find(set[k]);

}

 

int main()

{

       cin>>N;         //输入节点数

       memset(vis,0,sizeof(vis));

       intA,B;

       for(inti=2;i<=N;++i)

       {

              cin>>A>>B;

              set[B]=A;

       }

       cin>>A>>B;

       Find(A);

       cout<<Find(B)<<endl;

       return0;

}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章