leecode题解 94. Binary Tree Inorder Traversal

2019年4月20日 170次阅读 来源: cunfate

Given a binary tree, return the inorder traversal of its nodes’ values.
Example:
Input: [1,null,2,3]

   1
    \
     2
    /
   3

Output: [1,3,2]
Follow up: Recursive solution is trivial, could you do it iteratively?

这个题目可以说是面试和数据结构的入门题目了，递归的代码不算声明变量的话只要三行就能搞定，可以说没什么意思，主要还是用栈来实现递归式的遍历是考察的重点。
这里的话主要思考一下流程：
我们做中序遍历，肯定要先下沉到最左下角去保存最左下的元素，然后保存其父亲，然后进入父亲的右孩子也就是右兄弟字树去重复这个过程，也就是完成一次迭代。
这里面的问题主要就是

  1. 下沉后如何上浮？
  2. 终止条件是什么呢？
    对于第一个问题，我们能想到的当然就是栈——stack。后入先出嘛。
    那么结束条件是什么呢？就是当当前节点走到最右下的时候，这个时候栈中元素已经被清空（也就是说没有回溯的路径），而且也无法下沉（右孩子为空指针）。也就是说，迭代已经可以结束了，那么就果断的返回吧。
    最后代码大概是这样的：
/**
 * 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 {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        stack<TreeNode*> stk;
        vector<int> results;
        TreeNode* now = root;
        for(;;) {
            while(now != nullptr) {
                stk.push(now);
                now = now->left;
            }
            if(!stk.empty()) {
                now = stk.top();
                stk.pop();
                results.push_back(now->val);
            }
            if((now == nullptr || now->right == nullptr) && stk.empty()) {
                return results;
            }
            now = now->right;
        }
    }
};

大家都是这么做的，所以最后统计也beats 100%，不过也没什么可骄傲的就是了……

    原文作者：cunfate
    原文地址: https://www.jianshu.com/p/976aeda6d5b3
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