Atcoder dp_m Candies 前缀和优化dp

题意

给$n$个孩子发$k$颗糖,每个孩子最多拿到$a_i$颗,问有多少种分法.
$n\le 100,k\le 10^5$.

题解

考虑$dp.$
非常裸的$dp$如下.

for (int i=1;i<=n;++i) {
  for (int j=k;~j;--j)
    for (int l=j+1;l<=j+a[i];++l) 
      dp[l]=(dp[l]+dp[j])%mod;
}

复杂度为$O(n\times k^2)$,显然不能通过.
令$d{p}_j$为前$i$(滚动数组)个孩子有$k$颗糖没有发到的总分法数.
假设当前的孩子最多能够分到$k$颗糖而不是$a_i$颗,则
$d{p}_k=上一个d{p}_k$(这个孩子没分到糖).
$d{p}_{k-1}=上一个d{p}_{k-1}+d{p}_k$(这个孩子可以分到一颗糖,也可以不分到糖).
以此类推,$d{p}_0=上一个d{p}_0+d{p}_1+...+d{p}_{k-1}+d{p}_k$.
但是实际上这个孩子最多只能分到$a_i$颗糖,因此我们要把超过$a_i$的部分减掉.
由于刚才已经做过前缀和,对$j$只需要减掉$d{p}_{j+a_i+1}$这个位置的答案就可以了.
这样一来本题就解决了.
谢谢大家.

const int yuzu=2e5,mod=1e9+7;
typedef ll fuko[yuzu|10];
fuko dp;
int main() {
  int i,n,k,a;
  read(n),read(k);
  for (dp[k]=1;read(a);) {
    for (i=k;i--;) dp[i]+=dp[i+1];
    for (i=0;i<k;++i) dp[i]=(dp[i]-dp[i+a+1])%mod;
  }
  printf("%lld\n",*dp);
}