实现对二叉树的前后序非递归遍历
题解:
递归的实现就是:递去,归来。每一次递归调用都会把函数的局部变量、参数值和返回地址等压入调用栈中,然后递归返回的时候,从栈顶弹出上一次递归的各项参数,所以这就是递归为什么可以返回上一层位置的原因。所以可以用栈的方式实现遍历。
前序思路:
即遍历顺序为中-》左-》右
用栈实现就是从放入根节点开始循环,放入栈中记录栈顶节点,弹出后记录栈顶节点的值,然后判断栈顶节点有没有左右节点(从第二次循环开始,此栈顶节点就是上一节点的左节点),有右节点先放右节点进去,再放左节点,因为下一次循环会弹出栈顶节点并记录,所以要先放右后放左,这也是实现迭代遍历的精髓所在,当每一次循环时都先判断此次栈顶节点有没有右节点,先右后左,所以每一次循环开始记录下的就是左节点的值。当左节点都记录完后再开始从下往上弹右节点。
代码如下:
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> result;
if(NULL == root) return result;
st.push(root);
while(!st.empty()){
TreeNode* node = st.top();
st.pop();
result.push_back(node->val);
if (node->right) st.push(node->right);
if (node->left) st.push(node->left);
}
return result;
}
};
后序思路:
先序遍历是中左右,后续遍历是左右中,那么我们只需要调整一下先序遍历的代码顺序,就变成中右左的遍历顺序,然后在反转result数组,输出的结果顺序就是左右中了,如下图:
reverse(result.begin(), result.end());