94. Binary Tree Inorder Traversal

二叉樹的中序遍歷,可以是經(jīng)典的遞歸寫法。
能寫成遞歸就可以寫成迭代,但是迭代的話需要保存一下之前的結(jié)點(diǎn)。比如對(duì)root來說,這個(gè)結(jié)點(diǎn)在我訪問完左半部分之后才需要訪問,于是我們可以使用一個(gè)stack,保證其訪問順序?qū)τ诎瑀oot的左半部分來說是最后的。FILO。

遞歸

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> result =new ArrayList<>();
        inOrder(root,result);
        return result;
    }
    private void inOrder(TreeNode root,List<Integer> result)
    {
        if(root==null) return ;
        inOrder(root.left,result);
        result.add(root.val);
        inOrder(root.right,result);
    }
}

迭代:

 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> result =new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();
        if(root==null)return result;
        TreeNode cur  = root;
       while(cur!=null || !stack.empty())
       {
        while(cur!=null)
        {
            stack.push(cur);
            cur = cur.left;
        }
        cur = stack.pop();
        result.add(cur.val);
        cur=cur.right;
    }
        return result ;
    }
    
}

之前的寫法有點(diǎn)問題,主要是:
while(部分沒有寫對(duì),我總是取棧的peek為start,導(dǎo)致迭代回到這個(gè)最初的peek時(shí)又接著往下取了,陷入了死循環(huán)。

while(!stack.empty())
      {
     TreeNode cur = stack.peek();
       while(cur!=null)
       {
           cur = cur.left;
          stack.push(cur);
       }
       cur = stack.pop();
       result.add(cur.val);
      if(cur.right!=null)
      stack.push(cur.right);
   }
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容