Leetcode 188. Best Time to Buy and Sell Stock IV

Say you have an array for which the ith element is the price of a given stock on day i.
Design an algorithm to find the maximum profit. You may complete at most k transactions.

Note:
You may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).

思路:
自己沒有想出來解法刽宪,參見了其他人的解法雕凹。
動態(tài)規(guī)劃啥箭,局部最優(yōu)與全局最優(yōu)。
我們需要維護(hù)如下兩個量:
global[i][j]:當(dāng)前到達(dá)第i天最多可以進(jìn)行j次交易扰路,所得到的最大利潤。
local[i][j]:當(dāng)前到達(dá)第i天最多可以進(jìn)行j次交易舅桩,而且最后一次交易在當(dāng)天賣出靶草,所得到的最大利潤。
狀態(tài)轉(zhuǎn)移方程:
global[i][j] = max(local[i][j], global[i-1][j])
上述方程比較兩個量的大心豢选:①當(dāng)前局部最大值丢氢;②過往全局最大值。
local[i][j] = max(global[i-1][j-1] + max(diff, 0), local[i-1][j] + diff)
上述方程比較兩個量的大邢雀摹:
①全局到i-1天進(jìn)行j-1次交易疚察,然后加上今天的交易(如果今天的交易賺錢的話)。
②取局部第i-1天進(jìn)行j次交易仇奶,然后加上今天的差值(local[i-1][j]是第i-1天賣出的交易貌嫡,它加上diff后變成第i天賣出,并不會增加交易次數(shù)该溯。無論diff是正還是負(fù)都要加上岛抄,否則就不滿足local[i][j]必須在最后一天賣出的條件了)
另外需要注意的一個問題是,當(dāng)k遠(yuǎn)大于數(shù)組的大小時朗伶,上述算法將變得低效弦撩。因此將其改用不限交易次數(shù)的方式解決步咪。

public int maxProfit(int k, int[] prices) {
    int res = 0;
    if (k < 1 || prices == null || prices.length < 2) {
        return res;
    }
    if (k >= prices.length / 2) return quickSolve(prices);
    
    int[] local = new int[k + 1];
    int[] global = new int[k + 1];
    for (int i = 0; i < prices.length - 1; i++)
    {
        int diff = prices[i+1]-prices[i];
        for(int j = k; j >= 1; j--)
        {
            local[j] = Math.max(global[j-1]+(diff>0?diff:0), local[j]+diff);
            global[j] = Math.max(local[j],global[j]);
        }
    }
    return global[k];
}

private int quickSolve(int[] prices) {
    int len = prices.length, profit = 0;
    for (int i = 1; i < len; i++)
        // as long as there is a price gap, we gain a profit.
        if (prices[i] > prices[i - 1]) profit += prices[i] - prices[i - 1];
    return profit;
}

下面這種思路更好理解一些:

public int maxProfit1(int k, int[] prices) {
    int n = prices.length;
    if (n <= 1)
        return 0;

    //if k >= n/2, then you can make maximum number of transactions.
    if (k >=  n/2) {
        int maxPro = 0;
        for (int i = 1; i < n; i++) {
            if (prices[i] > prices[i-1])
                maxPro += prices[i] - prices[i-1];
        }
        return maxPro;
    }

    int[][] dp = new int[k+1][n];
    for (int i = 1; i <= k; i++) {
        int localMax = dp[i-1][0] - prices[0];
        for (int j = 1; j < n; j++) {
            dp[i][j] = Math.max(dp[i][j-1],  prices[j] + localMax);
            localMax = Math.max(localMax, dp[i-1][j] - prices[j]);
        }
    }
    return dp[k][n-1];
}
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末论皆,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子猾漫,更是在濱河造成了極大的恐慌点晴,老刑警劉巖,帶你破解...
    沈念sama閱讀 211,423評論 6 491
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件悯周,死亡現(xiàn)場離奇詭異粒督,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)禽翼,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,147評論 2 385
  • 文/潘曉璐 我一進(jìn)店門屠橄,熙熙樓的掌柜王于貴愁眉苦臉地迎上來族跛,“玉大人,你說我怎么就攤上這事锐墙〗负澹” “怎么了?”我有些...
    開封第一講書人閱讀 157,019評論 0 348
  • 文/不壞的土叔 我叫張陵溪北,是天一觀的道長桐绒。 經(jīng)常有香客問我,道長之拨,這世上最難降的妖魔是什么茉继? 我笑而不...
    開封第一講書人閱讀 56,443評論 1 283
  • 正文 為了忘掉前任,我火速辦了婚禮蚀乔,結(jié)果婚禮上烁竭,老公的妹妹穿的比我還像新娘。我一直安慰自己吉挣,他們只是感情好颖变,可當(dāng)我...
    茶點(diǎn)故事閱讀 65,535評論 6 385
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著听想,像睡著了一般腥刹。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上汉买,一...
    開封第一講書人閱讀 49,798評論 1 290
  • 那天衔峰,我揣著相機(jī)與錄音,去河邊找鬼蛙粘。 笑死垫卤,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的出牧。 我是一名探鬼主播穴肘,決...
    沈念sama閱讀 38,941評論 3 407
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼舔痕!你這毒婦竟也來了评抚?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 37,704評論 0 266
  • 序言:老撾萬榮一對情侶失蹤伯复,失蹤者是張志新(化名)和其女友劉穎慨代,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體啸如,經(jīng)...
    沈念sama閱讀 44,152評論 1 303
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡侍匙,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,494評論 2 327
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了叮雳。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片想暗。...
    茶點(diǎn)故事閱讀 38,629評論 1 340
  • 序言:一個原本活蹦亂跳的男人離奇死亡妇汗,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出说莫,到底是詐尸還是另有隱情铛纬,我是刑警寧澤,帶...
    沈念sama閱讀 34,295評論 4 329
  • 正文 年R本政府宣布唬滑,位于F島的核電站告唆,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏晶密。R本人自食惡果不足惜擒悬,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,901評論 3 313
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望稻艰。 院中可真熱鬧懂牧,春花似錦、人聲如沸尊勿。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,742評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽元扔。三九已至躯保,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間澎语,已是汗流浹背途事。 一陣腳步聲響...
    開封第一講書人閱讀 31,978評論 1 266
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留擅羞,地道東北人尸变。 一個月前我還...
    沈念sama閱讀 46,333評論 2 360
  • 正文 我出身青樓,卻偏偏與公主長得像减俏,于是被迫代替她去往敵國和親召烂。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 43,499評論 2 348

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