Auction SDOI2010 Code (count based DP)

SDOI2010 Code auction

---

---

\(solution:\)

Updating. . . . . .

---

---

\(code:\)

#include<iostream>
#include<cstdio>
#include<iomanip>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<ctime>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>

#define ll long long
#define db double
#define rg register int

using namespace std;

const int mod=999911659; //答案的模数

ll n;
ll g[505]; //记录余数为i的形如11111的数的个数
int p,vn,ans; //循环节长度，小模数
int a[505]; //记录余数上一次的位置（找循环节用）
int inv[11]; //逆元
int c[505][11]; //组合数
int f[505][505][11]; //计数类DP

inline int qr(){
    register char ch; register bool sign=0; rg res=0;
    while(!isdigit(ch=getchar()))if(ch=='-')sign=1;
    while(isdigit(ch))res=res*10+(ch^48),ch=getchar();
    if(sign)return -res; else return res;
}

int main(){
    //freopen(".in","r",stdin);
    //freopen(".out","w",stdout);
    scanf("%lld%d",&n,&p); inv[1]=1;
    if(n<=p) for(rg i=1;i<=n;++i) vn=(vn*10+1)%p,++g[vn];
    else{ rg tot=1%p,m,l;
        for(rg i=1;i<=p+1;++i){
            if(a[tot]){ l=a[tot]; m=i-l; break;} //找到循环节，计算循环节长度
            a[tot]=i; ++g[tot]; tot=(tot*10+1)%p; //记录当前余数出现位置，继续寻找
        }
        for(rg i=0;i<p;++i)
            if(a[i]&&a[i]>=l){
                g[i]+=(n-a[i])/m%mod; //记录所有余数为i的形如11111的数的个数
                if((a[i]-l+1)%m==(n-l+1)%m) vn=i;
            }
    }
    for(rg i=2;i<9;++i) inv[i]=(ll)(mod-mod/i)*inv[mod%i]%mod; //线性递推求逆元
    for(rg i=0;i<p;++i){ c[i][0]=1; if(!g[i])continue;
        for(rg j=1;j<9;++j,g[i]=(g[i]+1)%mod) //注意最后还有个++g[i]！！！
            c[i][j]=(ll)c[i][j-1]*g[i]%mod*inv[j]%mod; //用性质求i较大,j较小的逆元
    }
    f[0][vn][0]=1; //最开始有一个长度为n的形如1111111111的数
    for(rg i=0;i<p;++i)
        for(rg j=0;j<p;++j)
            for(rg o=0;o<9;++o)
                for(rg k=0;k<9-o;++k) //枚举取出多少个余数为i的形如的11111111的数
                    (f[i+1][(j+k*i)%p][k+o]+=(ll)f[i][j][o]*c[i][k]%mod)%=mod;
    for(rg i=0;i<9;++i) ans=(ans+f[p][0][i])%mod; //计算答案
    printf("%d\n",ans);
    return 0;
}