这个题 找父亲发现不唯一,但是找儿子确是唯一的,那么可以倒过来建树
描述
如果A,B是C的父母亲,则A,B是C的parent,C是A,B的child,如果A,B是C(外)祖父,祖母,则A,B是C的grandparent,C是A,B的grandchild,如果A,B是C的(外)曾祖父,曾祖母,则A,B是C的great-grandparent,C是A,B的great-grandchild,之后再多一辈,则在关系上加一个great-。
输入描述:
输入包含多组测试用例,每组用例首先包含2个整数n(0<=n<=26)和m(0<m<50), 分别表示有n个亲属关系和m个问题, 然后接下来是n行的形式如ABC的字符串,表示A的父母亲分别是B和C,如果A的父母亲信息不全,则用-代替,例如A-C,再然后是m行形式如FA的字符串,表示询问F和A的关系。
输出描述:
对于每组输入数据,如果所有顶点都是连通的,输出"YES",否则输出"NO"。
示例
输入: 4 3 1 2 2 3 3 2 3 2 1 2 2 3 0 0 输出: NO YES
参考代码:
代码语言:javascript复制路径查找的代码封装成函数更好,我这里重复了两遍
#include<iostream>
#include<map>
#include<vector>
using namespace std;
int son[30];
int height[30];//其实没用到
int n,m;
int s;
vector<int>v;
void Initial()
{
for(int i=0; i<=27; i )
{
son[i]=i;
height[i]=0;
}
}
int Find(int x)
{
s ;
// cout<<x<<" has a son is "<<son[x]<<endl;
v.push_back(x);
if(x!=son[x])
{
return Find(son[x]);
}
return son[x];
}
void Union(int x,int y)//这个题每个节点关系很明确,后面的是父母,倒过来建树父母的儿子就是前者,不需要进行常规的height判断
{
son[y]=x;
height[x] ;
return ;
}
bool is_parents(char p)
{
if(p>='A'&&p<='Z')return true;
return false;
}
void relation_find(int p1,int p2)
{
s=0;
int a =Find(p1);
int d1 =s;
// cout<<a<<" "<<s<<endl;
s=0;
int b=Find(p2);
int d2 =s;
// cout<<b<<" "<<s<<endl;
if(a !=b)//根源不同
{
cout<<"-"<<endl;
}
else
{
if(d1<d2) //d2更大,说明 p2走过的路径深,是p1的长辈
{
v.clear();
Find(p2);//打印p2路径
int f=0;
for(int i=0; i<v.size(); i )
{
if(v[i]==p1)f=1;//p1在p2走过的路径中,说明是直系
}
if(f)
{
int d = d2-d1;
if(d == 1)cout<<"child"<<endl;
else if(d ==2)cout<<"grandchild"<<endl;
else
{
string great="";
d =d-2;
while(d--)great ="great-";
great ="grandchild";
cout<<great<<endl;
}
}
else cout<<"-"<<endl;
}
else
{
v.clear();
Find(p1);
int f=0;
for(int i=0; i<v.size(); i )
{
if(v[i]==p2)f=1;
}
if(f)
{
int d = d1-d2;
if(d == 1)cout<<"parent"<<endl;
else if(d ==2)cout<<"grandparent"<<endl;
else
{
string great="";
d =d-2;
while(d--)great ="great-";
great ="grandparent";
cout<<great<<endl;
}
}
else cout<<"-"<<endl;
}
}
}
int main()
{
while(cin>>n>>m)
{
// cout<<n<<" "<<m<<endl;
Initial();
// cout<<n<<endl;
while(n--)
{
string str;
cin>>str;
int son = str[0]-'A';
int f1 = str[1]-'A';
int f2= str[2]-'A';
if(is_parents(str[1]))Union(son,f1);
if(is_parents(str[2]))Union(son,f2);
}
while(m--)
{
string str;
cin>>str;
int p1 = str[0]-'A';
int p2 = str[1]-'A';
relation_find(p1,p2);
}
}
}