有n个位置,每个位置最多可以放a[i]个物品,现在你有m个物品想要全部放到位子上,求共有多少种放法?

设dp[i][j] 为前i个位置放了j个物品的方案数

然后我们可以想到枚举位置;
然后在枚举前面一共放了多少个物品;
在枚举当前位置放多少个物品
代码:复杂度为O(n * m ^ 2) 很明显超时

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

很明显枚举放了多少个物品是可以用前缀和优化的

#include <bits/stdc++.h>
#include <iostream>
#include <string.h>
#include <math.h>

using namespace std;
const int maxn = 1e5 + 5;
const int mod = 1e9 + 7;
int dp[105][maxn], a[maxn];
int sum[maxn];

int main() {
//    freopen("in.in", "r", stdin);
    int n, m;
    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
    }
    dp[1][0] = 1; sum[0] = 1;
    for (int i = 1; i <= a[1]; i++) {
        dp[1][i] = 1; sum[i] = sum[i -1] + dp[1][i];
    }
    for (int i = 1; i <= m; i++) {
        sum[i] = sum[i - 1] + dp[1][i];
    }
    for (int i = 2; i <= n; i++) {
        for (int j = 0; j <= m; j++) {
            if (j > a[i]) { //前缀和优化
                dp[i][j] = (sum[j] - sum[j - a[i] - 1] + mod) % mod;
            } else {
                dp[i][j] = sum[j];
            }
        }
        sum[0] = dp[i][0];
        for (int j = 1; j <= m; j++) sum[j] = (sum[j - 1] + dp[i][j]) % mod;

    }
    cout << dp[n][m] << endl;
    return 0;
}