找出直系亲属【浙江大学】★

题目链接

解题思路

使用二叉树存亲子关系，然后通过深搜查找
可以多做几遍

#include<bits/stdc++.h>
using namespace std;
vector<int> G[30];
int opt;
void dfs(int s, int e, int step){
	if(s == e) {
		opt=step;
		return;
	}
	for(int j=0; j<G[s].size(); j++){
		int v = G[s][j];
		dfs(v, e, step+1);
	}
}
int main(){
	int m,n; 
	while(cin>>n>>m){
		char a,b,c;
		for(int i=0;i<30;i++){
			G[i].clear();
		}
		for(int i=0;i<n;i++){
			cin>>a>>b>>c;
			G[a-'A'].push_back(b-'A');
			G[a-'A'].push_back(c-'A');
		}
		for(int i=0;i<m;i++){
			cin>>a>>b;
			opt = -1;
			dfs(a-'A', b-'A', 0); 
			int p1 = opt;
			opt = -1;
			dfs(b-'A', a-'A', 0); 
			int p2 = opt;
			if(p1==-1 && p2==-1) cout<<"-"<<endl;
			else{
				int t= max(p1, p2);
				for(int i=t-1;i>=0;i--){
					if(i==0){
						
						if(t==p1) cout<<"child";
						else cout<<"parent";
					}
					else if(i==1) cout<<"grand";
					else cout<<"great-";
				}
				cout<<endl;
			}
		}
	}
	
	
	return 0;
}