二叉树的建立&查找最邻近的共同祖先（201703）

题目

输入若干行，每一行的第一个输入为家谱中的某个成员，该行接着输入的信息为每个孩子姓名，最后一行的输入为要求查找的两个家谱成员的关系。
要求：根据输入的家谱成员的信息，建立二叉树家谱关系图，并输出二位待查找成员在家谱中的关系，包括输出他们最近邻的共同祖先以及在家谱中相差的层次数。

输入样例：

Ye Shu Ba
Shu Ge Mei1
Ba self Mei2
Ge Son1 Son2
Son2 Mei1

输出样例：

Shu 1

解题代码

#include<stdio.h>
#include<string.h>
#include<algorithm>
const int maxn=100;

typedef struct node{//定义节点
	char name[20];
	int level;
	struct node *father;
}Node;

int test(char s[]){//判断有几个人
	int i=0;
	char *temp=strtok(s," ");//必须是双引号
	while(temp){
		i++;
		temp=strtok(NULL," ");
	}
	return i;
}

void AddNode(char s[],Node nodes[],int number){
	char *temp=strtok(s," ");
	while(temp){
		//只有是根节点才会执行，不是根节点的情况只需要存储每行后面的两个单词； 
		if(number==1){
			strcpy(nodes[number].name,temp);
			nodes[number].father=NULL;
			nodes[number].level=1;
			number++;
		}
		temp=strtok(NULL," ");
		if(temp==NULL)
			break;
		strcpy(nodes[number].name,temp);
		//father是一个指针,指向的是父节点的地址。
		nodes[number].father=&nodes[number/2];//找到父节点，根据父节点的深度，求当前节点的深度 
		nodes[number].level=nodes[number].father->level+1;
		number++;
	}
}

void FindRelationship(char s[],Node nodes[],int number){
	char *temp=strtok(s," ");
	char people1[20];//存放第一个人的名字
	int people1_level=0;
	char people2[20];//存放第二个人的名字
	int people2_level=0;
	strcpy(people1,temp);
	temp=strtok(NULL," ");
	strcpy(people2,temp);
	int k,find1,find2;//find1、find2用来找共同的父节点
	int find=0;
	for(k=1;k<number;k++){
		if(strcmp(people1,nodes[k].name)==0){
			people1_level=nodes[k].level;
			find1=k;
			find++;
		}
	}
	for(k=1;k<number;k++){
		if(strcmp(people2,nodes[k].name)==0){
			people2_level=nodes[k].level;
			find2=k;
			find++;
		}
	}
	if(find==2){
		int diff=people1_level-people2_level;
		int p=0;
		//变成同一级进行找共同的父亲
		if(diff>0){ //1的层级要大
			for(p=0;p<diff;p++)
				find1=find1/2;//父节点
		}else{ //2的层级要大
			for(p=0;p<abs(diff);p++)
				find2=find2/2;
		}
		//在同一级进行查找共同的父节点
		while(strcmp(nodes[find1].father->name,nodes[find2].father->name)!=0){
			find1=find1/2;
			find2=find2/2;
		}
		printf("%s %d\n",nodes[find1].father->name,abs(diff));
	}
}

int main(){

	char s[100];
	char copy_s[100];
	Node nodes[maxn];
	int number=1;
	freopen("in.txt","r",stdin);
	while(gets(s)!=NULL){
		strcpy(copy_s,s);
		//三个人进行二叉树的建立
		if(test(copy_s)==3){
			AddNode(s,nodes,number);
			if(number==1)	
				number+=3;
			else
				number+=2;
		}
		//两个人找关系
		else{
			//number记录的是已经有number-1个节点
			FindRelationship(s,nodes,number);
			break;
		}
	}
	return 0;
}