-
二叉樹的前序周遊
給定一個二叉樹,傳回它的 前序 周遊。
示例:
輸入: [1,null,2,3]
1
\
2
/
3
輸出: [1,2,3]
進階: 遞歸算法很簡單,你可以通過疊代算法完成嗎?
思路一:遞歸法
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
vector<int> nums;
public:
vector<int> preorderTraversal(TreeNode* root) {
if (root == NULL){
return nums;
}
nums.push_back(root->val);
preorderTraversal(root->left);
preorderTraversal(root->right);
return nums;
}
};
/*8ms,13.2MB*/
時間複雜度:O(n)
空間複雜度:最壞情況下需要空間O(n),平均情況為O(logn)。
思路二:疊代法
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> nums;
if (root == NULL){
return nums;
}
stack<TreeNode*> s;
s.push(root);
TreeNode* p = NULL;
while (!s.empty()){
p = s.top();
s.pop();
nums.push_back(p->val);
if (p->right != NULL){
s.push(p->right);
}
if (p->left != NULL){
s.push(p->left);
}
}
return nums;
}
};
/*8ms,10.8MB*/
時間複雜度:O(n)
空間複雜度:O(n)