798. 得分最高的最小轮调
给你一个数组 nums,我们可以将它按一个非负整数 k 进行轮调,这样可以使数组变为 [nums[k], nums[k + 1], … nums[nums.length - 1], nums[0], nums[1], …, nums[k-1]] 的形式。此后,任何值小于或等于其索引的项都可以记作一分。
例如,数组为 nums = [2,4,1,3,0],我们按 k = 2 进行轮调后,它将变成 [1,3,0,2,4]。这将记为 3 分,因为 1 > 0 [不计分]、3 > 1 [不计分]、0 <= 2 [计 1 分]、2 <= 3 [计 1 分],4 <= 4 [计 1 分]。
在所有可能的轮调中,返回我们所能得到的最高分数对应的轮调下标 k 。如果有多个答案,返回满足条件的最小的下标 k 。
示例 1:
输入:nums = [2,3,1,4,0]
输出:3
解释:
下面列出了每个 k 的得分:
k = 0, nums = [2,3,1,4,0], score 2
k = 1, nums = [3,1,4,0,2], score 3
k = 2, nums = [1,4,0,2,3], score 3
k = 3, nums = [4,0,2,3,1], score 4
k = 4, nums = [0,2,3,1,4], score 3
所以我们应当选择 k = 3,得分最高。
示例 2:
输入:nums = [1,3,0,2,4]
输出:0
解释:
nums 无论怎么变化总是有 3 分。
所以我们将选择最小的 k,即 0。
提示:
1 <= nums.length <= 105
0 <= nums[i] < nums.length
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/smallest-rotation-with-highest-score
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
今天的每日一题。
首先注意0 <= nums[i] < nums.length,即不存在论调后不能使分数+1的数字(任何数字放在最后都可以+1)。以下分析通过画图比较好理解。
1、我们对任意一个索引处的数字进行判断,不管这个数字是否大于索引,我们都可以找到这个数字与索引相等的索引位置,并且将这个数字向后移动(轮调的索引选择向后移动)一定可以得分,向左一定失分。
2、我们从索引0开始遍历,用变量 val 收集每个数字小于等于索引的值的数量,同时创建一个 int 数组 step ,数组中与每个数字相等的索引位置的值 +1 ,这个数组相当于记录了在索引 i + 1 处轮调会失分的数字数量, +1 是因为在索引 i 处 step 记录的是与索引相同的数字数量,这时是不失分的,但是轮调索引向后移动这些数字就全部是失分的。
3、我们再进行一次遍历(轮调的位置从 1 开始),每次轮调减去当前轮调会失分的数字数量(因为遍历是向右的,即轮调的索引是向右的,原来失分的数字都会向左移依然是失分的,所以减去的数字就不用再减了,只需要减去当前失分的索引),计算方法是 val = val - step[i - 1] + 1 ,在计算的同时找出最大的分数并记录当前索引。
题解(Java)
class Solution {
public int bestRotation(int[] nums) {
int n = nums.length, val = 0;
int[] step = new int[n];
for (int i = 0; i < n; i++) {
if (nums[i] <= i) {
step[i - nums[i]]++;
val++;
} else {
step[n - (nums[i] - i)]++;
}
}
int ans = 0, max = val;
for (int i = 1; i < n; i++) {
val = val - step[i - 1] + 1;
if (val > max) {
ans = i;
max = val;
}
}
return ans;
}
}