题目链接:K次操作后最大化顶端元素
题目描述:
给你一个下标从 0 开始的整数数组 nums ,它表示一个 栈 ,其中 nums[0] 是栈顶的元素。
每一次操作中,你可以执行以下操作 之一 :
如果栈非空,那么 删除 栈顶端的元素。
如果存在 1 个或者多个被删除的元素,你可以从它们中选择任何一个,添加 回栈顶,这个元素成为新的栈顶元素。
同时给你一个整数 k ,它表示你总共需要执行操作的次数。
请你返回 恰好 执行 k 次操作以后,栈顶元素的 最大值 。如果执行完 k 次操作以后,栈一定为空,请你返回 -1 。
分析:
1)如果k==0,无法进行任何操作,直接返回nums[0]即可。
2)如果length==1,即数组只有一个元素,那么当k为奇数时,栈一定为空;当k为偶数时,返回值一定是nums[0]。
3)如果k==1,那么返回值一定是nums[1]。
4)当k>length时,操作次数比数组还长,那么一定可以保证最大栈顶元素就是数组的最大值。
5)当k==length时,那么一定可以保证最大栈顶元素就是数组前k-1个元素的最大值。
6)剩下就是k<length的情况,最大栈顶元素是前k-1个元素的最大值和第k+1个元素之间的较大者。
代码如下:
var maximumTop = function (nums, k) {
if (k == 0)
return nums[0];
const length = nums.length;
if (length == 1){
if(k%2==1)
return -1;
return nums[0];
}
if (k == 1)
return nums[1];
let max = nums[0];
if (k > length) {
for (let i = 0; i < length; i++) {
max = Math.max(max, nums[i]);
}
return max;
}
for (let i = 0; i < k - 1; i++) {
max = Math.max(max, nums[i]);
}
if (k == length)
return max;
max = Math.max(max, nums[k])
return max;
};