牛客网多校第九场 H Prefix Sum【分块】

题目链接

题意： a[0] a [ 0 ] ~ a[k] a [ k ] ，一开始 a[0] a [ 0 ] 全为 0 0 ，且无论何时 a[i][1]=a[i−1][1](i>=1) a [ i ] [ 1 ] = a [ i − 1 ] [ 1 ] ( i >= 1 ) ， a[i][j]=a[i][j−1]+a[i−1][j](j>=2) a [ i ] [ j ] = a [ i ] [ j − 1 ] + a [ i − 1 ] [ j ] ( j >= 2 ) 。有两种操作，第一种是 a[0][x]+=y a [ 0 ] [ x ] + = y ，第二种是查询 a[k][x] a [ k ] [ x ] 的值。

优雅的分块暴力~

#include 
using namespace std;
typedef long long ll;
const ll mod=1000000007;

int n,m,k;

ll fac[200000+10];
ll inv[200000+10];

void init(){
    fac[0]=fac[1]=1;
    for (int i=2;i<=200001;i++) fac[i]=fac[i-1]*i%mod;
    inv[0]=inv[1]=1;
    for (int i=2;i<=200001;i++) inv[i]=inv[mod%i]*(mod-mod/i)%mod;
    for (int i=2;i<=200001;i++) inv[i]=inv[i]*inv[i-1]%mod; 
}

ll C(ll n,ll m){
    ll ans=fac[n];
    ans=ans*inv[m]%mod;
    ans=ans*inv[n-m]%mod;
    return ans;
}

int unit=2000;
int x[2000+10],y[2000+10];
int top=0;
int p;
ll a[50][100000+10];
int op;

int main(){
    init();
    scanf("%d %d %d",&n,&m,&k);
    for (int i=1;i<=m;i++){
        scanf("%d",&op);
        if (op==0){
            ++top;
            scanf("%d %d",&x[top],&y[top]);
        }
        else{
            scanf("%d",&p);
            ll ans=a[k][p];
            for (int i=1;i<=top;i++){
                if (x[i]>p) continue;
                ans+=y[i]*C(k+p-x[i]-1,p-x[i])%mod;
                ans=(ans+mod)%mod;
            }
            printf("%lld\n",ans);
        }
        if (i%unit==0){
            for (int i=1;i<=top;i++){
                a[0][x[i]]+=y[i];
                a[0][x[i]]=(a[0][x[i]]+mod)%mod;
            }
            top=0;
            for (int i=1;i<=k;i++){
                a[i][1]=a[i-1][1];
                for (int j=2;j<=n;j++){
                    a[i][j]=(a[i][j-1]+a[i-1][j]+mod)%mod;
                }
            }
        }
    }
}