题意:
给一个 N ≤ 5000 N \le 5000 N≤5000 的数组 A A A,求该数组的 K ≤ 1000000000 K \le 1000000000 K≤1000000000 次前缀和数组 ( m o d    1000000007 ) (\mod\ 1000000007) (mod 1000000007)。
分析:
考虑数组 i i i 位的数字对 K K K 次前缀和 j ( ≥ i ) j(\ge i) j(≥i) 位的贡献,为 ( K − 1 + j − i j − i ) A i \binom{K-1+j-i}{j-i}A_i (j−iK−1+j−i)Ai。由于 K , m o d ∼ 1 e 9 K, mod \sim1e9 K,mod∼1e9,所以二项式系数不能打阶乘表,也不能Lucas。但是其实 j − i < 5000 j-i \lt 5000 j−i<5000,所以可以考虑暴力计算 ( n m ) = n ( n − 1 ) … ( n − m + 1 ) m ( m − 1 ) … ( 1 ) \binom{n}{m} = \frac{n(n-1)\dots(n-m+1)}{m(m-1)\dots(1)} (mn)=m(m−1)…(1)n(n−1)…(n−m+1) 式子,复杂度 O ( M ) O(M) O(M)。枚举 i , j i, j i,j 复杂度 O ( N 2 ) O(N^2) O(N2), 所以总复杂度 O ( N 2 M ) O(N^2M) O(N2M),不能过。
观察系数 ( K − 1 + j − i j − i ) \binom{K-1+j-i}{j-i} (j−iK−1+j−i), 可以想到当 l e n = j − i len = j-i len=j−i 固定时系数也不变,并且可以得到递推式: ( K − 1 + l e n + 1 l e n + 1 ) = ( K − 1 + l e n + 1 ) l e n + 1 ( K − 1 + l e n l e n ) \binom{K-1+len+1}{len+1} = \frac{(K-1+len+1)}{len+1}\binom{K-1+len}{len} (len+1K−1+len+1)=len+1(K−1+len+1)(lenK−1+len)。所以可以 O ( N ) O(N) O(N) 枚举长度 l e n len len,同时得到系数,然后 O ( N ) O(N) O(N) 计算贡献,总复杂度 O ( N 2 ) O(N^2) O(N2),能过。
代码:
#include <bits/stdc++.h>
using namespace std;
const int N_MAX = 1e5 + 10;
const long long mod = 1e9 + 7;
inline long long power(long long a, long long k = mod-2) {
long long ans = 1;
while (k) {
if (k & 1) ans = ans * a % mod;
a = a * a % mod;
k >>= 1;
}
return ans;
}
int N;
long long arr[N_MAX], out[N_MAX], K;
int main() {
scanf("%d%lld", &N, &K);
K--;
for (int i = 1; i <= N; i++) {
scanf("%lld", arr + i);
}
long long up = 1, down = 1;
for (int len = 1; len <= N; len++) {
if (len > 1) up = up * (K + len - 1) % mod;
if (len > 1) down = down * (len - 1) % mod;
long long invdown = power(down);
long long fix = up * invdown % mod;
for (int l = 1, r = len; r <= N; l++, r++) {
out[r] = (out[r] + arr[l] * fix % mod) % mod;
}
}
for (int i = 1; i <= N; i++) {
printf("%lld\n", out[i]);
}
}