三叉树 北航机试题

一、题目描述

一个关于三叉树的题目，小于100的值代表树叶，大于100的值为分支点，建树的过程是水平方向建树。

输入格式：

先输入n，代表有n组数据，接下来n行，输入4个数，第一个数代表根节点，接下来分别代表三个子节点，-1代表子节点不存在，输入的顺序按照层次遍历的次序。接下来，要求寻找叶子节点的最短路径，最短路径是指不经过重复的边。输入方式，首先输入一个值m，代表m行，接下来m行输入m个叶子节点和对应的优先级，要求按优先级输出从上次到达的位置到该节点的最短路径，每条路径的最后一个节点要求输出目标叶子节点，最后要求回到根节点。

输入样例

10
100 101 108 107
101 1 102 2
108 103 104 105
107 17 109 18
102 3 4 5
103 7 8 9
104 10 106 11
105 15 16 -1
109 19 20 21
106 12 13 14
5
8 1
14 3
16 2
5 0
19 4

输出样例

100 101 102 5
102 101 100 108 103 8
103 108 105 16
105 108 104 106 14
106 104 108 100 107 109 19
109 107 100

思路

使用数组来进行建树，并且给每一个节点设置一个pre值，方便遍历，如果要求出两个叶子节点之间的最短路径，首先求出它们各自到达根节点的路径。然后判断两个路径中是否有共同的祖先节点，判断方法是设置一个hash数组，对于第二条路径上的所有节点标记为1，然后在第一条路径中找出hash值为1的第一个节点，这个节点就是它们的最小祖先节点。

#include<iostream>
#include<algorithm>
using namespace std;
struct node{
	int child[3],pre;
	node(){
		pre = -1;
	}
}nodes[10010];
struct leaf{
	int inde,priority;
	vector<int> path;
}leaves[10010];
bool cmp(leaf a,leaf b){
	return a.priority<b.priority;
}
int main(){
	int n,index,tmp;
	scanf("%d",&n);
	for(int i = 0;i<n;i++){
		scanf("%d",&index);
		for(int j = 0;j<3;j++){
			scanf("%d",&tmp);
			nodes[index].child[j] = tmp;
			nodes[tmp].pre = index;
		}
	}
	scanf("%d",&n);
	for(int i = 0;i<n;i++){
		scanf("%d %d",&index,&tmp);
		leaves[i].inde = index;
		leaves[i].priority = tmp;
	}
	sort(leaves,leaves+n,cmp);
	for(int i = 0;i<=n;i++){
		if(i == 0){
			int now = leaves[0].inde;
			while(now != -1){
				leaves[0].path.push_back(now);
				now = nodes[now].pre;
			}
			reverse(leaves[0].path.begin(),leaves[0].path.end());
			for(int j = 0;j<leaves[0].path.size();j++){
				printf("%d",leaves[0].path[j]);
				if(j<leaves[0].path.size()-1) printf(" ");
			}
			printf("\n");
		}else if(i < n){
			leaves[i-1].path.pop_back();
			reverse(leaves[i-1].path.begin(),leaves[i-1].path.end());
			int now = leaves[i].inde;
			while(now != -1){
				leaves[i].path.push_back(now);
				now = nodes[now].pre;
			}
			reverse(leaves[i].path.begin(),leaves[i].path.end());
			int hashSet[10010] = {0};
			for(int j = 0;j<leaves[i].path.size();j++){
				hashSet[leaves[i].path[j]] = 1;
			}
			int s,start = 0;
			for(int j = 0;j<leaves[i-1].path.size();j++){
				if(hashSet[leaves[i-1].path[j]] == 1){
					s = leaves[i-1].path[j];
					printf("%d ",s);
					break;
				}else{
					printf("%d ",leaves[i-1].path[j]);
				}
			}
			for(int j = 0;j<leaves[i].path.size();j++){
				if(start == 0){
					if(leaves[i].path[j] == s){
						start = 1;
					}
				}else{
					printf("%d",leaves[i].path[j]);
					if(j<leaves[i].path.size()-1) printf(" ");
				}
			}
			printf("\n");
		}else{
			leaves[i-1].path.pop_back();
			reverse(leaves[i-1].path.begin(),leaves[i-1].path.end());
			for(int j = 0;j<leaves[i-1].path.size();j++){
				printf("%d",leaves[i-1].path[j]);
				if(j<leaves[i-1].path.size()-1) printf(" ");
			}
			printf("\n");
		}
	}
    return 0;
}