798. 得分最高的最小輪調
給定一個數組 A,我們可以將它按一個非負整數 K 進行輪調,這樣可以使數組變爲 A[K], A[K+1], A{K+2], … A[A.length - 1], A[0], A[1], …, A[K-1] 的形式。此後,任何值小於或等於其索引的項都可以記作一分。
例如,如果數組爲 [2, 4, 1, 3, 0],我們按 K = 2 進行輪調後,它將變成 [1, 3, 0, 2, 4]。這將記作 3 分,因爲 1 > 0 [no points], 3 > 1 [no points], 0 <= 2 [one point], 2 <= 3 [one point], 4 <= 4 [one point]。
在所有可能的輪調中,返回我們所能得到的最高分數對應的輪調索引 K。如果有多個答案,返回滿足條件的最小的索引 K。
示例 1:
輸入:[2, 3, 1, 4, 0]
輸出:3
解釋:
下面列出了每個 K 的得分:
K = 0, A = [2,3,1,4,0], score 2
K = 1, A = [3,1,4,0,2], score 3
K = 2, A = [1,4,0,2,3], score 3
K = 3, A = [4,0,2,3,1], score 4
K = 4, A = [0,2,3,1,4], score 3
所以我們應當選擇 K = 3,得分最高。
示例 2:
輸入:[1, 3, 0, 2, 4]
輸出:0
解釋:
A 無論怎麼變化總是有 3 分。
所以我們將選擇最小的 K,即 0。
提示:
A 的長度最大爲 20000。
A[i] 的取值範圍是 [0, A.length]。
class Solution {
public int bestRotation(int[] A) {
int N = A.length;
int[] bad = new int[N];
for (int i = 0; i < N; ++i) {
//把這裏換成差分的形式
//對於A【i】小於等於i是加分的
//那麼 i - A[i] + 1 和 i之間是沒有分的
//這樣就可以轉換成left位置沒分,right有分
int left = (i - A[i] + 1 + N) % N;
int right = (i + 1) % N;
bad[left]--;
bad[right]++;
}
int best = -N;
int ans = 0, cur = 0;
for (int i = 0; i < N; ++i) {
cur += bad[i];
if (cur > best) {
best = cur;
ans = i;
}
}
return ans;
}
}