CodeForces 223 C.Partial Sums（dp+组合数学）

Description

给出一个长度为$n$的序列$a_i$，对其求$k$次前缀和得到序列$s_i$，求$s_i$

Input

第一行两个整数$n,k$，之后输入$n$个整数$a_i$$(1\le n\le 2000,0\le k\le 10^9,0\le a_i\le 10^9)$

Output

输出做$k$次前缀和后得到的$s$序列，结果模$10^9+7$

Sample Input

3 1
1 2 3

Sample Output

1 3 6

Solution

固定$a_j$，考虑求$k$次前缀和后$a_j$对$s_i(i\ge j)$的贡献次数$dp[k][i]$，显然有转移$dp[k][i]=dp[k][i-1]+dp[k-1][i]$

这个转移方程和组合数的递推式$C[i][j]=C[i][j-1]+C[i-1][j]$一样，我们取好初值去看$dp$和组合数的关系

$k=0$时，$dp[0][j]=1$， $i=j$时，$dp[k][j]=1$

令$dp[k][i]=C[k][i-j]$，简单验证知该式成立，故$s_i=\sum\limits_{j=1}^iC_k^{i-j}a_j\ mod\ 10^9+7$， 预处理$C[k][i]$，之后$O(n^2)$求出 s <script type="math/tex" id="MathJax-Element-29">s</script>序列

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=2005;
#define mod 1000000007
int inv[maxn];
void init(int n=2000)
{
    inv[1]=1;
    for(int i=2;i<=n;i++)inv[i]=mod-(ll)(mod/i)*inv[mod%i]%mod;
}
int n,k,a[maxn],b[maxn],c[maxn];
int main()
{
    init();
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    b[1]=1;
    if(k)for(int i=1;i<n;i++)b[i+1]=(ll)(k+i-1)*inv[i]%mod*b[i]%mod;
    for(int i=1;i<=n;i++)
    {
        c[i]=0;
        for(int j=i;j>=1;j--)c[i]=(c[i]+(ll)b[i-j+1]*a[j]%mod)%mod;
    }
    for(int i=1;i<=n;i++)printf("%d%c",c[i],i==n?'\n':' ');
    return 0;
}