超能粒子炮 · 改

超能粒子炮 · 改

t组询问,求(sum_{i=0}^kC_n^imod 2333),(tleq 10^5,n,kleq 10^{18})。

解

不难得知2333为质数，考虑lucus定理，设yyb为模数，于是我们有

[ans=sum_{i=0}^kC_{n/yyb}^{i/yyb}C_{n\%yyb}^{i\%yyb} ]

注意到整除形式，考虑整除分块，对单个整除的值考虑

[ans=C_{n/yyb}^{0}sum_{i=0}^{yyb-1}C_{n\%yyb}^i+C_{n/yyb}^{1}sum_{i=0}^{yyb-1}C_{n\%yyb}^i+...+ ]

[C_{n/yyb}^{k/yyb-1}sum_{i=0}^{yyb-1}C_{n\%yyb}^i+C_{n/yyb}^{k/yyb}sum_{i=0}^{k\%yyb}C_{n\%yyb}^i= ]

[sum_{i=0}^{k/yyb-1}C_{n/yyb}^i*sum_{i=0}^{yyb-1}C_{n\%yyb}^i+C_{n/yyb}^{k/yyb}sum_{i=0}^{k\%yyb}C_{n\%yyb}^i ]

于是设(f[n][m])表示(sum_{i=0}^nC_m^i),故原式可以被表示为

[ans=f[k/yyb-1][n/yyb]f[yyb-1][n\%yyb]+C_{n/yyb}^{k/yyb}f[k\%yyb][n\%yyb] ]

以此递归处理，额外lucus定理处理式子剩下的组合数。

参考代码：

#include <iostream>
#include <cstdio>
#define il inline
#define ri register
#define ll long long
#define yyb 2333
using namespace std;
int jc[yyb],jv[yyb],tr[yyb][yyb];
il void prepare();
il int pow(int,int),C(int,int),
    lucus(ll,ll),ddc(ll,ll);
int main(){
    int t;ll n,k;
    scanf("%d",&t),prepare();
    while(t--)scanf("%lld%lld",&n,&k),
                  printf("%d
",ddc(n,k));
    return 0;
}
il int ddc(ll n,ll r){
    if(r<0)return 0;if(!n||!r)return 1;
    return (ddc(n/yyb,r/yyb-1)*tr[n%yyb][yyb-1]%yyb
        +lucus(n/yyb,r/yyb)*tr[n%yyb][r%yyb]%yyb)%yyb;
}
il int lucus(ll n,ll r){
    int ans(1);
    while(r)ans=ans*C(n%yyb,r%yyb)%yyb,
                n/=yyb,r/=yyb;return ans;
}
il int pow(int x,int y){
    int ans(1);
    while(y){
        if(y&1)ans=ans*x%yyb;
        x=x*x%yyb,y>>=1;
    }return ans;
}
il int C(int n,int r){
    if(n<r)return 0;
    return jc[n]*jv[r]%yyb*jv[n-r]%yyb;
}
il void prepare(){
    int i,j;
    for(i=jc[0]=jv[0]=1;i<yyb;++i)
        jc[i]=jc[i-1]*i%yyb;
    --i,jv[i]=pow(jc[i],yyb-2);
    while(i>1)jv[i-1]=jv[i]*i%yyb,--i;
    for(i=0;i<yyb;++i)
        for(j=tr[i][0]=1;j<yyb;++j)
            tr[i][j]=(tr[i][j-1]+C(i,j))%yyb;
}