Codeforces Round #239 (Div. 1) 二项式差分

C - Curious Array

思路：对于区间[l, r]每个数加上C(i - l + k, k)， 可以在l处+1， 在r+1处-1， 然后做k+1次求前缀和操作，然后就可以写啦。

然后逐层求前缀和的时候r+1上都要减去一个组合数。

#include<bits/stdc++.h>
#define LL long long
#define fi first
#define se second
#define mk make_pair
#define PII pair<int, int>
using namespace std;

const int N = 2e5 + 7;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const int mod = 1e9 + 7;

int n, m;
LL a[N], b[N][107], inv[N], f[N], finv[N];

void init() {
    inv[1] = f[0] = finv[0] = 1;
    for(int i = 2; i < N; i++) inv[i]=(mod-mod/i)*inv[mod%i]%mod;
    for(int i = 1; i < N; i++) f[i] = f[i-1]*i%mod;
    for(int i = 1; i < N; i++) finv[i] = finv[i-1]*inv[i]%mod;
}
LL comb(int n, int m) {
    if(n < 0 || n < m) return 0;
    return f[n]*finv[m]%mod*finv[n-m]%mod;
}

inline void add(LL &a, LL b) {
    a += b; if(a >= mod) a -= mod;
}

int main() {
    init();
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i++) scanf("%lld", &a[i]);
    while(m--) {
        int l, r, k;
        scanf("%d%d%d", &l, &r, &k);
        add(b[l][k+1], 1);
        for(int j = k+1; j >= 1; j--)
            add(b[r+1][j], mod-comb(r-l+k+1-j, k+1-j));
    }
    for(int j = 100; j >= 0; j--) {
        LL s = 0;
        for(int i = 1; i <= n; i++) {
            add(s, b[i][j+1]);
            add(b[i][j], s);
        }
    }
    for(int i = 1; i <= n; i++)
        printf("%lld ", (b[i][0]+a[i])%mod);
    puts("");
    return 0;
}

/*
*/