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

题目链接

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

优雅的分块暴力~

#include <bits/stdc++.h>
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;
                }
            }
        }
    }
}