题目要求:
给你一个下标从 0 开始的整数数组 nums
,它表示一个 栈 ,其中 nums[0]
是栈顶的元素。
每一次操作中,你可以执行以下操作 之一 :
- 如果栈非空,那么 删除 栈顶端的元素。
- 如果存在 1 个或者多个被删除的元素,你可以从它们中选择任何一个,添加 回栈顶,这个元素成为新的栈顶元素。
同时给你一个整数 k
,它表示你总共需要执行操作的次数。
请你返回 恰好 执行 k
次操作以后,栈顶元素的 最大值 。如果执行完 k
次操作以后,栈一定为空,请你返回 -1
。
示例 1:
输入:nums = [5,2,2,4,0,6], k = 4 输出:5 解释: 4 次操作后,栈顶元素为 5 的方法之一为: - 第 1 次操作:删除栈顶元素 5 ,栈变为 [2,2,4,0,6] 。 - 第 2 次操作:删除栈顶元素 2 ,栈变为 [2,4,0,6] 。 - 第 3 次操作:删除栈顶元素 2 ,栈变为 [4,0,6] 。 - 第 4 次操作:将 5 添加回栈顶,栈变为 [5,4,0,6] 。 注意,这不是最后栈顶元素为 5 的唯一方式。但可以证明,4 次操作以后 5 是能得到的最大栈顶元素。
示例 2:
输入:nums = [2], k = 1 输出:-1 解释: 第 1 次操作中,我们唯一的选择是将栈顶元素弹出栈。 由于 1 次操作后无法得到一个非空的栈,所以我们返回 -1 。
题解:
这道题让我们操作一个栈,我们可以将元素弹出或是将弹出的元素的再次压入栈,每次弹出或压入算一个操作,求解出经过k次操作后可以得到的最大元素。
我们经过k次操作后有几种情况,当k的值大于栈的大小时我们可以得到栈中所有的元素,因此直接遍历栈就可以得到最大值。
而当k小于栈的大小时,我们可以得到k-1前的所有元素,和第k+1个元素(即所有的操作都在弹出元素。这种情况下,我们直接找到前k-1个元素中的最大值再和第k+1个元素进行对比即可,大的就是最大值。
还有一种特殊情况,当栈中只有一个元素,且操纵数为奇数时,我们无论怎么操纵都会将唯一的元素弹出,因此此时返回-1.
经过这样的分析,本题的逻辑就很清晰了,因此代码如下:
class Solution {
public int maximumTop(int[] nums, int k) {
int n = nums.length;
if(k%2==1&&n==1) return -1;
int max = 0;
int k1=0;
int k2=0;
for(int i=0;i<n;i++){
if(i<k-1) k1 = Math.max(k1,nums[i]);
if(i==k) k2 =nums[i];
max=Math.max(max,nums[i]);
}
if(k>n)return max;
return Math.max(k1,k2);
}
}