[SHOI2015] 超能粒子炮·改

题目链接:戳我

(f[n][k]=sum_{i=0}^kC_n^ipmod {2333})
那么根据卢卡斯定理我们知道(f[n][k]=sum_{i=0}^kC_{n/p}^{i/p} imes C_{n\%p}^{i\%p})

(=C_{n/p}^0 imes sum_{i=0}^{p-1}C_{n\%p}^i+C_{n/p}^1 imes sum_{i=0}^{p-1}C_{n\%p}^i+...+C_{n/p}^{k/p-1} imes sum_{i=0}^{p-1}C_{n\%p}^i+C_{n/p}^{k/p} imes sum_{i=0}^{k\%p}C_{n\%p}^i)

(=sum_{i=0}^{p-1}C_{n\%p}^i imes(C_{n/p}^0+C_{n/p}^1+...+C_{n/p}^{k/p-1})+sum_{i=0}^{k\%p}C_{n\%p}^i)

(=sum_{i=0}^{p-1}C_{n\%p}^i imes sum_{j=1}^{k/p-1}C_{n/p}^j+sum_{i=0}^{k\%p}C_{n\%p}^i)

(=f[n\%p][p-1] imes f[n/p][k/p-1]+C_{n/p}^{k/p} imes f[n\%p][k\%p])

取模的那些我们可以先预处理出来(小于等于2333),然后除法操作的那些我们递归计算即可。

代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define ll long long
#define p 2333
using namespace std;
int T;
ll n,k;
ll c[2500][2500],f[2500][2500];
inline void init()
{
    c[0][0]=1;
	for(int i=1;i<=2333;i++)
        c[i][0]=c[i][i]=1;	
    for(int i=1;i<=2333;i++)
        for(int j=1;j<i;j++)
            c[i][j]=(c[i-1][j-1]+c[i-1][j])%p;
}
inline ll lucas(ll down,ll up)
{
    if(down==up||up==0) return 1;
    if(down<up) return 0;
    return lucas(down/p,up/p)*c[down%p][up%p]%p;
}
inline ll F(ll down,ll up)
{
    if(!down||!up) return 1;
    if(up<0||down<0) return 0;
    if(down<2333&&up<2333) return f[down][up];
    return (f[down%p][p-1]*F(down/p,up/p-1)%p+lucas(down/p,up/p)*f[down%p][up%p]%p)%p;
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("ce.in","r",stdin);
    #endif
    scanf("%d",&T);
    init();
    f[0][0]=1;
    for(int i=1;i<=2333;i++) f[i][0]=1;
    for(int i=0;i<=2333;i++) 
        for(int j=1;j<=2333;j++)
            f[i][j]=(f[i][j-1]+c[i][j])%p;
    while(T--)
    {
        scanf("%lld%lld",&n,&k);
        printf("%lld
",F(n,k));
    }
    return 0;
}