C++ 动态规划问题

C++ 动态规划问题

小美拿到了一个大小为n的数组，她希望删除一个区间后，使得剩余所有元素的乘积末尾至少有k个 0。小美想知道，一共有多少种不同的删除方案？(美团面试题）

输入描述：
第一行输入两个正整数n,k。
第二行输入n个正整数a_i，代表小美拿到的数组。
1<=n,k <= 10^5
1<= a_i <= 10^9

示例1

输入例子：
5 2
2 5 3 4 20

输出例子：
4

例子说明：
第一个方案，删除[3]。
第二个方案，删除[4]。
第三个方案，删除[3,4]。
第四个方案，删除[2]。

#include <iostream>
#include <vector>

using namespace std;

int countDeletionOptions(vector<int>& nums, int k) {
    int n = nums.size();
    vector<vector<int>> dp(n + 1, vector<int>(k + 1, 0));

    int zeroCount = 0;
    for (int i = 0; i < n; i++) {
        if (nums[i] == 0) {
            zeroCount++;
        }
    }

    for (int i = 0; i <= n; i++) {
        dp[i][0] = 1;
    }

    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= k; j++) {
            if (nums[i - 1] == 0) {
                dp[i][j] = dp[i - 1][j];
            } else {
                dp[i][j] = dp[i - 1][j] + dp[i - 1][j - 1];
            }
        }
    }

    int totalOptions = 0;
    for (int i = 1; i <= n; i++) {
        for (int j = 0; j <= k; j++) {
            totalOptions += dp[i][j];
        }
    }

    return totalOptions/10;
}

int main() {
    int n,k;
    cin >> n >> k;
    vector<int> nums(n);
    for(int i=0;i<n;i++){
        cin >> nums[i];
    }

    int options = countDeletionOptions(nums, k);

    cout << options << endl;

    return 0;
}