BJ模拟 简单粗暴的题目【二项式定理】

解题思路：

已知$n,k$和长度为n的数列$a_i$，对每个$1\le i\le n$，求$\sum\limits_{j=1}^i(\sum\limits_{l=j}^{i}a[l])^k$。
$1\le n\le 500000,1\le k\le 100$

解题思路：

设$s_i$为后缀和，则:
$ans_i=\sum\limits_{j=1}^{i}(s_j-s_{i+1})^k=\sum\limits_{j=1}^{i}\sum\limits_{l=0}^{k}\binom{k}{l}(-1)^ls_{i+1}^ls_j^{k-l}=\sum\limits_{l=0}^k\binom{k}{l}(-1)^ls_{i+1}^l\sum\limits_{j=1}^{i}s_j^{k-l}$
预处理一下即可做到$O(nk)$

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')c=getchar(),f=-1;
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
const int N=500005,K=105,mod=1e9+7;
int n,k;
int a[N],s[N],c[K][K],ss[N][K];
int main()
{
    //freopen("lx.in","r",stdin);
    n=getint(),k=getint();
    for(int i=1;i<=n;i++)a[i]=getint();
    for(int i=n;i;i--)s[i]=(s[i+1]+a[i])%mod;
    for(int i=1;i<=n;i++)
        for(int j=0,tmp=1;j<=k;j++,tmp=1ll*tmp*s[i]%mod)
            ss[i][j]=(ss[i-1][j]+tmp)%mod;
    for(int i=0;i<=k;i++)
    {
        c[i][0]=1;
        for(int j=1;j<=i;j++)
            c[i][j]=(c[i-1][j-1]+c[i-1][j])%mod;
    }
    for(int i=1;i<=n;i++)
    {
        int ans=0;
        for(int j=0,tmp=1;j<=k;j++,tmp=1ll*tmp*s[i+1]%mod)
            ans=(ans+1ll*((j&1)?mod-1:1)*c[k][j]%mod*tmp%mod*ss[i][k-j]%mod)%mod;
        printf("%d ",ans);
    }
    return 0;
}