1. 程式人生 > >劍指offer刷題第四題

劍指offer刷題第四題

arr null true tps 復雜度 分享 system brush www

第四題 重建二叉樹

技術分享圖片

思路:通過前序序列確定根節點,在中序序列找到其位置,確定左右子樹,並把對應左右子樹的數組根據根節點位置拷貝到新數組中,遞歸調用得到左右子樹。

時間復雜度:O(n)。

代碼:

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        if(pre.length == 0)
			return null;
		int val = pre[0];
		TreeNode root = new TreeNode(val);
		int rootPositionInOrder = 0;
		for(int i = 0;i < in.length;i++)
			if(in[i] == val) {
				rootPositionInOrder = i;
				break;
			}
		int[] leftPre = new int[rootPositionInOrder];
		int[] leftIn = new int[rootPositionInOrder];
		int[] rightPre = new int[in.length - 1 - rootPositionInOrder];
		int[] rightIn = new int[in.length - 1 - rootPositionInOrder];
		System.arraycopy(pre, 1, leftPre, 0, leftPre.length);
		System.arraycopy(pre, leftPre.length + 1, rightPre, 0, rightPre.length);
		System.arraycopy(in, 0, leftIn, 0, leftIn.length);
		System.arraycopy(in, rootPositionInOrder + 1, rightIn, 0, rightIn.length);
		root.left = reConstructBinaryTree(leftPre, leftIn);
		root.right = reConstructBinaryTree(rightPre,rightIn);
		return root;
    }
}

  

劍指offer刷題第四題