数组题目:得分最高的最小轮调

文章目录

题目

标题和出处

标题:得分最高的最小轮调

出处:798. 得分最高的最小轮调

难度

8 级

题目描述

要求

给定一个数组 nums \texttt{nums} nums,我们可以将它按一个非负整数 k \texttt{k} k 进行轮调,这样可以使数组变为 nums[k],   nums[k   +   1],   nums[k   +   2],   … ,   nums[nums.length   -   1],   nums[0],   nums[1],   … ,   nums[k   -   1] \texttt{nums[k], nums[k + 1], nums[k + 2], \ldots, nums[nums.length - 1], nums[0], nums[1], \ldots, nums[k - 1]} nums[k], nums[k + 1], nums[k + 2], …, nums[nums.length - 1], nums[0], nums[1], …, nums[k - 1] 的形式。此后,任何值小于或等于其索引的项都可以计 1 \texttt{1} 1 分。

  • 例如,如果数组为 [2,   4,   1,   3,   0] \texttt{[2, 4, 1, 3, 0]} [2, 4, 1, 3, 0],我们按 k   =   2 \texttt{k = 2} k = 2 进行轮调后,它将变成 [1,   3,   0,   2,   4] \texttt{[1, 3, 0, 2, 4]} [1, 3, 0, 2, 4]。这将计 3 \texttt{3} 3 分,因为 1 > 0 \texttt{1} > \texttt{0} 1>0( 0 \texttt{0} 0 分), 3 > 1 \texttt{3} > \texttt{1} 3>1( 0 \texttt{0} 0 分), 0 ≤ 2 \texttt{0} \le \texttt{2} 0≤2( 1 \texttt{1} 1 分), 2 ≤ 3 \texttt{2} \le \texttt{3} 2≤3( 1 \texttt{1} 1 分), 4 ≤ 4 \texttt{4} \le \texttt{4} 4≤4( 1 \texttt{1} 1 分)。

在所有可能的轮调中,返回我们所能得到的最高分数对应的轮调索引 k \texttt{k} k。如果有多个答案,返回满足条件的最小的索引 k \texttt{k} k。

示例

示例 1:

输入: [2,   3,   1,   4,   0] \texttt{[2, 3, 1, 4, 0]} [2, 3, 1, 4, 0]
输出: 3 \texttt{3} 3
解释:
下面列出了每个 k \texttt{k} k 的得分:
k   =   0 \texttt{k = 0} k = 0, nums   =   [2,3,1,4,0] \texttt{nums = [2,3,1,4,0]} nums = [2,3,1,4,0], 2 \texttt{2} 2 分
k   =   1 \texttt{k = 1} k = 1, nums   =   [3,1,4,0,2] \texttt{nums = [3,1,4,0,2]} nums = [3,1,4,0,2], 3 \texttt{3} 3 分
k   =   2 \texttt{k = 2} k = 2, nums   =   [1,4,0,2,3] \texttt{nums = [1,4,0,2,3]} nums = [1,4,0,2,3], 3 \texttt{3} 3 分
k   =   3 \texttt{k = 3} k = 3, nums   =   [4,0,2,3,1] \texttt{nums = [4,0,2,3,1]} nums = [4,0,2,3,1], 4 \texttt{4} 4 分
k   =   4 \texttt{k = 4} k = 4, nums   =   [0,2,3,1,4] \texttt{nums = [0,2,3,1,4]} nums = [0,2,3,1,4], 3 \texttt{3} 3 分
所以我们应当选择 k   =   3 \texttt{k = 3} k = 3,得分最高。

示例 2:

输入: [1,   3,   0,   2,   4] \texttt{[1, 3, 0, 2, 4]} [1, 3, 0, 2, 4]
输出: 0 \texttt{0} 0
解释: nums \texttt{nums} nums 无论怎么变化总是有 3 \texttt{3} 3 分。所以我们将选择最小的 k \texttt{k} k,即 0 \texttt{0} 0。

数据范围

  • 1 ≤ nums.length ≤ 10 5 \texttt{1} \le \texttt{nums.length} \le \texttt{10}^\texttt{5} 1≤nums.length≤105
  • 0 ≤ nums[i] < nums.length \texttt{0} \le \texttt{nums[i]} < \texttt{nums.length} 0≤nums[i]<nums.length

解法

思路和算法

最直观的做法是对所有可能的轮调分别计算得分,记录最高得分和最高得分对应的轮调索引。当数组 nums \textit{nums} nums 的长度为 n n n 时,共有 n n n 种可能的轮调,对于每种轮调计算得分需要 O ( n ) O(n) O(n) 的时间,因此总时间复杂度为 O ( n 2 ) O(n^2) O(n2),会超出时间限制,因此必须优化。

对于数组 nums \textit{nums} nums 中的元素 x x x,只有当 x x x 的索引大于或等于 x x x 时, x x x 才会计 1 1 1 分,因此 x x x 的索引在 [ x , n − 1 ] [x, n-1] [x,n−1] 范围内时会计 1 1 1 分,该范围的长度为 n − x n-x n−x,即有 n − x n-x n−x 个轮调使得 x x x 计 1 1 1 分,其余 x x x 个轮调使得 x x x 不计分。

如果初始时, x x x 位于索引 i i i,即 nums [ i ] = x \textit{nums}[i]=x nums[i]=x,则当轮调索引为 k k k 时, x x x 将位于索引 ( i − k + n )   m o d   n (i-k+n) \bmod n (i−k+n)modn。当 ( i − k + n )   m o d   n ≥ x (i-k+n) \bmod n \ge x (i−k+n)modn≥x 时, x x x 将会计 1 1 1 分。

