數(shù)組區(qū)間問題(動態(tài)規(guī)劃)

寫在前

矩陣的問題一般需要這樣考慮:當前坐標(row, col)是如何“走”到的。矩陣問題的動態(tài)規(guī)劃即為填表問題。

1.數(shù)組區(qū)間和 (303-易)

題目描述:給定一個整數(shù)數(shù)組 nums,求出數(shù)組從索引 iji ≤ j)范圍內元素的總和,包含 i、j兩點。

示例

Given nums = [-2, 0, 3, -5, 2, -1]

sumRange(0, 2) -> 1
sumRange(2, 5) -> -1
sumRange(0, 5) -> -3

思路:本題關鍵:若求解i - j區(qū)間范圍的元素和,可以用sum[j + 1] - sum[i] ,注:sum[i]為 0 - i-1的累加和。 那么要注意這時sum數(shù)組開辟空間【nums.length + 1】,保證sum[i + 1]可以取到。

代碼:動態(tài)規(guī)劃

private int[] sums;
public NumArray(int[] nums) {
    sums = new int[nums.length + 1];
    for (int i = 1; i <= nums.length; ++i) {
        sums[i] = sums[i - 1] + nums[i - 1];
    }
}
// 區(qū)間和方法入口
public int sumRange(int i, int j) {
    return sums[j + 1] - sums[i];
}

2.不同路徑II (63-中)

題目描述:如果一個至少有三個元素的等差數(shù)列。函數(shù)要返回數(shù)組 A 中所有為等差數(shù)組的子數(shù)組個數(shù)。

示例

A = [0, 1, 2, 3, 4]

return: 6

[0, 1, 2],
[1, 2, 3],
[0, 1, 2, 3],
[0, 1, 2, 3, 4],
[ 1, 2, 3, 4],
[2, 3, 4]

思路:關鍵定義dp數(shù)組,dp[i]:表示以A[i]作為結尾的所有等差數(shù)組的子數(shù)組的個數(shù)。故,當A[i] - A[i-1] == A[i-1] - A[i-2],即A[i]、A[i-1]、A[i-2]構成等差數(shù)列,狀態(tài)轉移方程為dp[i] = dp[i-1] + 1。例

dp[2] = 1
    [0, 1, 2]
dp[3] = dp[2] + 1 = 2
    [0, 1, 2, 3], // [0, 1, 2] 之后加一個 3
    [1, 2, 3]     // 新的遞增子區(qū)間
dp[4] = dp[3] + 1 = 3
    [0, 1, 2, 3, 4], // [0, 1, 2, 3] 之后加一個 4
    [1, 2, 3, 4],    // [1, 2, 3] 之后加一個 4
    [2, 3, 4]        // 新的遞增子區(qū)間

代碼:動態(tài)規(guī)劃

public int numberOfArithmeticSlices(int[] A) {
    int n = A.length;
    int ret = 0;
    int[] dp = new int[n];
    for (int i = 2; i < n; ++i) {
        // A[i]、A[i-1]、A[i-2]為等差數(shù)列
        if (A[i] - A[i - 1] == A[i - 1] - A[i - 2]) {
            dp[i] = dp[i - 1] + 1;
            ret += dp[i];
        }
    }
    return ret;
}

3.戳氣球 (312-中)

題目描述:有 n 個氣球,編號為0 到 n - 1,每個氣球上都標有一個數(shù)字,這些數(shù)字存在數(shù)組 nums 中。

現(xiàn)在要求你戳破所有的氣球。戳破第 i 個氣球,你可以獲得 nums[i - 1] * nums[i] * nums[i + 1] 枚硬幣。 這里的 i - 1 和 i + 1 代表和 i 相鄰的兩個氣球的序號。如果 i - 1或 i + 1 超出了數(shù)組的邊界,那么就當它是一個數(shù)字為 1 的氣球。

求所能獲得硬幣的最大數(shù)量。

示例

輸入:nums = [3,1,5,8]
輸出:167
解釋:
nums = [3,1,5,8] --> [3,5,8] --> [3,8] --> [8] --> []
coins =  3*1*5    +   3*5*8   +  1*3*8  + 1*8*1 = 167

思路:@狗大王,就先別管前面是怎么戳的,你只要管這個區(qū)間最后一個被戳破的是哪個氣球,最后一個被戳爆的氣球就設置為 k!!!

  • dp數(shù)組:假設 dp[i][j] 表示開區(qū)間 (i,j) 內你能拿到的最多金幣
  • 狀態(tài)轉移方程:假設你戳中第k個氣球(第k個氣球最后被戳爆),dp[i][j] = dp[i][k] + val[i]*val[k]*val[j] + dp[k][j]

ps:根據(jù)題意我們可以使用輔助數(shù)組,并且在數(shù)組的首尾添加1,為啥是開區(qū)間,我們不能戳中假邊界。

代碼:動態(tài)規(guī)劃

class Solution {
    public int maxCoins(int[] nums) {
        int n = nums.length;
        int[] tmp = new int[n + 2];
        tmp[0] = 1;
        tmp[n + 1] = 1;
        for (int i = 0; i < n; i++) {
            tmp[i + 1] = nums[i];
        }
        int[][] dp = new int[n + 2][n + 2];

        // len表示開區(qū)間的長度(對每一個區(qū)間長度進行循環(huán))
        for (int len = 3; len <= n + 2; len++) {
            // i表示區(qū)間的左邊界,i + len - 1表示區(qū)間的右邊界
            for (int i = 0; i <= n + 2 - len; i++) {
                // k表示開區(qū)間內的索引(代表區(qū)間最后一個被戳破的氣球)
                int j = i + len - 1;
                for (int k = i + 1; k < j; k++) {
                    dp[i][j] = Math.max(dp[i][j], dp[i][k] + tmp[i]*tmp[k]*tmp[j] + dp[k][j]);
                }
            }
        }
        return dp[0][n + 1];
    }
}
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
【社區(qū)內容提示】社區(qū)部分內容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內容

友情鏈接更多精彩內容