bzoj 4667: 小y的密码

不错的题啊 然而我太菜了orz

想一想 不难发现其实从小到大搜索哪几个数并不会超时
然后就算一下就好了。。。

至于怎么算
就枚举哪个位开始低于n的对应位 剩下的位随便填就用组合数算
如果位数比n少直接算就好了 首位不为0即可

#include<bits/stdc++.h>using namespace std;typedef long long LL;inline int read(){    char ch=getchar(); int x=0,f=1;    while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}    while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0'; ch=getchar();}    return x*f;}int n,k,m,a[11],al,ans,c[11],tot,jc[11];LL d[11];bool check(){    if(!tot || !c[tot])return 0;    LL s=0; int u=c[(tot+1)>>1];    for(int i=1;i<=tot;++i)    {        int g=c[i]-u; LL o= g>=0?d[g]:d[-g];        if(g<0) s+= k&1?-o:o;        else s+=o;    }    if(s>m)return 0;    return 1;}int num[11];void add(int x){    LL s=jc[x];    for(int i=0;i<=9;++i)s/=jc[num[i]];    ans+=s;}void cal(){    for(int i=0;i<=9;++i) num[i]=0;    for(int i=1;i<=tot;++i) num[c[i]]++;    if(tot<al){        for(int j=9;j>0;--j)            if(num[j]>0){                num[j]--;                add(tot-1);                num[j]++;            }        return;    }    for(int i=tot;i>=1;--i){        int o= i==tot?0:-1;        for(int j=a[i]-1;j>o;--j)            if(num[j]>0){                num[j]--;                add(i-1);                num[j]++;            }        if(!num[a[i]])break;        num[a[i]]--; if(i==1)ans++;    }}void dfs(int x){    if(check())cal();    if(tot==al)return;    ++tot;    for(int i=x;i<=9;++i)        c[tot]=i,dfs(i);    --tot;}int main(){    n=read(),k=read(),m=read();    while(n) a[++al]=n%10,n/=10;    jc[0]=1; for(int i=1;i<=al;++i)jc[i]=jc[i-1]*i;    for(int i=0;i<=9;++i){        d[i]=1;        for(int j=1;j<=k;++j)d[i]*=i;    }    dfs(0);    printf("%d\n",ans);    return 0;}