将上述不等式进行转换可以得到:当 x x x 计 1 1 1 分时,轮调索引 k k k 应满足 k ≤ ( i − x + n )   m o d   n k \le (i-x+n) \bmod n k≤(i−x+n)modn。又由于使得 x x x 计 1 1 1 分的 k k k 有 n − x n-x n−x 个取值,因此当 x x x 计 1 1 1 分时,有 k ≥ ( i + 1 )   m o d   n k \ge (i+1) \bmod n k≥(i+1)modn。

由于上述 k k k 的表示有取余运算,因此 k k k 的实际取值范围由 x x x、 i i i 和 n n n 决定:

  • 当 i < x i<x i<x 时, i + 1 ≤ k ≤ i + n − x i+1 \le k \le i+n-x i+1≤k≤i+n−x;

  • 当 i ≥ x i \ge x i≥x 时, k ≥ i + 1 k \ge i+1 k≥i+1 或 k ≤ i − x k \le i-x k≤i−x。

对于数组 nums \textit{nums} nums 中的任意元素 x x x,都可以得到 x x x 计 1 1 1 分的轮调索引范围。只要找到数组 nums \textit{nums} nums 中的每一个元素对应的计 1 1 1 分的轮调索引范围,然后在所有轮调索引中找到计 1 1 1 分的次数最多的轮调索引,即为得分最高的轮调索引。如果存在多个得分最高的轮调索引,则取其中的最小轮调索引。

创建长度为 n n n 的数组 points \textit{points} points,其中 points [ k ] \textit{points}[k] points[k] 表示轮调索引为 k k k 时的得分。为了优化时间复杂度,可以使用差分数组的做法计算 points \textit{points} points 的每个元素值,具体做法如下:

  1. 遍历数组 nums \textit{nums} nums,对于 nums [ i ] = x \textit{nums}[i]=x nums[i]=x,令 low = ( i + 1 )   m o d   n \textit{low}=(i+1) \bmod n low=(i+1)modn, high = ( i + n − x + 1 )   m o d   n \textit{high}=(i+n-x+1) \bmod n high=(i+n−x+1)modn,然后将 points [ low ] \textit{points}[\textit{low}] points[low] 的值加 1 1 1,将 points [ high ] \textit{points}[\textit{high}] points[high] 的值减 1 1 1,如果 low > high \textit{low}>\textit{high} low>high,则再将 points [ 0 ] \textit{points}[0] points[0] 的值加 1 1 1;

  2. 维护最高得分和最高得分下标,初始值都是 0 0 0,遍历数组 points \textit{points} points 并计算其前缀和,使用前缀和更新最高得分和最高得分下标。遍历结束时即可得到最高得分,此时的最高得分下标即为得分最高的最小轮调索引。

上述差分数组的做法的正确性说明如下:

  • 对于数组 nums \textit{nums} nums 中的任意元素 x x x,如果 x x x 计 1 1 1 分的轮调索引范围是连续的,则该范围是 [ i + 1 , i + n − x ] [i+1, i+n-x] [i+1,i+n−x],等价于 [ low , high − 1 ] [\textit{low},\textit{high}-1] [low,high−1],将 points [ low ] \textit{points}[\textit{low}] points[low] 的值加 1 1 1,将 points [ high ] \textit{points}[\textit{high}] points[high] 的值减 1 1 1,然后更新前缀和,该范围内的得分都会加 1 1 1;

  • 对于数组 nums \textit{nums} nums 中的任意元素 x x x,如果 x x x 计 1 1 1 分的轮调索引范围是不连续的,则该范围是 [ 0 , i − x ] ∪ [ i + 1 , n − 1 ] [0, i-x] \cup [i+1, n-1] [0,i−x]∪[i+1,n−1],等价于 [ 0 , high − 1 ] ∪ [ low , n − 1 ] [0, \textit{high}-1] \cup [\textit{low}, n-1] [0,high−1]∪[low,n−1],将 points [ 0 ] \textit{points}[0] points[0] 和 points [ low ] \textit{points}[\textit{low}] points[low] 的值分别加 1 1 1,将 points [ high ] \textit{points}[\textit{high}] points[high] 的值减 1 1 1,然后更新前缀和,该范围内的得分都会加 1 1 1。

代码

class Solution {
    public int bestRotation(int[] nums) {
        int length = nums.length;
        int[] points = new int[length];
        for (int i = 0; i < length; i++) {
            int low = (i + 1) % length;
            int high = (i + length - nums[i] + 1) % length;
            points[low]++;
            points[high]--;
            if (low > high) {
                points[0]++;
            }
        }
        int maxIndex = 0;
        int maxScore = 0;
        int curScore = 0;
        for (int i = 0; i < length; i++) {
            curScore += points[i];
            if (curScore > maxScore) {
                maxIndex = i;
                maxScore = curScore;
            }
        }
        return maxIndex;
    }
}

复杂度分析

  • 时间复杂度: O ( n ) O(n) O(n),其中 n n n 是数组 nums \textit{nums} nums 的长度。整个过程需要遍历数组 nums \textit{nums} nums 两次。

  • 空间复杂度: O ( n ) O(n) O(n),其中 n n n 是数组 nums \textit{nums} nums 的长度。需要创建一个长度为 n n n 的数组 points \textit{points} points。

上一篇:725. 分隔链表难度中等214给你一个头结点为 head 的单链表和一个整数 k ,请你设计一个算法将链表分隔为 k 个连续的部分。每部分的长度应该尽可能的相等:任意两部分的


下一篇:【最佳解法】剑指 Offer 42. 连续子数组的最大和