作者: admin
232.用栈实现队列
目标
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push、pop、peek、empty):
实现 MyQueue 类:
- void push(int x) 将元素 x 推到队列的末尾
- int pop() 从队列的开头移除并返回元素
- int peek() 返回队列开头的元素
- boolean empty() 如果队列为空,返回 true ;否则,返回 false
说明:
- 你 只能 使用标准的栈操作 —— 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
- 你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
示例 1:
输入:
["MyQueue", "push", "push", "peek", "pop", "empty"]
[[], [1], [2], [], [], []]
输出:
[null, null, null, 1, 1, false]
解释:
MyQueue myQueue = new MyQueue();
myQueue.push(1); // queue is: [1]
myQueue.push(2); // queue is: [1, 2] (leftmost is front of the queue)
myQueue.peek(); // return 1
myQueue.pop(); // return 1, queue is [2]
myQueue.empty(); // return false
说明:
- 1 <= x <= 9
- 最多调用 100 次 push、pop、peek 和 empty
- 假设所有操作都是有效的 (例如,一个空的队列不会调用 pop 或者 peek 操作)
进阶:
你能否实现每个操作均摊时间复杂度为 O(1) 的队列?换句话说,执行 n 个操作的总时间复杂度为 O(n) ,即使其中一个操作可能花费较长时间。
思路
这道题是让我们仅使用push压入栈顶、peek获取栈顶元素、pop弹栈等API实现先进先出队列。
关键是要想明白一点,入栈与出栈操作的是不同的栈,出栈s2中的元素是从入栈s1中获取的,并且,只能在s2为空的时候,将s1中的元素一个个弹栈再入s2,这样才能保证顺序反转。
代码
import java.util.ArrayDeque;
import java.util.Deque;
/**
* @date 2024-03-04 0:42
*/
public class MyQueue {
private final Deque<Integer> s1;
private final Deque<Integer> s2;
public MyQueue() {
s1 = new ArrayDeque<>();
s2 = new ArrayDeque<>();
}
public void push(int x) {
s1.push(x);
}
public int pop() {
if (s2.isEmpty()) {
while (!s1.isEmpty()) {
s2.push(s1.pop());
}
}
return s2.pop();
}
public int peek() {
if (s2.isEmpty()) {
while (!s1.isEmpty()) {
s2.push(s1.pop());
}
}
if (s2.isEmpty()) {
throw new RuntimeException();
}
return s2.peek();
}
public boolean empty() {
return s2.isEmpty() && s1.isEmpty();
}
}
Stack注释中有这么一段话,告诉我们应该优先使用Deque接口及其实现。
A more complete and consistent set of LIFO stack operations is provided by the Deque interface and its implementations, which should be used in preference to this class.
For example: Deque<Integer> stack = new ArrayDeque<Integer>();
在Java中,Stack类是基于Vector实现的,而Vector是一个线程安全的、可动态增长的数组。虽然Stack也是数组实现的,但由于Vector的一些内部机制(例如,每次增长时都会分配更大的数组,并将旧数组的内容复制到新数组中),它可能在某些操作上不如ArrayDeque高效。
相比之下,ArrayDeque是基于循环数组实现的,它避免了不必要的内存分配和复制操作。循环数组意味着当数组的一端达到容量限制时,元素会从另一端开始填充,从而充分利用了数组空间。这种实现方式通常比Vector或Stack更高效。
因此,即使Stack也是数组实现的,但由于Deque(如ArrayDeque)使用了不同的内部机制和优化,它在某些情况下可能会提供更好的性能。
另外,值得注意的是,在Java中,Stack类已经被标记为遗留(legacy),不建议在新的代码中使用。相反,应该使用Deque接口及其实现,如ArrayDeque,因为它们提供了更完整、更一致的操作集合,并且通常具有更好的性能。
性能
时间复杂度:push(数组赋值),empty为O(1),pop与peek虽然涉及到移动数据,但只有在出栈为空的时候才执行,均摊后为O(1)。
空间复杂度为O(n)。
225.用队列实现栈
目标
请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(push、top、pop 和 empty)。
实现 MyStack 类:
void push(int x) 将元素 x 压入栈顶。
int pop() 移除并返回栈顶元素。
int top() 返回栈顶元素。
boolean empty() 如果栈是空的,返回 true ;否则,返回 false 。
注意:
你只能使用队列的基本操作 —— 也就是 push to back、peek/pop from front、size 和 is empty 这些操作。
你所使用的语言也许不支持队列。 你可以使用 list (列表)或者 deque(双端队列)来模拟一个队列 , 只要是标准的队列操作即可。
示例:
输入:
["MyStack", "push", "push", "top", "pop", "empty"]
[[], [1], [2], [], [], []]
输出:
[null, null, null, 2, 2, false]
解释:
MyStack myStack = new MyStack();
myStack.push(1);
myStack.push(2);
myStack.top(); // 返回 2
myStack.pop(); // 返回 2
myStack.empty(); // 返回 False
说明:
1 <= x <= 9
最多调用100 次 push、pop、top 和 empty
每次调用 pop 和 top 都保证栈不为空
进阶:你能否仅用一个队列来实现栈。
思路
这道题的目的是只使用标准库的部分API实现栈,即后进先出。
Java中可以使用Queue接口,offer插入队尾,peek获取队首元素,poll从队首获取并删除元素。
官网给出了两种方法,一种是使用两个队列,offer总是将新元素放到空队列q中,然后将另一队列q1从头至尾放入q,这样就实现了顺序的反转。
例如,依次入栈1,2,3,4:
说明:队首在右边
offer(1):
q: 1
q1:
----------------------------
offer(2):
q: 1
q1: 2
q1.offer(q.poll()):
q:
q1: 1 2
----------------------------
offer(3):
q: 3
q1: 1 2
while(!q1.isEmpty){q.offer(q1.poll())}:
q: 1 2 3
q1:
----------------------------
offer(4):
q: 1 2 3
q1: 4
while(!q.isEmpty){q1.offer(q.poll())}:
q:
q1: 1 2 3 4
另一种只用一个队列的方法是:每次offer前先记录队列数据数量n,然后再offer,之后将前面n个数依次poll并offer到队尾。
例如,依次入栈1,2,3,4:
说明:队首在右边
offer(1):
q: 1
----------------------------
n = 1
offer(2):
q: 2 1
p.offer(p.poll()):
q: 1 2
----------------------------
n = 2
offer(3):
q: 3 1 2
循环2次:p.offer(p.poll())
q: 1 2 3
----------------------------
n = 3
offer(4):
q: 4 1 2 3
循环3次:p.offer(p.poll())
q:1 2 3 4
性能
这两种方法的性能基本相同。
时间复杂度:入栈操作均为O(n),其余为O(1)。
空间复杂度:O(n)
121.买卖股票的最佳时机
目标
你只能选择 某一天 买入这只股票,并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。
返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润,返回 0。
示例 1:
输入:[7,1,5,3,6,4]
输出:5
解释:在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。
注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格;同时,你不能在买入前卖出股票。
示例 2:
输入:prices = [7,6,4,3,1]
输出:0
解释:在这种情况下, 没有交易完成, 所以最大利润为 0。
说明:
- 1 <= prices.length <= 105
- 0 <= prices[i] <= 104
思路
虽然这是一道简单题,但是我也想了一个多小时。算法实现起来简单,但是算法的设计,如何得到满足条件的结果也不是那么直观。
说回这道题,要求得最大利润,我们都明白要低买高卖。难道这道题就是让求数组的最大值与最小值?显然不是。我们的目标是求取一个买入点,再其之后的某天卖出获得的利润最大。只能操作一次,可以看成是长线交易(后面还有一道可以多次操作的题)。
暴力解法是遍历股票价格数组,然后向后循环,求得最大利润。时间复杂度是O(n!),不可行。
思考下面的问题:
取得最大利润是否必须在最低点买入?不是,比如[2,8,1,6]
。
取得最大利润是否必须在最高点卖出?不是,比如[3,8,1,7]
。
但是,如果遇到了更低的价格,那么它一定是当前最佳的买入点。因为不管后续价格怎么波动,想要获得最大收益,被减数当然越小越好。
因此,问题可以转化为:任取一个买入点,向后遍历并记录最大利润,直到出现一个更低的买入点,在这之前的一段时间内,我们所选的就是最佳买入点,可以记录这段时间内的最大利润。然后以新的最佳买入点继续向后遍历并记录最大利润,直到一个新的最佳买入点出现。在这一过程中如果最大利润比之前的值更大就替换掉。这样我们就得到了问题的答案。
其实,一开始我并没有这么清晰的思路。只是跟着感觉走:
- 首先,如果遍历时发现股票价格是递减的,那么我们肯定不能买入。这里,我们需要计算后一天减去前一天的收益,如果连着都是负数,那么直接跳过。
- 然后,如果找到了第一个正收益的点,那么我们买入。之后,我们开始累加收益,这样累加得到的是当天到买入点的收益。即 如果a是我们的买入点,那么往后利润的累加和
profitSum
为(b - a) + (c - b) + (d - c) + (e - d) = e - a
。在这一过程中,我们记录最大收益profitMax
。 - 接下来,我们需要找到更低的买入点。如果我们的买入点是i,当我们遍历到第j天的时候,
profitSum = prices[j] - prices[i]
,而当天与前一天的利润为profit = prices[j] - prices[j-1]
,如果profit > profitSum
,那么prices[j] - prices[j-1] > prices[j] - prices[i]
即prices[j-1] < prices[i]
。j-1
就是新的最佳买入点。这时我们需要将利润和重置,然后再比较最大利润。
最开始写的时候忽略了查找最新买入点的这个步骤,增加的这个判断 profit > profitSum
是根据错误案例调试出来的,感觉就是如果一天的收益就比之前的历史收益大了,那就没必要再累加前面的利润了,应该重新开始。
代码
/**
* @date 2024-03-03 1:44
*/
public class MaxProfit {
public int maxProfit(int[] prices) {
int profitSum = 0;
int profitMax = 0;
for (int i = 1; i < prices.length; i++) {
int profit = prices[i] - prices[i - 1];
if (profitSum <= 0 && profit <= 0) {
continue;
}
profitSum += profit;
if (profit > profitSum) {
// 最开始的时候没写这个判断
profitSum = profit;
}
if (profitSum > profitMax) {
profitMax = profitSum;
}
}
return profitMax;
}
public static void main(String[] args) {
MaxProfit main = new MaxProfit();
System.out.println(main.maxProfit(new int[]{3, 3, 5, 0, 0, 3, 1, 4}));
}
}
理清楚之后发现许多操作是没必要的,比如累加每天的利润,实际就是当天价格减去买入点价格。
public int maxProfit(int[] prices) {
int buyPoint = 0;
int profitMax = 0;
for (int i = 1; i < prices.length; i++) {
int profit = prices[i] - prices[buyPoint];
if (profit <= 0) {
buyPoint = i;
} else if (profit > profitMax) {
profitMax = profit;
}
}
return profitMax;
}
性能
优化后
169.多数元素
目标
给定一个大小为 n 的数组 nums ,返回其中的多数元素。多数元素是指在数组中出现次数 大于 ⌊ n/2 ⌋ 的元素。
你可以假设数组是非空的,并且给定的数组总是存在多数元素。
示例 1:
输入:nums = [3,2,3]
输出:3
示例 2:
输入:nums = [2,2,1,1,1,2,2]
输出:2
说明:
- n == nums.length
- 1 <= n <= 5 * 104
- -109 <= nums[i] <= 109
进阶:尝试设计时间复杂度为 O(n)、空间复杂度为 O(1) 的算法解决此问题。
思路
今天做个简单题吧。要求时间复杂度是O(n),那么就不能嵌套循环,空间复杂度度是O(1),也就不能开辟新数组。很自然的可以想到:累加当前出现次数 count 最多的元素 res,如果遇到其它元素则减1,当 count 为负数时,说明当前出现次数最多的元素 可能
发生改变,将res替换为当前元素,并将count置1。
这里说 可能
,是因为只要遇到与所选元素值不相等的, count 就减1。我们并不清楚这些其它元素值是否都相同,只能够推出当初所选的多数被其它少数反超了。但是从总体来考虑,如果我们所选择的真是多数元素,那么它一定会在后面再次反超。
官网介绍了一种投票算法 Boyer-Moore,应该也是这种思路吧。
官网还给出了一种分治算法,主要思想是:如果将数组分成两部分,那么数组的众数至少是一部分的众数。递归求解,然后回溯合并,确定子数组众数。不过时间复杂度O(nlogn),参考算法导论P53主定理。空间复杂度:O(logn),递归用到的栈空间。
代码
/**
* @date 2024-03-02 22:24
*/
public class MajorityElement {
public int majorityElement(int[] nums) {
int res = nums[0];
int count = 1;
for (int i = 1; i < nums.length; i++) {
if (nums[i] != res) {
count--;
// 出错点:反超时应当将值设为1,参考错误用例[10,9,9,9,10]
if (count < 0) {
res = nums[i];
count = 1;
}
} else {
count++;
}
// 本以为加上可以提高性能,谁知道还慢了2ms
// if (count > Math.floor(nums.length / 2.0)) {
// break;
// }
}
return res;
}
public static void main(String[] args) {
MajorityElement main = new MajorityElement();
System.out.println(main.majorityElement(new int[]{10, 9, 9, 9, 10}));
}
}
性能
本以为判断条件可以提高效率,谁知道还慢了2ms,耗时增加了2倍,因为每次算出答案基本上也都循环完了。
2673.使二叉树所有路径值相等的最小代价
目标
给你一个整数 n 表示一棵 满二叉树 里面节点的数目,节点编号从 1 到 n 。根节点编号为 1 ,树中每个非叶子节点 i 都有两个孩子,分别是左孩子 2 i 和右孩子 2 i + 1 。
树中每个节点都有一个值,用下标从 0 开始、长度为 n 的整数数组 cost 表示,其中 cost[i] 是第 i + 1 个节点的值。每次操作,你可以将树中 任意 节点的值 增加 1 。你可以执行操作 任意 次。
你的目标是让根到每一个 叶子结点 的路径值相等。请你返回 最少 需要执行增加操作多少次。
注意:
- 满二叉树 指的是一棵树,它满足树中除了叶子节点外每个节点都恰好有 2 个子节点,且所有叶子节点距离根节点距离相同。
- 路径值 指的是路径上所有节点的值之和。
说明:
- 3 <= n <= 105
- n + 1 是 2 的幂
- cost.length == n
- 1 <= cost[i] <= 104
思路
操作首先要知道操作的最终状态:各路径相等。那么哪一个路径值可以使操作总数最小?既然明确不了哪一个路径又如何操作,又怎会知道最小?最开始以为可能是路径的中位数,但是我也没法证明。后来发现想复杂了,操作只能加不能减,那么问题就变成了将所有路径变成最大最少需要几次操作。
首先可以根据节点的父子关系将子路径值层层累加,那么最底下的叶子层就是各条路径的值。用最大路径值分别减去其它路径值得到相应的操作数,但这时操作数不是最小的。注意到,如果左右孩子所需操作均不为零,可以将较小孩子节点的操作数提到父节点上,并将操作数置零,而另一个孩子节点同时减去该操作数,这样就完成了一步最小化。依次向上计算,直到根节点,然后统计各节点操作数即可。
代码
/**
* @date 2024-02-28 8:52
*/
public class MinIncrements {
public int minIncrements(int n, int[] cost) {
for (int i = 1; i < cost.length; i++) {
if (i % 2 == 1) {
cost[i] += cost[(i - 1) / 2];
} else {
cost[i] += cost[(i - 2) / 2];
}
}
int max = 0;
double l = Math.floor(Math.log(Double.parseDouble(String.valueOf(n))) / Math.log(2.0));
for (int i = (int) Math.pow(2.0, l) - 1; i < cost.length; i++) {
if (cost[i] > max) {
max = cost[i];
}
}
for (int i = (int) Math.pow(2.0, l) - 1; i < cost.length; i++) {
cost[i] = max - cost[i];
}
for (int i = cost.length - 1; i > 2; i -= 2) {
if (cost[i] == 0 || cost[i - 1] == 0) {
cost[(i - 2) / 2] = 0;
continue;
}
if (cost[i] <= cost[i - 1]) {
cost[(i - 2) / 2] = cost[i];
cost[i - 1] = cost[i - 1] - cost[i];
cost[i] = 0;
} else if (cost[i] >= cost[i - 1]) {
cost[(i - 2) / 2] = cost[i - 1];
cost[i] = cost[i] - cost[i - 1];
cost[i - 1] = 0;
}
}
int res = 0;
for (int i = 1; i < cost.length; i++) {
res += cost[i];
}
return res;
}
public static void main(String[] args) {
MinIncrements main = new MinIncrements();
// int[] cost = new int[]{1, 5, 2, 2, 3, 3, 1};
int[] cost = new int[]{764, 1460, 2664, 764, 2725, 4556, 5305, 8829, 5064, 5929, 7660, 6321, 4830, 7055, 3761};
System.out.println(main.minIncrements(cost.length, cost));
}
}
性能
又是勉强过关。看看官网给的答案:
class Solution {
public int minIncrements(int n, int[] cost) {
int ans = 0;
for (int i = n - 2; i > 0; i -= 2) {
ans += Math.abs(cost[i] - cost[i + 1]);
// 叶节点 i 和 i+1 的双亲节点下标为 i/2(整数除法)
cost[i / 2] += Math.max(cost[i], cost[i + 1]);
}
return ans;
}
}
称之为自底向上的贪心算法。所谓贪心算法,它在每一步都做出当时看起来最佳的选择,也就是说,它总是做出局部最优的选择。贪心算法并不保证得到最优解,但对很多问题确实可以求得最优解。
2581.统计可能的树根数目
有空补上
2867.统计树中的合法路径数目
目标
给你一棵 n 个节点的无向树,节点编号为 1 到 n 。给你一个整数 n 和一个长度为 n - 1 的二维整数数组 edges ,其中 edges[i] = [ui, vi] 表示节点 ui 和 vi 在树中有一条边。
请你返回树中的 合法路径数目 。
如果在节点 a 到节点 b 之间 恰好有一个 节点的编号是质数,那么我们称路径 (a, b) 是 合法的 。
注意:
- 路径 (a, b) 指的是一条从节点 a 开始到节点 b 结束的一个节点序列,序列中的节点 互不相同 ,且相邻节点之间在树上有一条边。
- 路径 (a, b) 和路径 (b, a) 视为 同一条 路径,且只计入答案 一次 。
思路
质数是指在大于1的自然数(非负整数)中,除了1和它本身以外不再有其他因数的自然数。
现在有一颗 n 个节点的无向树,要求任意两个连通节点间恰好有一个质数节点的路径数。树是一种无环连通图,问题可以转化为从树中选取两个节点,节点之间的路径只经过一个质数节点。由于没有环,所以两点之间的路径是唯一的。
- 两个都是质数节点要排除掉。
- 以质数节点为中心,与它邻接的非质数节点符合条件。即以质数节点为中心加上与之相连的非质数节点任取两个均可。我们可以称直接与质数节点相连的非质数节点为直接节点。
- 直接节点连通的非质数节点也可能满足条件,需要要减去直接节点向外连通的路径,即直接节点加上其向外连通的节点之间任取两个的路径数。
按照上面的思路,先要找到所有的质数节点,涉及到质数判断。同时保存与之直接相连的非质数节点。然后保存非质数节点的边,使用Map保存,边的两个端点都保存进去,方便后续向外查找连通的节点。
得到满足条件的节点总数,根据排列组合公式C(n,2) = n!/(2!(n-2)!) = (n-1)n/2
求得路径总数D。
将外围节点k向外连通节点总数记为Ik,无效路径数为(Ik-1)Ik/2
。
最终的结果就是D - Σ(Ik-1)Ik/2
代码
/**
* @date 2024-02-27 0:22
*/
public class CountPaths {
public Map<Integer, Set<Integer>> primeEdges = new HashMap<>();
public Map<Integer, Set<Integer>> notPrimeEdges = new HashMap<>();
public Map<Integer, Integer> indirectNodesNumMap = new HashMap<>();
Set<Integer> counter = new HashSet<>();
public long countPaths(int n, int[][] edges) {
for (int i = 0; i < n - 1; i++) {
int[] edge = edges[i];
boolean i0 = isPrimeNumber(edge[0]);
boolean i1 = isPrimeNumber(edge[1]);
if (i0 && !i1) {
primeEdges.computeIfAbsent(edge[0], k -> new HashSet<>());
primeEdges.get(edge[0]).add(edge[1]);
} else if (!i0 && i1) {
primeEdges.computeIfAbsent(edge[1], k -> new HashSet<>());
primeEdges.get(edge[1]).add(edge[0]);
} else if(!i0){
notPrimeEdges.computeIfAbsent(edge[0], k -> new HashSet<>());
notPrimeEdges.computeIfAbsent(edge[1], k -> new HashSet<>());
notPrimeEdges.get(edge[0]).add(edge[1]);
notPrimeEdges.get(edge[1]).add(edge[0]);
}
}
long res = 0;
for (Integer primeNode : primeEdges.keySet()) {
Set<Integer> nonPrimeNodesOfPrimeEdge = primeEdges.get(primeNode);
counter.clear();
int total = 0;
for (int nonPrimeNode : nonPrimeNodesOfPrimeEdge) {
counter.add(nonPrimeNode);
if (indirectNodesNumMap.get(nonPrimeNode) == null) {
indirectNodesNumMap.put(nonPrimeNode, 1);
countEdges(nonPrimeNode, nonPrimeNode);
}
total += indirectNodesNumMap.get(nonPrimeNode);
}
total = total + 1;
res += total * (total - 1L) / 2L;
for (int nonPrimeNode : nonPrimeNodesOfPrimeEdge) {
int indirectNodesNum = indirectNodesNumMap.get(nonPrimeNode);
res -= indirectNodesNum * (indirectNodesNum - 1L) / 2L;
}
}
return res;
}
public Set<Integer> countEdges(int key, int nonPrimeNode) {
if (notPrimeEdges.get(nonPrimeNode) != null) {
for (Integer node : notPrimeEdges.get(nonPrimeNode)) {
if (!counter.contains(node)) {
indirectNodesNumMap.put(key, indirectNodesNumMap.get(key) + 1);
counter.add(node);
countEdges(key, node);
}
}
}
return counter;
}
public boolean isPrimeNumber(int num) {
if (num == 1) {
return false;
}
if (num == 2) {
return true;
}
if (num % 2 == 0) {
return false;
}
for (int i = 3; i * i <= num; i+=2) {
if (num % i == 0) {
return false;
}
}
return true;
}
public static void main(String[] args) {
// int[][] edges = new int[][]{new int[]{1, 2}, new int[]{1, 3}, new int[]{2, 4}, new int[]{2, 5}};
int[][] edges = new int[][]{new int[]{1, 2}, new int[]{4, 1}, new int[]{3, 4}};
CountPaths main = new CountPaths();
// System.out.println(main.countPaths(5, edges));
System.out.println(main.countPaths(4, edges));
}
}
性能
勉强通过。有时间再回来看看题解吧。
938.二叉搜索树的范围和
目标
给定二叉搜索树的根结点 root,返回值位于范围 [low, high] 之间的所有结点的值的和。
说明:
- 树中节点数目在范围 [1, 2 * 10^4] 内
- 1 <= Node.val <= 10^5
- 1 <= low <= high <= 10^5
- 所有 Node.val 互不相同
思路
二叉搜索树,也叫二叉查找树(Binary Search Tree, BST)。BST是一颗二叉树,其中的每个节点都含有一个可比较的Key,并且每个节点的Key都大于其左子树中的任意节点的Key,而小于其右子树的任意节点的Key。
比较每个节点是否在给定的范围内,如果节点Key小于low去左子树找,大于high则去右子树找,如果在二者之间,累加和,继续遍历左右子树。
代码
/**
* @date 2024/2/26 10:37
*/
public class RangeSumBST {
public class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode() {
}
TreeNode(int val) {
this.val = val;
}
TreeNode(int val, TreeNode left, TreeNode right) {
this.val = val;
this.left = left;
this.right = right;
}
@Override
public String toString() {
return "TreeNode{" +
"val=" + val +
", left=" + left +
", right=" + right +
'}';
}
}
public int sum = 0;
/** 省去了节点为空的判断嵌套*/
public int rangeSumBST_v1(TreeNode root, int low, int high) {
if (root == null) {
return 0;
}
if (low > root.val) {
rangeSumBST_v1(root.right, low, high);
}
if (high < root.val) {
rangeSumBST_v1(root.left, low, high);
}
if (high >= root.val && low <= root.val) {
sum += root.val;
rangeSumBST_v1(root.left, low, high);
rangeSumBST_v1(root.right, low, high);
}
return sum;
}
public int rangeSumBST(TreeNode root, int low, int high) {
if (low > root.val) {
if (root.right != null) {
rangeSumBST(root.right, low, high);
}
}
if (high < root.val) {
if (root.left != null) {
rangeSumBST(root.left, low, high);
}
}
if (high >= root.val && low <= root.val){
sum += root.val;
if (root.left != null) {
rangeSumBST(root.left, low, high);
}
if (root.right != null) {
rangeSumBST(root.right, low, high);
}
}
return sum;
}
}
性能
235.二叉搜索树的最近公共祖先
目标
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。
思路
我第一次想到的方法是先找到这两个指定节点,然后交替求取位置索引较大节点的父节点。然后比较,如果值相等则为公共祖先。但是,题目给出的树结构中不包含指向父节点的引用,很自然地想到先将值保存到数组中,中序遍历二叉搜索树得到正序序列。然后二分查找出index,计算父节点index:左孩是奇数index,父index为(child-1)/2,右孩是偶数index,父index为(child-2)/2,注意排除root,index为0。接着比较父节点的值,还应该考虑输入节点本身就是父子关系的情况,如果pindex > qindex,应先求p节点的父index,因为index大,层数可能更高。再与qindex对应的值比较,不相等的话再求q节点的父index,依此类推。题目中有提到,树中的值都是唯一的。这个想法看起来正确,但真正去实现的时候就会发现是不可行的。首先是存储空间问题,要在数组中保留树的结构,在有许多空节点的情况下是不可行的。这个在 二叉搜索树最近节点查询 中提到过。然后是二分查找需要先排序的问题,这本身就打乱了树节点的index关系。
这几天刷题有一个感受,就是如果算法实现起来太过复杂那么一定是有问题的,很有可能是求解的方向不对。
于是我换了一种思路,还是先找到这两个节点,但是在找的过程中记录下访问的路径,然后再找到路径节点中所有相同节点中的最后一个即可。那么路径应该保存到哪里,又如何获取相同节点序列的最后一个呢?
在 Java 的标准库中,java.util.Stack 类是使用数组实现的。然而,从 Java 6 开始,java.util.Stack 类被标记为遗留(legacy),并建议使用 java.util.Deque 接口及其实现(如 ArrayDeque 或 LinkedList)来替代。Deque 接口表示双端队列,它支持在两端插入和删除元素,因此非常适合实现栈和队列。----AI的回答
无论是使用数组还是链表,最后查找的时候,都可以从头开始同时比较,直到第一个不相等节点出现即可。无需额外申请空间。
如果从尾开始比较的话,需要借助HashSet,将其中一个路径序列存入集合,然后循环弹另一个路径栈,第一个在集合中的被弹出元素就是最近的公共祖先。需要额外的空间。有的时候条件反转一下可以简化处理逻辑,而有时则相反!
另外数组长度固定,超过的话就会有重新分配空间的开销。
代码
public TreeNode lowestCommonAncestor_v1(TreeNode root, TreeNode p, TreeNode q) {
Stack<TreeNode> pStack = findPath(root, p.val, new Stack<>());
Stack<TreeNode> qStack = findPath(root, q.val, new Stack<>());
HashSet<Integer> set = new HashSet<>();
while (!pStack.empty()){
set.add(pStack.pop().val);
}
TreeNode res = null;
while (!qStack.empty()){
res = qStack.pop();
if (set.contains(res.val)) {
break;
}
}
return res;
}
public Stack<TreeNode> findPath(TreeNode subRoot, int value, Stack<TreeNode> stack) {
stack.push(subRoot);
if (subRoot.val == value) {
return stack;
} else if (subRoot.val > value) {
return findPath(subRoot.left, value, stack);
} else {
return findPath(subRoot.right, value, stack);
}
}
/**
* 还有一个更好的做法是一次遍历同时比较p,q,如果一个大于等于,一个小于等于 当前节点值,表明已经/准备分叉了
* 这个是看了题解之后发现的
*/
public TreeNode lowestCommonAncestor_v2(TreeNode root, TreeNode p, TreeNode q) {
if (root.val > p.val && root.val > q.val) {
return lowestCommonAncestor_v2(root.left, p, q);
} else if (root.val < p.val && root.val < q.val) {
return lowestCommonAncestor_v2(root.right, p, q);
} else {
return root;
}
}