-
Notifications
You must be signed in to change notification settings - Fork 0
/
0094.cpp
91 lines (86 loc) · 1.64 KB
/
0094.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
#include <iostream>
#include <vector>
#include <stack>
using namespace std;
struct TreeNode
{
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x) :val(x), left(NULL), right(NULL) {}
};
class Solution
{
public:
vector<int> inorderTraversal(TreeNode* root)
{
vector<int> ans;
traversal(root, ans);
return ans;
}
void traversal(TreeNode* root, vector<int>& ans)
{
if (root == NULL)
return;
traversal(root->left, ans);
ans.push_back(root->val);
traversal(root->right, ans);
}
// 中序遍历 非递归 注意处理顺序和访问顺序不一致
vector<int> inorderTraversal(TreeNode* root)
{
stack<TreeNode*> st;
vector<int> ans;
if (root == NULL)
return ans;
// st.push(root);
TreeNode* cur = root;
while (cur != NULL || !st.empty())
{
if (cur != NULL)
{
st.push(cur);
cur = cur->left; // 左
}
else
{
cur = st.top();
ans.push_back(cur->val); // 中
st.pop();
cur = cur->right; // 右
}
}
return ans;
}
// 中序遍历 统一迭代风格 另一种标记法
vector<int> inorderTraversal(TreeNode* root)
{
vector<int> ans;
stack<TreeNode*> st;
if (root != NULL)
st.push(root);
while (!st.empty())
{
TreeNode* node = st.top();
if (node != NULL)
{
st.pop(); // 将该节点弹出,避免重复操作
if (node->right)
st.push(node->right); // 加入右结点
st.push(node);
st.push(NULL); // 中结点访问过还没处理,加入空结点
if (node->left)
st.push(node->left); // 加入左结点
}
// 遇到空结点,处理结点(访问过但是还没处理)
else
{
st.pop();
node = st.top();
st.pop();
ans.push_back(node->val); // 加入到结果集
}
}
return ans;
}
};