【bzoj4591】超能粒子炮·改

Portal-->bzoj4591

Solution

　　首先这个模数是一个质数

　　然后看一下那个\(k\)和\(n\)的范围。。行吧Lucas定理咯

　　但是如果直接求：
\[ \sum\limits_{i=0}^{k}\binom n i \]
　　那。。稳稳的T啊。。。所以要化一下式子，我们令\(k=ap+b\)：
\[ \begin{aligned} \sum\limits_{i=0}^{k}\binom n i&\equiv \sum\limits_{i=0}^k \binom {i/p} {n/p}\binom {i\% p}{n\%p}(mod\ p)\\ &\equiv \sum\limits_{i=0}^{ap-1}\binom {i/p} {n/p}\binom {i\% p}{n\%p}+\sum\limits_{i=ap}^{ap+b}\binom {i/p} {n/p}\binom {i\% p}{n\%p}(mod\ p)\\ &\equiv \sum\limits_{i=0}^{a-1}\binom {i} {n/p}\sum\limits_{i=0}^{p-1}\binom {i}{n\%p}+\binom a {n/p}\sum\limits_{i=0}^b\binom {i}{n\%p} \end{aligned} \]
　　然后因为\(p\)比较小（只有\(2333\)真是2333）

　　所以我们可以直接暴力处理出\(n,m<=2333\)的\(\binom n m\)的的前缀和

　　然后对于范围内的直接调用，范围外的用上面那个式子递归处理就好了

　　

　　代码大概长这个样子：

#include<iostream>
#include<cstdio>
#include<cstring>
#define ll long long
using namespace std;
const int MOD=2333;
ll c[MOD+10][MOD+10],sum[MOD+10][MOD+10];
ll n,k,T,ans;
void prework(int n);
ll Lucas(ll n,ll m);
ll Min(ll x,ll y){return x<y?x:y;}
ll f(ll n,ll k);

int main(){
#ifndef ONLINE_JUDGE
    freopen("a.in","r",stdin);
#endif
    ll a,b;
    scanf("%lld",&T);
    prework(MOD);
    for (int o=1;o<=T;++o){
        scanf("%lld%lld",&n,&k);
        printf("%lld\n",f(n,k));
    }
}

void prework(int n){
    c[0][0]=1;
    for (int i=1;i<=n;++i){
        c[i][0]=1; c[i][i]=1;
        for (int j=1;j<i;++j)
            c[i][j]=(c[i-1][j-1]+c[i-1][j])%MOD;
    }
    for (int i=0;i<=n;++i){
        sum[i][0]=c[i][0];
        for (int j=1;j<=n;++j)
            sum[i][j]=(sum[i][j-1]+c[i][j])%MOD;
    }
}

ll Lucas(ll n,ll m){
    if (n<m) return 0;
    if (n<MOD&&m<MOD) return c[n][m];
    return c[n%MOD][m%MOD]*Lucas(n/MOD,m/MOD)%MOD;
}

ll f(ll n,ll k){
    if (k<0) return 0;
    if (n<MOD&&k<MOD) return sum[n][k];
    return (f(n/MOD,min(k/MOD-1,n/MOD))*sum[n%MOD][MOD-1]%MOD+Lucas(n/MOD,k/MOD)*sum[n%MOD][k%MOD]%MOD)%MOD;
}