LeetCode筆記:396. Rotate Function

問題:

Given an array of integers A and let n to be its length.
Assume Bk to be an array obtained by rotating the array A k positions clock-wise, we define a "rotation function" F on A as follow:
F(k) = 0 * Bk[0] + 1 * Bk[1] + ... + (n-1) * Bk[n-1].
Calculate the maximum value of F(0), F(1), ..., F(n-1).
Note:
n is guaranteed to be less than 105.
Example:

A = [4, 3, 2, 6]
F(0) = (0 * 4) + (1 * 3) + (2 * 2) + (3 * 6) = 0 + 3 + 4 + 18 = 25
F(1) = (0 * 6) + (1 * 4) + (2 * 3) + (3 * 2) = 0 + 4 + 6 + 6 = 16
F(2) = (0 * 2) + (1 * 6) + (2 * 4) + (3 * 3) = 0 + 6 + 8 + 9 = 23
F(3) = (0 * 3) + (1 * 2) + (2 * 6) + (3 * 4) = 0 + 2 + 12 + 12 = 26
So the maximum value of F(0), F(1), F(2), F(3) is F(3) = 26.

大意:

給出一個整型數(shù)組A雄人,設(shè)n為其長度。
假設(shè)Bk是將A進行k此順時針旋轉(zhuǎn)后的數(shù)組层皱,我們定義一個A的“旋轉(zhuǎn)函數(shù)”F识椰,如下:
F(k) = 0 * Bk[0] + 1 * Bk[1] + ... + (n-1) * Bk[n-1].
注意:
n保證不會超過10的5次方
例子:

A = [4, 3, 2, 6]
F(0) = (0 * 4) + (1 * 3) + (2 * 2) + (3 * 6) = 0 + 3 + 4 + 18 = 25
F(1) = (0 * 6) + (1 * 4) + (2 * 3) + (3 * 2) = 0 + 4 + 6 + 6 = 16
F(2) = (0 * 2) + (1 * 6) + (2 * 4) + (3 * 3) = 0 + 6 + 8 + 9 = 23
F(3) = (0 * 3) + (1 * 2) + (2 * 6) + (3 * 4) = 0 + 2 + 12 + 12 = 26
所以 F(0), F(1), F(2), F(3) 中最大的值為 F(3) = 26.

思路:

這個題目的意思就是對A進行旋轉(zhuǎn)求多項式,每次旋轉(zhuǎn)系數(shù)移動一次鄙漏,旋轉(zhuǎn)一次求出一個結(jié)果嗤谚,看哪個最大就返回哪個棺蛛。

我的做法是直接進行每一個的計算然后找最大的,代碼挺簡單巩步,時間復(fù)雜度是O(n平方)旁赊,很長。

代碼(Java):

public class Solution {
    public int maxRotateFunction(int[] A) {
        if (A.length == 0) return 0;
        int result = -2147483648;
        for (int i = 0; i < A.length; i++) {
            int sum = sum(A, A.length - i);
            if (sum > result) result = sum;
        }
        return result;
    }
    
    public int sum(int[] A, int index) {
        int sum = 0;
        for (int i = 0; i < A.length; i++) {
            if (index >= A.length) index = 0;
            sum += i * A[index];
            index ++;
        }
        return sum;
    }
}

他山之石:

public class Solution {
    public int maxRotateFunction(int[] A) {
        int allSum = 0;
        int len = A.length;
        int F = 0;
        for (int i = 0; i < len; i++) {
            F += i * A[i];
            allSum += A[i];
        }
        int max = F;
        for (int i = len - 1; i >= 1; i--) {
            F = F + allSum - len * A[i];
            max = Math.max(F, max);
        }
        return max; 
    }
}

這個做法的好處在于只需要O(n)的時間椅野,快很多终畅。他對題目的要求進行了一些數(shù)學(xué)計算,然后得出了一個方便計算的式子竟闪,過程如下:

