【hihocoder1430】 A Boring Problem

来源:互联网 发布:知途旅行网 编辑:程序博客网 时间:2024/06/06 19:30

大意:
给定n个数字序列S,定义
F(j,i)=(S[j]+S[j+1]++S[i1]+S[i])k,
Ans(i)=F[1,i]+F[2,i]++F[i1,i]+F[i,i].
求出所有的 Ans(i) ,答案对10^9+7取模。 S的每个数都在0~9的范围内。
数据范围: N <= 50000, k <= 100.

Solution

呃……暴力拆开然后合并同次项
具体推导可以看我同学的(懒癌晚期)
http://blog.csdn.net/di4covery/article/details/53968919

#include<stdio.h>#include<cstring>#define mod 1000000007#define N 100050#define K 105#define For(i,a,b) for (int i=a;i<=b;i++)int n,k,C[K][K],ans[N],sum[N],S[N][K],T[N][K];char a[N];inline void PlanB(){    scanf("%d%d",&n,&k);    For(i,0,k) C[i][0]=C[i][i]=1;    For(i,1,k) For(j,1,i) C[i][j]=(C[i-1][j-1]+C[i-1][j]) % mod;    scanf("%s",a+1);    For(i,1,n) sum[i]=(sum[i-1]+a[i]-'0') % mod;    For(i,0,n) S[i][0]=1;    For(i,1,n) For(j,1,k) S[i][j]=(long long)S[i][j-1]*sum[i] % mod;    For(j,0,k) T[0][j]=S[0][j];    For(i,1,n) For(j,0,k) T[i][j]=(T[i-1][j]+S[i][j]) % mod;    memset(ans,0,sizeof(ans));    For(i,1,n) For(j,0,k) ans[i]=(ans[i]+(j&1?-1LL:1LL)*C[k][j]*S[i][k-j]%mod*T[i-1][j]%mod+mod)%mod;    For(i,1,n) printf("%d%c",ans[i],i==n?'\n':' ');}int main(){    int t;scanf("%d",&t);    while (t--) PlanB();}
0 0
原创粉丝点击