题目
Description
在又一次消灭林登·万的战斗中,指挥官moreD缴获了一个神奇的盒子。盒子异常的坚固,以至于完全无法摧毁,唯一打开的方式是通过盒上的密码锁。
经过仔细的调查,研究人员一致认为这个盒子中隐藏了林登·万和他的弟弟林登·图的秘密。然而moreD使用了许多办法,都没能打开这个盒子。最后只好将这个盒子封存在了仓库的底层。
事情并没有结束。moreD之所以没能打开这个盒子,是因为老牌的调查员/邪教徒LCJ隐瞒了它的调查结果。LCJ经过不懈的努力,得出了结论。即:给你一个长度不超过17的由0~9组成的无前导0的字符串S,S中的数字排列组成的无前导零的能被17整除的整数中字典序第K小的那个数就是密码。
尽管解开了密码,然而处于对未知的恐惧,LCJ最终并没有打开盒子。然而另一个资历较浅的调查员/邪教徒,你,YDMan不知通过什么办法得知了上述信息,并得到了S和K。现在你决定要解开这个密码,来取得“终极的智慧”。
Input
一行,一个由0~9组成的字符串S和一个不超过17!的正整数K。
Output
一行,即密码。数据保证有解。
Sample Input
输入1:
17 1
输入2:
2242223 2
Sample Output
输出1:
17
输出2:
2242232
分析
对于40%的数据,按字典序从小到大搜索原串的每一个排列。同时开一个数组,G[n][S][k],表示前n位使用的数字集合为S,对17取模的结果为k的情况下,是否有合法的数字能被17整除。当搜索一个出现过的状态,就可以直接用f判断是否有解,来减去大部分无用状态。
对于100%的数据,使用数位DP。F[n][S][k],表示前n位使用的数字集合为S,对17取模的结果为k的情况下,后续的位置数字排列的合法方案数。转移可以用记忆化搜索实现。求解的过程只要从高位开始从小到大枚举答案的每一位,判断是否有足够数目的解,即可。
程序
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
long long f[1<<17][17],a[18],fac[18],b[18],c[18],K;
using namespace std;
int main(){
char ch[20];
scanf("%s",ch);
scanf("%lld",&K);
int l=strlen(ch);
for (int i=1;i<=l;i++)
a[i]=ch[i-1]-'0';
sort(a+1,a+l+1);
f[(1<<l)-1][0]=1;
for (int i=(1<<l)-1;i>0;i--){
int p=1;
for (int j=1;j<=l;j++)
if ((i&(1<<(j-1)))==0) p=(p*10)%17;
for (int j=1;j<=l;j++)
if (i&(1<<(j-1)))
for (int k=0;k<=16;k++)
f[i^(1<<(j-1))][(k+a[j]*p)%17]+=f[i][k];
}
b[0]=1;
for (int i=1;i<=17;i++)
b[i]=b[i-1]*i;
for (int i=(1<<l)-1;i>0;i--){
for (int j=0;j<=9;j++) c[j]=0;
for (int j=1;j<=l;j++)
if ((i&1<<(j-1))==0) c[a[j]]++;
for (int j=0;j<=9;j++)
for (int k=0;k<=16;k++)
f[i][k]/=b[c[j]];
}
int s=0,p=0;
fac[0]=1;
for (int i=1;i<=l;i++)
fac[i]=(fac[i-1]*10)%17;
for (int i=1;i<=l;i++){
int pre=-1;
for (int j=1;j<=l;j++)
if ((s&(1<<(j-1)))==0){
if (a[j]==pre) continue;
pre=a[j];
int k=(17-(p+a[j]*fac[l-i])%17)%17;
if (i==1&&a[j]==0) continue;
if (f[s|1<<(j-1)][k]>=K){
s|=1<<(j-1);
p+=a[j]*fac[l-i];
p%=17;
cout<<a[j];
break;
}
else K-=f[s|1<<(j-1)][k];
}
}
for (int i=1;i<=l;i++)
if ((s&1<<(i-1))==0) cout<<a[i];
return 0;
}