题目链接
把每次求前缀和操作在纸上画一下就会发现每个位置的值就等于这次的上一个位置的值加上上次的这个位置的值之和,就可以想到和机器人走方格有关,找规律之后我们发现求k次前缀和之后
a
i
a_i
ai对位置
j
(
j
>
=
i
)
j (j>=i)
j(j>=i)的贡献的个数为下式,因为j-i最多n种,预处理逆元递推计算出这n个值就完事了。
C
k
−
1
+
(
j
−
i
)
j
−
i
=
∏
p
=
k
k
−
1
+
j
−
i
p
(
j
−
i
)
!
C_{k - 1 + (j - i)}^{j - i}=\frac{\prod_{p=k}^{k-1+j-i}p}{(j-i)!}
Ck−1+(j−i)j−i=(j−i)!∏p=kk−1+j−ip
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 5e3 + 5, mod = 1e9 + 7;
ll inv[maxn], f[maxn], a[maxn], ans[maxn];
int main()
{
int n, k;
cin >> n >> k;
for (int i = 1; i <= n; ++i) cin >> a[i];
inv[1] = 1;
for (int i = 2; i <= n; ++i) inv[i] = (ll)(mod - mod / i) * inv[mod % i] % mod;
f[0] = 1;
for (int i = 1; i <= n; ++i) f[i] = f[i - 1] * (k + i - 1) % mod * inv[i] % mod;
for (int i = 1; i <= n; ++i){
for (int j = i; j <= n; ++j){
ans[j] += f[j - i] * a[i] % mod;
ans[j] %= mod;
}
}
for (int i = 1; i <= n; ++i) cout << ans[i] << '\n';
return 0;
}