1. 程式人生 > >【[SHOI2015]超能粒子炮·改】

【[SHOI2015]超能粒子炮·改】

就是運用\(Lucas\)推一個柿子

首先是前置芝士\(Lucas\)定理

\[C_{n}^{m}\%p=C_{n/p}^{m/p}*C_{n\%p}^{m\%p}\%p\]

至於證明

我建議去問一下Lucas本人

至於這道題,我們要求的是這個柿子

\[\sum_{i=0}^kC_{n}^i\%p\]

於是我們設\(f(n,k)=\sum_{i=0}^kC_{n}^i\)

我們就可以化柿子啦

\[f(n,k)=\sum_{i=0}^kC_{n}^i\]

\[\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }\text{ }=\sum_{i=0}^kC_{n/p}^{i/p}*C_{n\%p}^{i\%p}\]

這個東西一看就很熟悉,\(n/p\)啊,顯然跟整除分塊差不多啊

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

前面有\(0\)\(k/p-1\)這些個整塊,於是我們可以將\(\sum_{i=0}^{p-1}C_{n\%p}^i\)提出來

變成

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

那這個東西豈不是可以寫成

\[f(n\%p,p-1)*f(n/p,k/p-1)\]

在加上那個不完整的塊

\(\sum_{i=0}^{k\%p}C_{n\%p}^i\)可以寫成\(f(n\%p,k\%p)\)

於是就有

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

由於\(n\%p\)還有\(k\%p\)都小於\(2333\),所以\(f(n\%p,p-1)\)還有\(f(n\%p,k\%p)\)可以直接預處理好可以直接求出來

至於那個\(C_{n/p}^{k/p}\)就直接上\(Lucas\)好了

時間複雜度\(O(p^2+Tlog_{2333}^2n)\)

程式碼

非常sb的把\(C_0^0\)

當成\(0\)WA了好幾發

#include<iostream>
#include<cstring>
#include<cstdio>
#define re register
#define LL long long
#define maxn 2335
const int P=2333;
LL c[maxn+2][maxn+2];
LL f[maxn+2][maxn+2];
inline LL Lucas(LL n,LL m)
{
    if(!m) return 1;
    if(n==m) return 1;
    if(n<m) return 0;
    return c[n%P][m%P]*Lucas(n/P,m/P)%P;
}
inline LL F(LL n,LL k)
{
    if(k<0) return 0;
    if(!n) return 1;
    if(!k) return 1;
    if(n<P&&k<P) return f[n][k];
    return (F(n/P,k/P-1)*f[n%P][P-1]%P+Lucas(n/P,k/P)*f[n%P][k%P]%P)%P;
}
int main()
{
    int T;
    scanf("%d",&T);
    c[0][0]=1;
    for(re int i=1;i<=maxn;i++) 
        c[i][i]=c[i][0]=1;
    for(re int i=1;i<=maxn;i++)
        for(re int j=1;j<i;j++)
            c[i][j]=(c[i-1][j]+c[i-1][j-1])%P;
    f[0][0]=1;
    for(re int i=1;i<=maxn;i++) 
        f[i][0]=1;
    for(re int i=0;i<=maxn;i++)
        for(re int j=1;j<=maxn;j++)
            f[i][j]=(c[i][j]+f[i][j-1])%P;
    LL n,k;
    while(T--)
    {
        scanf("%lld%lld",&n,&k);
        printf("%lld\n",F(n,k));
    }
    return 0;
}