自己编码。。。不会

原創

2020-02-21 02:05

以前没有做过自己编码的数位dp..涨了点姿势。。发出来慢慢膜

在又一次消灭林登·万的战斗中，指挥官moreD缴获了一个神奇的盒子。盒子异常的坚固，以至于完全无法摧毁，唯一打开的方式是通过盒上的密码锁。经过仔细的调查，研究人员一致认为这个盒子中隐藏了林登·万和他的弟弟林登·图的秘密。然而moreD使用了许多办法，都没能打开这个盒子。最后只好将这个盒子封存在了仓库的底层。
事情并没有结束。moreD之所以没能打开这个盒子，是因为老牌的调查员/邪教徒LCJ隐瞒了它的调查结果。LCJ经过不懈的努力，得出了结论。即：给你一个长度不超过17的由0~9组成的无前导0的字符串S，S中的数字排列组成的无前导零的能被17整除的整数中字典序第K小的那个数就是密码。
尽管解开了密码，然而处于对未知的恐惧，LCJ最终并没有打开盒子。然而另一个资历较浅的调查员/邪教徒，你，YDMan不知通过什么办法得知了上述信息，并得到了S和K。现在你决定要解开这个密码，来取得“终极的智慧”。

输入格式：

一行，一个由0~9组成的字符串S和一个不超过17！的正整数K。

输出格式：

一行，即密码。数据保证有解。

样例输入：

样例一
17 1
样例二
2242223 2

样例输出：

样例一
17
样例二
2242232

数据范围：

对于40%的数据，字符串S长度<=12,K<=2*10^6。对于100%的数据，字符串S长度<=17,K<=17!。

#include
#include
#include
#define fd(i,a,b) for(int i=a;i>=b;--i)
#define fo(i,a,b) for(int i=a;i<=b;++i)
using namespace std;
typedef long long LL;
const int mn=20,mm=20000;
char st[20];
int D[mn],S[mn],n,m=0;
LL f[mn][mn][mm],K;
LL dp(int n,int v,int statue)
{
    if(n==0)return v==0;
    LL&res=f[n][v][statue];
    if(res!=-1)return res;
    res=0;
    fo(i,0,9)
    {
        int ns=statue/D[i]%(S[i]+1);
        if(ns>=S[i])continue;
        res+=dp(n-1,(v*10+i)%17,statue+D[i]);
    }
    return res;
}
int main()
{
// freopen("password.in","r",stdin);
  //  freopen("password.out","w",stdout);
    cin>>st>>K;
    n=strlen(st);
    fo(i,0,n-1)
        ++S[st[i]-48];
    memset(f,255,sizeof f);
    D[0]=1;
    fo(i,1,9)D[i]=D[i-1]*(S[i-1]+1),m+=D[i]*S[i];
    int v1=0,s1=0;
    bool flag=0;
    LL now=0;
    fd(i,n,1)
    {
        fo(j,0,9)
        {
            if((i==n)&&j==0)continue;
            int ns=s1/D[j]%(S[j]+1);
            if(ns>=S[j])continue;
            if(now+dp(i-1,(v1*10+j)%17,s1+D[j])>=K)
            {
                flag=1;
                v1=(v1*10+j)%17,s1+=D[j];
                cout<                 break;
            }
            now+=dp(i-1,(v1*10+j)%17,s1+D[j]);
        }
        if(!flag)
        {
            cout<<-1<             return 0;
        }
    }
    cout<     return 0;
}