码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode 144. Binary Tree Preorder Traversal

时间:2016-03-01 22:31:17      阅读:183      评论:0      收藏:0      [点我收藏+]

标签:

前序遍历

递归:

 1 /**
 2  * Definition for a binary tree node.
 3  * struct TreeNode {
 4  *     int val;
 5  *     TreeNode *left;
 6  *     TreeNode *right;
 7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 8  * };
 9  */
10 class Solution {
11 public:
12     vector<int> res;
13     void help(TreeNode* root){
14         // if(!root) return;
15         res.push_back(root->val);
16         if(root->left) help(root->left);
17         if(root->right) help(root->right);
18     }
19     vector<int> preorderTraversal(TreeNode* root) {
20         if(!root) return res;
21         help(root);
22         return res;
23     }
24     
25     
26 };

 

迭代:

 1 /**
 2  * Definition for a binary tree node.
 3  * struct TreeNode {
 4  *     int val;
 5  *     TreeNode *left;
 6  *     TreeNode *right;
 7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 8  * };
 9  */
10 class Solution {
11 public:
12     vector<int> preorderTraversal(TreeNode* root){
13         vector<int> res;
14         if(!root) return res;
15         stack<TreeNode*> stk;
16         stk.push(root);
17         
18         while(!stk.empty()){
19             TreeNode* cur = stk.top();
20             stk.pop();
21             res.push_back(cur->val);
22             
23             if(cur->right) stk.push(cur->right);
24             if(cur->left) stk.push(cur->left);
25             
26         }
27         return res;
28     }
29     
30 };

stk存放待处理节点的倒序,即top为最先处理的节点。

每次处理完top后,将top的右、左儿子依次push进stk中,所以下一次先处理的是top的左儿子(再深一步想也即先处理top的左子树,在该子树中优先处理其中的左子树,以此类推),符合前序遍历的规则。

 

另有算法未看,留坑。

LeetCode 144. Binary Tree Preorder Traversal

标签:

原文地址:http://www.cnblogs.com/co0oder/p/5232570.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!