Java實現 LeetCode 798 得分最高的最小輪調 (暴力分析)

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;
    } 

}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章