mirror of
https://github.com/youngyangyang04/leetcode-master.git
synced 2025-07-07 15:45:40 +08:00
二叉树的统一迭代法.md 为"boolean标记法"加上C++的实现代码
This commit is contained in:
@ -36,7 +36,7 @@
|
|||||||
|
|
||||||
### 迭代法中序遍历
|
### 迭代法中序遍历
|
||||||
|
|
||||||
中序遍历代码如下:(详细注释)
|
> 中序遍历(空指针标记法)代码如下:(详细注释)
|
||||||
|
|
||||||
```CPP
|
```CPP
|
||||||
class Solution {
|
class Solution {
|
||||||
@ -75,6 +75,45 @@ public:
|
|||||||
|
|
||||||
可以看出我们将访问的节点直接加入到栈中,但如果是处理的节点则后面放入一个空节点, 这样只有空节点弹出的时候,才将下一个节点放进结果集。
|
可以看出我们将访问的节点直接加入到栈中,但如果是处理的节点则后面放入一个空节点, 这样只有空节点弹出的时候,才将下一个节点放进结果集。
|
||||||
|
|
||||||
|
> 中序遍历(boolean 标记法):
|
||||||
|
```c++
|
||||||
|
class Solution {
|
||||||
|
public:
|
||||||
|
vector<int> inorderTraversal(TreeNode* root) {
|
||||||
|
vector<int> result;
|
||||||
|
stack<pair<TreeNode*, bool>> st;
|
||||||
|
if (root != nullptr)
|
||||||
|
st.push(make_pair(root, false)); // 多加一个参数,false 为默认值,含义见下文注释
|
||||||
|
|
||||||
|
while (!st.empty()) {
|
||||||
|
auto node = st.top().first;
|
||||||
|
auto visited = st.top().second; //多加一个 visited 参数,使“迭代统一写法”成为一件简单的事
|
||||||
|
st.pop();
|
||||||
|
|
||||||
|
if (visited) { // visited 为 True,表示该节点和两个儿子位次之前已经安排过了,现在可以收割节点了
|
||||||
|
result.push_back(node->val);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
// visited 当前为 false, 表示初次访问本节点,此次访问的目的是“把自己和两个儿子在栈中安排好位次”。
|
||||||
|
|
||||||
|
// 中序遍历是'左中右',右儿子最先入栈,最后出栈。
|
||||||
|
if (node->right)
|
||||||
|
st.push(make_pair(node->right, false));
|
||||||
|
|
||||||
|
// 把自己加回到栈中,位置居中。
|
||||||
|
// 同时,设置 visited 为 true,表示下次再访问本节点时,允许收割。
|
||||||
|
st.push(make_pair(node, true));
|
||||||
|
|
||||||
|
if (node->left)
|
||||||
|
st.push(make_pair(node->left, false)); // 左儿子最后入栈,最先出栈
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
```
|
||||||
|
|
||||||
此时我们再来看前序遍历代码。
|
此时我们再来看前序遍历代码。
|
||||||
|
|
||||||
### 迭代法前序遍历
|
### 迭代法前序遍历
|
||||||
@ -110,7 +149,7 @@ public:
|
|||||||
|
|
||||||
### 迭代法后序遍历
|
### 迭代法后序遍历
|
||||||
|
|
||||||
后续遍历代码如下: (**注意此时我们和中序遍历相比仅仅改变了两行代码的顺序**)
|
> 后续遍历代码如下: (**注意此时我们和中序遍历相比仅仅改变了两行代码的顺序**)
|
||||||
|
|
||||||
```CPP
|
```CPP
|
||||||
class Solution {
|
class Solution {
|
||||||
@ -141,6 +180,42 @@ public:
|
|||||||
};
|
};
|
||||||
```
|
```
|
||||||
|
|
||||||
|
> 迭代法后序遍历(boolean 标记法):
|
||||||
|
```c++
|
||||||
|
class Solution {
|
||||||
|
public:
|
||||||
|
vector<int> postorderTraversal(TreeNode* root) {
|
||||||
|
vector<int> result;
|
||||||
|
stack<pair<TreeNode*, bool>> st;
|
||||||
|
if (root != nullptr)
|
||||||
|
st.push(make_pair(root, false)); // 多加一个参数,false 为默认值,含义见下文
|
||||||
|
|
||||||
|
while (!st.empty()) {
|
||||||
|
auto node = st.top().first;
|
||||||
|
auto visited = st.top().second; //多加一个 visited 参数,使“迭代统一写法”成为一件简单的事
|
||||||
|
st.pop();
|
||||||
|
|
||||||
|
if (visited) { // visited 为 True,表示该节点和两个儿子位次之前已经安排过了,现在可以收割节点了
|
||||||
|
result.push_back(node->val);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
// visited 当前为 false, 表示初次访问本节点,此次访问的目的是“把自己和两个儿子在栈中安排好位次”。
|
||||||
|
// 后序遍历是'左右中',节点自己最先入栈,最后出栈。
|
||||||
|
// 同时,设置 visited 为 true,表示下次再访问本节点时,允许收割。
|
||||||
|
st.push(make_pair(node, true));
|
||||||
|
|
||||||
|
if (node->right)
|
||||||
|
st.push(make_pair(node->right, false)); // 右儿子位置居中
|
||||||
|
|
||||||
|
if (node->left)
|
||||||
|
st.push(make_pair(node->left, false)); // 左儿子最后入栈,最先出栈
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
```
|
||||||
## 总结
|
## 总结
|
||||||
|
|
||||||
此时我们写出了统一风格的迭代法,不用在纠结于前序写出来了,中序写不出来的情况了。
|
此时我们写出了统一风格的迭代法,不用在纠结于前序写出来了,中序写不出来的情况了。
|
||||||
|
Reference in New Issue
Block a user