共同祖先

发布时间 2023-09-11 16:24:32作者: COMEIN

题目

https://kamacoder.com/problem.php?id=1010

小明发现和小宇有共同祖先!现在小明想知道小宇是他的长辈,晚辈,还是兄弟。

输入包含多组测试数据。每组首先输入一个整数N(N<=10),接下来N行,每行输入两个整数a和b,表示a的父亲是b(1<=a,b<=20)。小明的编号为1,小宇的编号为2。
输入数据保证每个人只有一个父亲。

对于每组输入,如果小宇是小明的晚辈,则输出“You are my younger”,如果小宇是小明的长辈,则输出“You are my elder”,如果是同辈则输出“You are my brother”。

主要考察对拓扑的理解,不需要使用排序,而是要从共同祖先这个点入手,所以需要不停地向上取值,然后根据路径长度来判断。

题解

#include <iostream>
#include <vector>
using namespace std;
int main()
{
	int n;
	while (cin >> n)
	{
	    vector<int> vec(21, 0);
        int a, b;
	    while (n--)
	    {
	        cin >> a >> b;
	        vec[a] = b;
	    }
	    int lenMing = 0, lenYu = 0;
	    int ancestorMing = vec[1], ancestorYu = vec[2];
	    while (ancestorMing != 0)
	    {
	        ancestorMing = vec[ancestorMing];
	        lenMing++;
	    }
	    while (ancestorYu != 0)
	    {
	        ancestorYu = vec[ancestorYu];
	        lenYu++;
	    }
	    if (lenMing == lenYu) cout << "You are my brother" << endl;
	    else if (lenMing > lenYu) cout << "You are my elder" << endl;
	    else cout << "You are my younger" << endl;
	}

	return 0;
}