面試題 04.06. 後繼者

LeetCode： 面試題 04.06. 後繼者

這個令人傷心的網路 >> 讓我不想做題了

二叉搜素樹的指定節點的下一個節點 >> 中序後序

中序後繼

根據題目的意思 >> 應該是不會出現相同的節點的 >> 因為指定節點 >> 兩個相同的就不知道是哪個了

    List<Integer> list = new ArrayList<>();
    public TreeNode inorderSuccessor(TreeNode root, TreeNode p) {
        order(root);
        int index = list.indexOf(p.val);
        TreeNode ans = null;
        if((index + 1) < list.size())
            ans = new TreeNode(list.get(index + 1));
        return ans;
    }

    public void order(TreeNode node){
        if(node == null) return ;
        // 左 根 右
        order(node.left);
        list.add(node.val);
        order(node.right);
    }