写法一:递归
class Solution {
public:
void preorder(TreeNode* root, vector<int> &res) {
if (root == NULL) {
return;
}
res.push_back(root->val);
preorder(root->left, res);
preorder(root->right, res);
}
vector<int> preorderTraversal(TreeNode* root) {
vector<int> res;
preorder(root, res);
return res;
}
};
写法二:栈
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> res;
if (root == NULL) return res;
st.push(root);
while (!st.empty()) {
TreeNode* temp = st.top();
st.pop();
res.push_back(temp->val);
if (temp->right) st.push(temp->right);
if (temp->left) st.push(temp->left);
}
return res;
}
};
通过这两种方法可以发现,递归的底层其实是栈,递归的过程就是每次递归会把函数的变量,参数值,返回地址压到栈中,等递归返回的时候,再把栈顶弹出去。感觉就很奇妙!
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> res;
TreeNode* cur = root;
while (cur != nullptr || !st.empty()) {
if (cur != nullptr) {
st.push(cur);
cur = cur->left; //左
} else {
cur = st.top(); //中
res.push_back(cur->val);
st.pop();
cur = cur->right; //右
}
}
return res;
}
};
中序遍历有点特殊,需要一个指针,先指到最左端,当cur为null时,让cur为栈顶元素然后输出其值,相当于已经到了“中”这个地步,然后向右走一格。
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> result;
if (root == NULL) return result;
st.push(root);
while (!st.empty()) {
TreeNode* node = st.top();
st.pop();
result.push_back(node->val);
if (node->left) st.push(node->left); // 相对于前序遍历,这更改一下入栈顺序 (空节点不入栈)
if (node->right) st.push(node->right); // 空节点不入栈
}
reverse(result.begin(), result.end()); // 将结果反转之后就是左右中的顺序了
return result;
}
};
其实后序遍历就是改了一下先序遍历,因为先序遍历是中左右,我先把他变为中右左,然后再把他反转,变为左右中,则是后序遍历。