PAT-ADVANCED1086——Tree Traversals Again

版权声明:我的GitHub:https://github.com/617076674。真诚求星! https://blog.csdn.net/qq_41231926/article/details/83619950

我的PAT-ADVANCED代码仓:https://github.com/617076674/PAT-ADVANCED

原题链接:https://pintia.cn/problem-sets/994805342720868352/problems/994805380754817024

题目描述:

题目翻译:

1086 树的再次遍历

可以使用堆栈以非递归方式实现二叉树的中序遍历。 例如,当遍历6节点二叉树(编号为1到6的键)时,堆栈操作为:push(1); push(2); push(3); pop(); pop(); push(4); pop(); pop(); push(5); push(6); pop(); pop()。 然后可以从该操作序列生成唯一的二叉树(如下图所示)。 你的任务是给出这棵树的后序遍历序列。

输入格式:

每个输入文件包含一个测试用例。在每个测试用例中,第一行给出一个正整数N(<= 30),代表树的节点个数(节点编号为1 ~ N)。接下来的2N行,每行以下述形式描述了一次栈操作:"Push X",这里X指进栈的节点编号;或"Pop",弹出栈顶元素。

输出格式:

对每个测试用例,在一行中打印后序遍历的结果。题目保证一定有且只有一个解。每个数字间隔一个空格,行末不得有多余空格。

输入样例:

6
Push 1
Push 2
Push 3
Pop
Pop
Push 4
Pop
Pop
Push 5
Push 6
Pop
Pop

输出样例:

3 4 2 6 5 1

知识点:二叉树前序遍历和中序遍历的非递归实现、二叉树的后序遍历

思路:用前序遍历和中序遍历重建二叉树,再后序遍历得到结果

前序遍历的本质是:在第一次访问该节点时就输出该节点信息。

中序遍历的本质是:在第二次访问该节点时就输出该节点信息。

对应本题,节点的入栈顺序就是其前序遍历结果节点的出栈顺序就是其中序遍历结果。而任给一棵二叉树前序遍历和中序遍历信息,我们一定能唯一地重构出这棵二叉树。

时间复杂度和空间复杂度均是O(N)。

C++代码:

#include<iostream>
#include<string>
#include<vector>
#include<stack>

using namespace std;

struct node {
	int lchild;
	int rchild;
};

int N;
node Node[31];
int index = 1;
vector<int> preOrder;
vector<int> inOrder;
vector<int> postOrder;

int create(int leftPre, int rightPre, int leftIn, int rightIn);
void postOrderTraversal(int root); 

int main(){
	cin >> N;
	string operate;
	int num;
	stack<int> Stack;
	for(int i = 0; i < 2 * N; i++){
		cin >> operate;
		if(operate.compare("Push") == 0){
			cin >> num;
			preOrder.push_back(num);
			Stack.push(num);
		}else if(operate.compare("Pop") == 0){
			inOrder.push_back(Stack.top());
			Stack.pop();
		}
	}
	int root = create(0, preOrder.size() - 1, 0, inOrder.size() - 1);
	postOrderTraversal(root);
	for(int i = 0; i < postOrder.size(); i++){
		cout << postOrder[i];
		if(i != postOrder.size() - 1){
			cout << " ";
		}
	}
	cout << endl;
	return 0;
}

int create(int leftPre, int rightPre, int leftIn, int rightIn){
	if(leftPre > rightPre){
		return -1;
	}
	int root = preOrder[leftPre];
	int k;
	for(k = leftIn; k <= rightIn; k++){
		if(inOrder[k] == preOrder[leftPre]){
			break;
		}
	}
	int numLeft = k - leftIn;
	Node[root].lchild = create(leftPre + 1, leftPre + numLeft, leftIn, k - 1);
	Node[root].rchild = create(leftPre + numLeft + 1, rightPre, k + 1, rightIn);
	return root;
}

void postOrderTraversal(int root){
	if(root == -1){
		return;
	}
	postOrderTraversal(Node[root].lchild);
	postOrderTraversal(Node[root].rchild);
	postOrder.push_back(root);
}

C++解题报告:

猜你喜欢

转载自blog.csdn.net/qq_41231926/article/details/83619950
今日推荐