LeetCode
好子数组的最大分数
题目链接:1793. 好子数组的最大分数 - 力扣(LeetCode)
题目描述
给你一个整数数组 nums
(下标从 0 开始) 和一个整数 k
。
一个子数组 (i, j)
的 分数 定义为 min(nums[i], nums[i+1], ..., nums[j]) * (j - i + 1)
。一个 好 子数组的两个端点下标需要满足 i <= k <= j
。
请你返回 好 子数组的最大可能 分数 。
示例 1:
输入:nums = [1,4,3,7,4,5], k = 3
输出:15
解释:最优子数组的左右端点下标是 (1, 5) ,分数为 min(4,3,7,4,5) * (5-1+1) = 3 * 5 = 15 。
示例 2:
输入:nums = [5,5,4,5,4,1,1,1], k = 0
输出:20
解释:最优子数组的左右端点下标是 (0, 4) ,分数为 min(5,5,4,5,4) * (4-0+1) = 4 * 5 = 20 。
提示:
1 <= nums.length <= 105
1 <= nums[i] <= 2 * 104
0 <= k < nums.length
思路
双指针
代码
C++
class Solution {
public:
int maximumScore(vector<int>& nums, int k) {
int n = nums.size();
int ans = nums[k], min_h = nums[k];
int i = k, j = k;
for(int t = 0; t < n - 1; t++){
if(j == n - 1 || i && nums[i - 1] > nums[j + 1]){
min_h = min(min_h,nums[--i]);
} else{
min_h = min(min_h,nums[++j]);
}
ans = max(ans,min_h * (j - i + 1));
}
return ans;
}
};
Java
class Solution {
public int maximumScore(int[] nums, int k) {
int n = nums.length;
int ans = nums[k], minH = nums[k];
int i = k, j = k;
for(int t = 0; t < n - 1; t++){
if(j == n - 1 || i > 0 && nums[i - 1] > nums[j + 1]){
minH = Math.min(minH,nums[--i]);
} else{
minH = Math.min(minH, nums[++j]);
}
ans = Math.max(ans,minH * (j - i + 1));
}
return ans;
}
}