https://hihocoder.com/problemset/problem/1430
思路:
要用前缀去推Sigma总公式,比较方便。https://blog.csdn.net/weixin_37517391/article/details/83821752
1 #define IOS ios_base::sync_with_stdio(0); cin.tie(0);
2 #include <cstdio>//sprintf islower isupper
3 #include <cstdlib>//malloc exit strcat itoa system("cls")
4 #include <iostream>//pair
5 #include <fstream>//freopen("C:\\Users\\13606\\Desktop\\草稿.txt","r",stdin);
6 #include <bitset>
7 //#include <map>
8 //#include<unordered_map>
9 #include <vector>
10 #include <stack>
11 #include <set>
12 #include <string.h>//strstr substr
13 #include <string>
14 #include <time.h>//and(((unsigned)time(NULL))); Seed n=rand()%10 - 0~9;
15 #include <cmath>
16 #include <deque>
17 #include <queue>//priority_queue<int, vector<int>, greater<int> > q;//less
18 #include <vector>//emplace_back
19 //#include <math.h>
20 //#include <windows.h>//reverse(a,a+len);// ~ ! ~ ! floor
21 #include <algorithm>//sort + unique : sz=unique(b+1,b+n+1)-(b+1);+nth_element(first, nth, last, compare)
22 using namespace std;//next_permutation(a+1,a+1+n);//prev_permutation
23 #define rint register int
24 #define fo(a,b,c) for(rint a=b;a<=c;++a)
25 #define fr(a,b,c) for(rint a=b;a>=c;--a)
26 #define mem(a,b) memset(a,b,sizeof(a))
27 #define pr printf
28 #define sc scanf
29 #define ls rt<<1
30 #define rs rt<<1|1
31 typedef long long ll;
32 void swapp(int &a,int &b);
33 double fabss(double a);
34 int maxx(int a,int b);
35 int minn(int a,int b);
36 int Del_bit_1(int n);
37 int lowbit(int n);
38 int abss(int a);
39 const double E=2.718281828;
40 const double PI=acos(-1.0);
41 //const ll INF=(1LL<<60);
42 const int inf=(1<<30);
43 const double ESP=1e-9;
44 const int mod=(int)1e9+7;
45 const int N=(int)5e4+10;
46
47 long long C[110][110];
48 void Init() {
49 C[0][0] = 1;
50 for(int i = 1;i <= 100;++i) {
51 C[i][0] = 1;
52 for(int j = 1;j <= i;++j) {
53 C[i][j] = (C[i-1][j-1] + C[i-1][j]) % mod;
54 }
55 }
56 }
57 //======================================================Cnk预处理;
58 ll S[105][N],SS[105][N],ans[N];
59 char s[N];
60
61 void solve()
62 {
63 ll n,k;
64 sc("%lld%lld",&n,&k);
65 sc("%s",s);
66
67 for(int i = 0;i <= n;++i) S[0][i] = 1;
68 for(int i = 1;i <= n;++i) S[1][i] = (s[i-1]-'0') + S[1][i-1] ;
69 for(int i = 2;i <= k;++i)
70 for(int j = 1;j <= n;++j)
71 S[i][j] = S[1][j] * S[i-1][j] % mod;
72
73 SS[0][0] = 1;
74
75 for(int i = 0;i <= k;++i) {
76 for(int j = 1;j <= n;++j)
77 SS[i][j] = (SS[i][j-1] + S[i][j])%mod;
78 }
79
80 for(int i=1;i<=n;++i)
81 {
82 ll sum=0;
83 for(int j=0;j<=k;++j)
84 {
85 ll temp=C[k][j]*S[k-j][i]%mod*SS[j][i-1]%mod;
86 if(j&1)
87 sum=(sum-temp+mod)%mod;
88 else
89 sum=(sum+temp)%mod;
90 }
91 ans[i]=sum;
92 }
93 for(int i=1;i<=n;++i)
94 pr("%lld%c",ans[i],i==n?'\n':' ');
95 }
96
97 int main()
98 {
99 Init();
100 int T;
101 sc("%d",&T);
102 while(T--)solve();
103 return 0;
104 }
105
106 /**************************************************************************************/
107
108 int maxx(int a,int b)
109 {
110 return a>b?a:b;
111 }
112
113 void swapp(int &a,int &b)
114 {
115 a^=b^=a^=b;
116 }
117
118 int lowbit(int n)
119 {
120 return n&(-n);
121 }
122
123 int Del_bit_1(int n)
124 {
125 return n&(n-1);
126 }
127
128 int abss(int a)
129 {
130 return a>0?a:-a;
131 }
132
133 double fabss(double a)
134 {
135 return a>0?a:-a;
136 }
137
138 int minn(int a,int b)
139 {
140 return a<b?a:b;
141 }