原题链接
是一道要你倒推的题。你照着题目倒推一遍就发现是中序历遍。所以题目就是告诉你栈是怎么变化的,让你还原树的原貌。你只需要按照中序历遍莱倒着建立树就可以了。
下面给出accept代码:
#include<stdio.h>
#include<iostream>
#include<string>
using namespace std;
const int Maxsize = 10010;
int result[Maxsize] = { 0 };
int resultIndex = 0;
class TreeClass {
public:
int data;
TreeClass *Left;
TreeClass *Right;
TreeClass() {
this->data = -1;
this->Left = NULL;
this->Right = NULL;
}
};
class Stack {
public:
TreeClass *ans[Maxsize];
int index;
Stack() {
this->index = -1;
}
void push(TreeClass *a) {
this->ans[++this->index] = a;
}
TreeClass* pop() {
TreeClass *temp = this->ans[this->index];
this->index--;
return temp;
}
}stack;
void builtTree(TreeClass *tree) {
string a;
TreeClass *p = tree;
int number;
getline(cin, a);
number = atoi(a.c_str());
int count = 1;
for (int i = 0; i < number * 2; i++) {
getline(cin, a);
char temp = a[a.length() - 1];
int ans = -1;
if (temp != 'p') {
int o = 0;
char count = a[a.length() - 2];
if (count == ' ') {
o = temp - '0';
}
else {
o = count - '0';
o *= 10;
o += temp - '0';
}
ans = o;
}
if (ans != -1) {
p->data = ans;
p->Left = new TreeClass();
p->Right = new TreeClass();
stack.push(p);
p = p->Left;
}
else {
p = stack.pop();
p = p->Right;
}
}
}
void find(TreeClass* tree) {
if (tree == NULL)
return;
find(tree->Left);
find(tree->Right);
if(tree->data!=-1)
result[resultIndex++] = tree->data;
}
int main() {
TreeClass *tree;
tree = new TreeClass();
builtTree(tree);
find(tree);
for (int i = 0; i < resultIndex; i++) {
printf("%d", result[i]);
if (i != resultIndex - 1)
printf(" ");
}
return 0;
}
其实我是想写TreeNode的结果手快打成TreeClass,加上vs联想蛮好用的就一路按着这么名字一点都没有影响哈哈哈哈
讲讲思路:
- 这道题很明显就是想让你模拟栈写非递归算法。所以就写一个基本栈来实现。
- 接下来就是录入,处理数据。
- 请从一开始就用getline接收数据再处理数据。因为涉及到空格。用cin之类的接受第一个数字会导致缓存区有一个==\n==,再用getline会字符串数组超限因为读成了一个空串。
- 由于数据有可能是两位数,还得判断一下。
其实有更好的办法处理数据,不用getline也能做,我这个有点麻烦 - 如果是push,储存在根节点,压栈,下个目标是压栈元素的left。
- pop,出栈,下个目标是这个出栈元素的right。
- 建完树,输出答案。