【牛客练习赛13】D 幸运数字Ⅳ 【康拓展开】

链接：https://www.nowcoder.com/acm/contest/70/D
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld
题目描述
定义一个数字为幸运数字当且仅当它的所有数位都是4或者7。
比如说，47、744、4都是幸运数字而5、17、467都不是。
现在想知道在1…n的第k小的排列（permutation，https://en.wikipedia.org/wiki/Permutation）中，有多少个幸运数字所在的位置的序号也是幸运数字。
输入描述:
第一行两个整数n，k。
1 <= n，k <= 1000,000,000
输出描述:
一个数字表示答案。
如果n没有k个排列，输出-1。
示例1
输入

7 4
输出

1
说明

1 2 3 4 6 7 5
示例2
输入

4 7
输出

1
说明

2 1 3 4

分析: 对于有n个数字来说，其所有的序列为n!种。
我们发现题目中的k只有1e9那么大，因为13！就大于1e9了，所以即使n再大，会变的也就是只有后面的13位，前面的都是位置和数字一样。
所以我们可以分为两部分
1》当n小于等于13的时候，因为有可能k大于n!所以特判一下，如果k小于n!，那么我们可以用康托展开逆运算直接得到第k个序列为什么，然后逐一判断就行了。
2》当n大于13的时候，从1到n-13 都是位置和数字一样，只有后面的13位才会变动，一样用康托展开逆运算求就好。

代码。

#include<bits/stdc++.h>
using namespace std;
#define LL long long 

const  int N = 1e5+11;
const int M = 1e6+11;
const int mod = 1e9+7;
const int inf = 0x3f3f3f3f;

LL fac[15];
void init(){
    fac[0]=1;
    for(int i=1;i<15;i++)  fac[i]=fac[i-1]*i;

}

int tmp[15],sz;
void kangto(vector<int>ve,int k){
    sz=1; k--; int len=ve.size();
    for(int i=len-1;i>=0;i--){
        int t=k/fac[i];
        tmp[sz++]=ve[t]; 
        ve.erase(t+ve.begin());
        k-=t*fac[i];
    }
}

bool Judge(int x){
    while(x){
        int t=x%10;
        if(t!=7 && t!=4) return false;
        x/=10;
    }
    return true;
}

vector<LL>S;
void dfs(LL x){
    if(x>fac[14]) return ;
    if(x!=0) S.push_back(x);
    dfs(x*10+4);
    dfs(x*10+7);
}
int main(){
    init(); 
    dfs(0); sort(S.begin(),S.end());

    int n,k; cin>>n>>k;
    if(n<15 && fac[n]<k) puts("-1");
    else {
        if(n<15) {
            vector<int>ve; 
            for(int i=1;i<=n;i++) ve.push_back(i);
            kangto(ve,k);
            int ans=0;
            for(int i=1;i<=n;i++) {
                if(Judge(i) && Judge(tmp[i]))  ans++;
            }
            cout<<ans<<endl;
        } else {
            vector<int>ve; 
            int ans=lower_bound(S.begin(),S.end(),n-14)-S.begin();
            for(int i=n-14+1;i<=n;i++)  ve.push_back(i);
            kangto(ve,k);
            for(int i=1;i<sz;i++) {
                if(Judge(i+n-14) && Judge(tmp[i]))  ans++;
            }
            cout<<ans<<endl;
        }
    } 
return 0;
}