889. 根據(jù)前序和后序遍歷構(gòu)造二叉樹

返回與給定的前序和后序遍歷匹配的任何二叉樹。
pre 和 post 遍歷中的值是不同的正整數(shù)。
示例:
輸入:pre = [1,2,4,5,3,6,7], post = [4,5,2,6,7,3,1]
輸出:[1,2,3,4,5,6,7]
提示:1 <= pre.length == post.length <= 30
pre[] 和 post[] 都是 1, 2, ..., pre.length 的排列
每個輸入保證至少有一個答案。如果有多個答案,可以返回其中一個。

思路

令左分支有 LL 個節(jié)點。我們知道左分支的頭節(jié)點為 pre[1],但它也出現(xiàn)在左分支的后序表示的最后。所以 pre[1] = post[L-1](因為結(jié)點的值具有唯一性),因此 L = post.indexOf(pre[1]) + 1。
現(xiàn)在在我們的遞歸步驟中,左分支由 pre[1 : L+1] 和 post[0 : L] 重新分支,而右分支將由 pre[L+1 : N] 和 post[L : N-1] 重新分支。

代碼

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode constructFromPrePost(int[] pre, int[] post) {
        if (pre.length != post.length) {
            return null;
        } 
        
        return buildMyTree(pre, 0, pre.length-1, post, 0, post.length-1);
    }
    
    public TreeNode buildMyTree(int[] pre,
                                int preStart,
                                int preEnd,
                                int[] post,
                                int postStart,
                                int postEnd) {
        if (preStart == preEnd) {
            return new TreeNode(pre[preStart]);
        }
        if (preStart > preEnd) {
            return null;
        }
        if (postStart > postEnd) {
            return null;
        }
        
        int root_value = pre[preStart];
        TreeNode root = new TreeNode(root_value);
        int root_left_value = pre[preStart+1];
        int position = findPosition(post, postStart, postEnd, root_left_value);
        if (position == -1) {
            return null;
        }
        
        root.left = buildMyTree(pre, preStart+1, preStart+1+position-postStart, post, postStart, position);
        root.right = buildMyTree(pre, preStart+position-postStart+2, preEnd, post, position+1, postEnd-1);
        
        return root;
    }
    
    public int findPosition(int[] post, int start, int end, int key) {
        for (int i = start; i <= end; i++) {
            if (post[i] == key) {
                return i;
            }
        }
        return -1;
    }
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

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

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