Leetcode 144 Binary Tree Preorder Traversal
#include "header.h"
#include <stack>
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
先序遍历 根左右,遍历到每一个节点,都先入栈,然后去找左孩子的左孩子的左孩子的左孩子……直到有一个节点没有左孩子,
这个节点可能还有右孩子也可能没有(如果有右孩子,那就把这个点弹出去,然后对它的右子树,做一次类似的过程,
弹出去的目的是,这个节点的右子树遍历完了之后,栈顶是这个节点的父节点(可能是空的,如果这个节点是根节点)。
也可能没有右孩子,那样就继续弹栈,直到有一个节点有右孩子,再继续类似的过程)。
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> tranversal;
stack<TreeNode*> tree;
TreeNode* node = root;
if(root == nullptr)
return tranversal;
while(!tree.empty()||node != nullptr) {
while (node != nullptr) {
tree.push(node);
tranversal.push_back(node->val);
node = node->left;
}
if(node == nullptr) {
node = tree.top();
tree.pop();
node = node->right;
}
}
return tranversal;
}
};