654.最大二叉树及变式

654.最大二叉树

题目分析

• 很容易想到首先比较找出最大值作为根节点，递归找出根节点的左子树的根节点和右子树的根节点

Solution

    public TreeNode constructMaximumBinaryTree(int[] nums) {
    	if (nums == null) return null;
    	return findRoot(nums, 0, nums.length);
    }
    
    /**
     * 找出[l, r)范围的根节点
     */
    private TreeNode findRoot(int[] nums, int l, int r) {
    	if (l == r) return null;
    	
    	// 找出[l, r)范围内最大值的索引
    	int maxIdx = l;
    	for (int i = l + 1; i < r; i++) {
			if (nums[i] > nums[maxIdx]) maxIdx = i;
		}
    	
    	TreeNode root = new TreeNode(nums[maxIdx]);
    	root.left = findRoot(nums, l, maxIdx);
    	root.right = findRoot(nums, maxIdx + 1, r);
    	return root;
    }

---

变式题

假如题目要求，返回一个数组，数组里面存着每个节点的父节点的索引（如果没有父节点，就存 -1）

题目分析

Solution

    public int[] parentIndexes(int[] nums) {
    	if (nums == null || nums.length == 0) return null;
    	/*
    	 * 就是找第一个比它大的，左边第一个比它大的，那么他就作为其左子树，右边第一个比它大的，那么他就作为其右子树
    	 * 1.扫描一遍所有的元素
    	 * 2.保持栈从栈底到栈顶是单调递减的
    	 */
    	int[] lis = new int[nums.length];
    	int[] ris = new int[nums.length];
    	// 栈中存放的是索引
    	Stack<Integer> stack = new Stack<>();
    	// 初始化
    	for (int i = 0; i < nums.length; i++) {
			ris[i] = -1;
			lis[i] = -1;
		}
    	for (int i = 0; i < nums.length; i++) {
    		// 找到比要加入的元素大的，把它弹出，并把它作为右边第一个比它大的数
			while (!stack.isEmpty() && nums[i] > nums[stack.peek()]) {
				ris[stack.pop()] = i;
			}
			// 栈顶位置为左边第一个比他小的
			if (!stack.isEmpty()) {
				lis[i] = stack.peek();
			}
			stack.push(i);
		}
		// 构建父索引数组
    	int[] pis = new int[nums.length];
    	for (int i = 0; i < pis.length; i++) {
    		// 左右都没有比它大的
    		if (lis[i] == -1 && ris[i] == -1) {
    			// i位置的是根节点
    			pis[i] = -1;
    			continue;
    		}
    		
    		if (lis[i] == -1) {
				pis[i] = ris[i];
    		} else if (ris[i] == -1) {
				pis[i] = lis[i];
			// 左右都有比它大的，那么找较小的
            // 3> 2> 1> 6  
            // 6 作为根节点，3作为其左子树
            // 1 的父节点一定为2， 而不是6，因为最大二叉树满足越大的会作为更高的根节点
    		} else if (nums[lis[i]] < nums[ris[i]]) {
				pis[i] = lis[i];
			} else {
				pis[i] = ris[i];
			}
		}
    	return pis;
    }

Reference：小码哥MJ