[51nod - 1161] Partial Sums

题意：

给一个 $N\le 5000$ 的数组 $A$，求该数组的 $K\le 1000000000$ 次前缀和数组 $(\mathrm{m}\mathrm{o}\mathrm{d}1000000007)$。

分析：

考虑数组 $i$ 位的数字对 $K$ 次前缀和 $j(\ge i)$ 位的贡献，为 $\left(\genfrac{}{}{0ex}{}{K-1+j-i}{j-i}\right)A_i$。由于 $K,mod\sim 1e9$，所以二项式系数不能打阶乘表，也不能Lucas。但是其实 $j-i<5000$，所以可以考虑暴力计算 $\left(\genfrac{}{}{0ex}{}{n}{m}\right)=\frac{n(n-1)\dots (n-m+1)}{m(m-1)\dots (1)}$ 式子，复杂度 $O(M)$。枚举 $i,j$ 复杂度 $O(N^2)$, 所以总复杂度 $O(N^{2}M)$，不能过。

观察系数 $\left(\genfrac{}{}{0ex}{}{K-1+j-i}{j-i}\right)$, 可以想到当 $len=j-i$ 固定时系数也不变，并且可以得到递推式：$\left(\genfrac{}{}{0ex}{}{K-1+len+1}{len+1}\right)=\frac{(K-1+len+1)}{len+1}\left(\genfrac{}{}{0ex}{}{K-1+len}{len}\right)$。所以可以 $O(N)$ 枚举长度 $len$，同时得到系数，然后 $O(N)$ 计算贡献，总复杂度 $O(N^2)$，能过。

代码：

#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]);
  }
}