PTA 7-1 列出所有祖先结点

问题描述：

对于给定的二叉树，本题要求你按从上到下顺序输出指定结点的所有祖先结点。

输入格式:

首先第一行给出一个正整数 N（≤10），为树中结点总数。树中的结点从 0 到 N−1 编号。

随后 N 行，每行给出一个对应结点左右孩子的编号。如果某个孩子不存在，则在对应位置给出 “-”。编号间以 1 个空格分隔。

最后一行给出一个结点的编号i（0≤i≤N-1）。

输出格式:

在一行中按规定顺序输出i的所有祖先结点的编号。编号间以 1 个空格分隔，行首尾不得有多余空格。

输入样例:

7
2 -
- 6
- -
0 5
- -
4 1
- -
4

输出样例:

3 5

试题分析：

       本题目必须要考虑的一个问题是，先找到的祖先节点，后输出，也就是说最先打印的是根节点。以本题为例。编号为4的结点的父节点是5,5的父节点是3，而3是根节点。最后输出的结果是3 5。所以说这里需要用到栈，即“先进后出”。我们可以自己定义栈，如果不想自己写栈，就可以用C++库中提供的stack栈类型。具体的使用方法，小编通过注释的形式加到了代码中。

C++实例代码：

#include<bits/stdc++.h>
using namespace std;

/*定义一个结构体用来存储每一个结点*/
typedef struct {
	int left;
	int right;
} node;

int main() {
	int N;
	cin>>N;
	node a[N];           //存储所有结点的数组
	stack<int> result;   //定义一个整数类型的栈，存储找到的祖先节点

	/*读入所有的结点*/
	for(int i=0; i<N; i++) {
		char c1,c2;
		getchar();     //把换行符吞掉 
		cin>>c1;
		getchar();     //把空格吞掉 
		cin>>c2;
		if((c1>='0')&&(c1<='9')) {
			int l;
			l=c1-'0';
			a[i].left=l;
		} else {
			a[i].left=100;
		}
		if((c2>='0')&&(c2<='9')) {
			int r;
			r=c2-'0';
			a[i].right=r;
		} else {
			a[i].right=100;
		}
	}

	/*读入待查找结点的编号*/
	int k;
	cin>>k;
	int flag=1;
	while(flag==1) {
		int p;
		for(p=0; p<N; p++) {
			if((a[p].left==k)||(a[p].right==k)) {
				result.push(p);      //把p(已经找到的一个祖先节点)入栈 
				k=p;                 //此时应该寻找当前找到的这个祖先节点的祖先节点 
				break;
			}
		}
		if(p==N) {           //如果这次循环没有找到一个祖先节点，说明此时已经找到了根节点 ，退出循环
			flag=0;
		}
	}

	/*输出所有祖先节点*/
	if(N<=1) {              //当没有结点，后者只有一个结点时，是没有祖先节点的 
		cout<<"";
	} else {
		cout<<result.top();     //输出栈顶元素 
		result.pop();           //弹出栈顶元素 
		while(result.empty()!=true) {
			cout<<" "<<result.top();
			result.pop();
		}
	}
	return 0;

}

本期分享到这里就结束啦，大家如果对小编的代码有疑问或者想要小编做其他的程序，都欢迎来私信小编呀！