最大二叉树-二叉树-栈

https://leetcode-cn.com/problems/maximum-binary-tree/

思路1:用递归的思想来解决问题

思路2: 利用栈来解决,找出左边第一个比较大值,和右边第一个比较大值,两两比较取大的一个

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;
}

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];
} else if (nums[lis[i]] < nums[ris[i]]) {
pis[i] = lis[i];
} else {
pis[i] = ris[i];
}
}
return pis;
}

最大二叉树-二叉树-栈

上一篇:各种请求


下一篇:pytest入门与基础用法