F(k) = 0 * Bk[0] + 1 * Bk[1] + ... + (n-1) * Bk[n-1]
F(k-1) = 0 * Bk-1[0] + 1 * Bk-1[1] + ... + (n-1) * Bk-1[n-1]
   = 0 * Bk[1] + 1 * Bk[2] + ... + (n-2) * Bk[n-1] + (n-1) * Bk[0]

那么离福,

F(k) - F(k-1) = Bk[1] + Bk[2] + ... + Bk[n-1] + (1-n)Bk[0]
      = (Bk[0] + ... + Bk[n-1]) - nBk[0]
      = sum - nBk[0]

因此,

F(k) = F(k-1) + sum - nBk[0]

那Bk[0]是什么呢炼蛤?

k = 0; B[0] = A[0];
k = 1; B[0] = A[len-1];
k = 2; B[0] = A[len-2];
...

這樣妖爷,也就有了上面的代碼了。

合集:https://github.com/Cloudox/LeetCode-Record


查看作者首頁

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末理朋,一起剝皮案震驚了整個濱河市絮识,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌暗挑,老刑警劉巖笋除,帶你破解...
    沈念sama閱讀 211,194評論 6 490
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異炸裆,居然都是意外死亡垃它,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,058評論 2 385
  • 文/潘曉璐 我一進店門烹看,熙熙樓的掌柜王于貴愁眉苦臉地迎上來国拇,“玉大人,你說我怎么就攤上這事惯殊〗戳撸” “怎么了?”我有些...
    開封第一講書人閱讀 156,780評論 0 346
  • 文/不壞的土叔 我叫張陵土思,是天一觀的道長务热。 經(jīng)常有香客問我,道長己儒,這世上最難降的妖魔是什么崎岂? 我笑而不...
    開封第一講書人閱讀 56,388評論 1 283
  • 正文 為了忘掉前任,我火速辦了婚禮闪湾,結(jié)果婚禮上冲甘,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好江醇,可當(dāng)我...
    茶點故事閱讀 65,430評論 5 384
  • 文/花漫 我一把揭開白布濒憋。 她就那樣靜靜地躺著,像睡著了一般陶夜。 火紅的嫁衣襯著肌膚如雪凛驮。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 49,764評論 1 290
  • 那天律适,我揣著相機與錄音辐烂,去河邊找鬼遏插。 笑死捂贿,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的胳嘲。 我是一名探鬼主播厂僧,決...
    沈念sama閱讀 38,907評論 3 406
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼了牛!你這毒婦竟也來了颜屠?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 37,679評論 0 266
  • 序言:老撾萬榮一對情侶失蹤鹰祸,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后蛙婴,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體粗井,經(jīng)...
    沈念sama閱讀 44,122評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡街图,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 36,459評論 2 325
  • 正文 我和宋清朗相戀三年浇衬,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片餐济。...
    茶點故事閱讀 38,605評論 1 340
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖醉冤,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情韵吨,我是刑警寧澤椿疗,帶...
    沈念sama閱讀 34,270評論 4 329
  • 正文 年R本政府宣布糠悼,位于F島的核電站铝条,受9級特大地震影響班缰,放射性物質(zhì)發(fā)生泄漏埠忘。R本人自食惡果不足惜馒索,卻給世界環(huán)境...
    茶點故事閱讀 39,867評論 3 312
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望鉴腻。 院中可真熱鬧疯趟,春花似錦信峻、人聲如沸产镐。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,734評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽玻孟。三九已至,卻和暖如春黍翎,著一層夾襖步出監(jiān)牢的瞬間趟紊,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 31,961評論 1 265
  • 我被黑心中介騙來泰國打工碱璃, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人谐丢。 一個月前我還...
    沈念sama閱讀 46,297評論 2 360
  • 正文 我出身青樓,卻偏偏與公主長得像衷佃,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 43,472評論 2 348

推薦閱讀更多精彩內(nèi)容