bzoj4737: 组合数问题

终于过了肝了一天啊，怎么我最近都在做细节码农题啊

（这种水平NOIP凉凉？？）

luacs大家都可以想到用吧，一开始我的思路是把所有在p以内的%p==0的组合数预处理出来，那C(n/p,m/p)任取，但是好像有重算

仔细思考，其实一个组合数可以拆分成很多个C(x%p,y%p) x<y就有是k的倍数，把这个东西看成p进制，得到推论C(i,j)不是p的倍数当且仅当k进制下i的每一位分别大于等于j

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
const LL mod=1e9+;

int alen,a[],blen,b[];
LL S(LL x){return x%=mod,x*(x+)/%mod;}
LL cal(LL a,LL b)
{
    if(a<b)b=a;
    return (S(a)-S(a-b))%mod;
}
LL f[][][];//第i个位，是否在n的上界边缘，是否在m的上界边缘，每一位i都大于等于j的方案数
int main()
{
    freopen("a.in","r",stdin);
    freopen("a.out","w",stdout);
    int T,K;
    scanf("%d%d",&T,&K);
    while(T--)
    {
        LL n,m,k;
        scanf("%lld%lld",&n,&m);if(n<m)m=n;
        alen=,k=n;while(k>)a[++alen]=k%K,k/=K;
        blen=,k=m;while(k>)b[++blen]=k%K,k/=K;
        for(int i=blen+;i<=alen;i++)b[i]=;

        int len=alen;
        memset(f,,sizeof(f));
        for(int u=;u<=a[len];u++)
            for(int v=;v<=b[len];v++)
            {
                if(u>=v)
                {
                    if(u==a[len]&&v==b[len])f[len][][]++;
                    else if(u==a[len])f[len][][]++;
                    else if(v==b[len])f[len][][]++;
                    else f[len][][]++;
                }
            }
        int c;
        for(int i=len-;i>=;i--)
        {
            c=;
            for(int u=;u<K;u++)
                for(int v=;v<K;v++)
                {
                    if(u>=v)
                    {
                        if(u<a[i])
                        {
                            if(v<b[i])
                            {
                                c++;
                                f[i][][]=(f[i][][]+f[i+][][]+f[i+][][]+f[i+][][]+f[i+][][])%mod;
                            }
                            if(v==b[i])
                            {
                                f[i][][]=(f[i][][]+f[i+][][]+f[i+][][])%mod;
                                f[i][][]=(f[i][][]+f[i+][][]+f[i+][][])%mod;
                            }
                            if(v>b[i])
                            {
                                f[i][][]=(f[i][][]+f[i+][][]+f[i+][][])%mod;
                            }
                        }
                        if(u==a[i])
                        {
                            if(v<b[i])
                            {
                                f[i][][]=(f[i][][]+f[i+][][]+f[i+][][])%mod;
                                f[i][][]=(f[i][][]+f[i+][][]+f[i+][][])%mod;
                            }
                            if(v==b[i])
                            {
                                f[i][][]=(f[i][][]+f[i+][][])%mod;
                                f[i][][]=(f[i][][]+f[i+][][])%mod;
                                f[i][][]=(f[i][][]+f[i+][][])%mod;
                                f[i][][]=(f[i][][]+f[i+][][])%mod;
                            }
                            if(v>b[i])
                            {
                                f[i][][]=(f[i][][]+f[i+][][])%mod;
                                f[i][][]=(f[i][][]+f[i+][][])%mod;
                            }
                        }
                        if(u>a[i])
                        {
                            if(v<b[i])
                            {
                                f[i][][]=(f[i][][]+f[i+][][]+f[i+][][])%mod;
                            }
                            if(v==b[i])
                            {
                                f[i][][]=(f[i][][]+f[i+][][])%mod;
                                f[i][][]=(f[i][][]+f[i+][][])%mod;
                            }
                            if(v>b[i])
                            {
                                f[i][][]=(f[i][][]+f[i+][][])%mod;
                            }
                        }
                    }
                }
        }

        LL ans=cal(n+,m+);
        for(int i=;i<=;i++)
            for(int j=;j<=;j++)
                ans=(ans-f[][i][j])%mod;
        printf("%lld\n",(ans+mod)%mod);
    }
    return ;
}