LC 3366. Minimum Array Sum

https://leetcode.com/problems/minimum-array-sum/description/

Tag: DP

class Solution {
public:
    int twoOp(int num, int k) {
        if (ceil(1.0 * num / 2) < k) {
            return ceil(1.0 * (num - k) / 2);
        } else {
            return ceil(1.0 * num / 2) - k;
        }
    }

    int minArraySum(vector<int>& nums, int l, int op1, int op2) {
        int n = nums.size();
        vector<vector<vector<int>>> f(n, vector<vector<int>>(op1 + 1, vector<int>(op2 + 1, INT_MAX)));
        f[0][0][0] = nums[0];
        if (op1)
            f[0][1][0] = ceil(1.0 * nums[0] / 2);
        if (op2 && nums[0] >= l)
            f[0][0][1] = nums[0] - l;
        if (op1 && op2 && nums[0] >= l)
            f[0][1][1] = twoOp(nums[0], l);
        for (int i = 1; i < n; i++) {
            for (int j = 0; j <= op1; j++) {
                for (int k = 0; k <= op2; k++) {
                    if (f[i - 1][j][k] != INT_MAX)
                        f[i][j][k] = f[i - 1][j][k] + nums[i];
                    if (j - 1 >= 0 && f[i - 1][j - 1][k] != INT_MAX)
                        f[i][j][k] = min(f[i][j][k], int(f[i - 1][j - 1][k] + ceil(1.0 * nums[i] / 2)));
                    if (k - 1 >= 0 && f[i - 1][j][k - 1] != INT_MAX && nums[i] >= l) 
                        f[i][j][k] = min(f[i][j][k], f[i - 1][j][k - 1] + nums[i] - l);
                    if (j - 1 >= 0 && k - 1 >= 0 && f[i - 1][j - 1][k - 1] != INT_MAX && nums[i] >= l)
                        f[i][j][k] = min(f[i][j][k], f[i - 1][j - 1][k - 1] + twoOp(nums[i], l));
                }
            }
        }
        int ans = INT_MAX;
        for (int i = 0; i <= op2; i++)
            ans = min(ans, f[n - 1][op1][i]);
        return ans;
    }
};

一個數字有三種可能，分別是只進行 op1 或是 op2 或是同時執行 op1 以及 op2，若遇到同時執行 op1 以及 op2 的需要注意一下執行順序，有些可以先做 op1 再做 op2 但是有些必須先做 op2 再做 op1。