51nod 1161 Partial Sums

找规律可以得到，经过k次变换后，a[j]在i上的累加是a[j]*C(i-j+k-1,k-1)。

由于k-1不变，因此可以用递推公式

C(n+1,m)=C(n,m)*(n+1)/(n-m+1)

计算组合数。

#include<bits/stdc++.h>
using namespace std;

const long long mod=1e9+7;
const int MAXN=5050;
long long a[MAXN],w[MAXN];

void read(long long&a){
    char ch;while(!((ch=getchar())>='0')&&(ch<='9'));
    a=ch-'0';while(((ch=getchar())>='0')&&(ch<='9'))a*=10,a+=ch-'0';
}

inline void prin_d(long long x)
{
    if (x > 9)
    {
        prin_d(x / 10);
    }
    putchar(x % 10 + '0');
    return ;
}

long long powmod(long long x,long long p)
{
	long long ret=1;
	while(p)
	{
		if(p&1)
			ret=x*ret%mod;
		p>>=1;
		x=x*x%mod;
	}
	return ret;
}

int main()
{
	long long i,j,n,k,ans,tmp;
	while(~scanf("%lld%lld",&n,&k))
	{
		for(i=1;i<=n;i++)
			read(a[i]);
		w[0]=1;
		tmp=1;
		for(i=1;i<n;i++)
		{
			tmp=tmp*(i+k-1)%mod*powmod(i,mod-2)%mod;
			w[i]+=tmp;
			if(w[i]>mod)
				w[i]%=mod;
		}
		for(i=1;i<=n;i++)
		{
			ans=0;
			for(j=i;j>=1;j--)
			{
				ans+=w[i-j]*a[j]%mod;
				if(ans>mod)
					ans%=mod;
			}
			prin_d(ans);
			puts("");
		} 
	}
}