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

D 幸运数字Ⅳ

链接：https://www.nowcoder.com/acm/contest/70/D
来源：牛客网
题目描述
定义一个数字为幸运数字当且仅当它的所有数位都是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

康托展开
康托展开及其逆运算 详解
思路：看了会康拓展开和逆运算发现并不太难且很实用，但是bug却调了好久。。。
把阶乘和幸运数字打表，由于n很大k却不大，所以先剪枝，把-1的情况输出。我是把15作为一个界限分两段，你也可以有其他界限，保证前n-limit个数为自然数排列。由于n很大，数组下标一律用n - i，分类讨论时注意下标（有关数的位置）。

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

const LL MAXN = 1e4 + 10;
LL a[MAXN >> 1], b[MAXN >> 1], dp[MAXN];
LL cnt = 0;
bool vis[MAXN];

inline void dfs(LL x, LL y) {
    if(y == 0) {
        dp[cnt++] = x;
        return ;
    }
    dfs(x * 10 + 4, y - 1);
    dfs(x * 10 + 7, y - 1);
}

inline void init() {
    memset(vis, false, sizeof(vis));
    a[0] = 1;
    for(LL i = 1; ; ++i) {
        a[i] = i * a[i - 1];
        if(a[i] > 1e15) break;
    }
    for(LL k = 1; k <= 10; ++k) {
        dfs(4, k - 1);
        dfs(7, k - 1);
    }
}

bool judge(LL x) {
    while(x) {
        LL ans = x % 10;
        x /= 10;
        if(ans != 4 && ans != 7) {
            return false;
        }
    }
    return true;
}

int main() {
    LL n, k; init();
    scanf("%lld %lld", &n, &k);
    if(n <= 15) {
        if(k > a[n]) {
            puts("-1");
            return 0;
        }
    }
    LL p = 1; k--;
    for(LL i = max(1LL, n - 15); i <= n; ++i) {
        LL ans = k / a[n - i];
        k = k % a[n - i];
        LL t = 0;
        for(LL j = max(1LL, n - 15); j <= n; ++j) {
            if(t == ans && !vis[n - j]) {
                t = j;
                vis[n - j] = 1;
                break;
            }
            if(!vis[n - j]) t++;
        }
        b[p++] = t;
    }
//  for(LL i = 1; i < p; ++i) {
//      printf(" %lld", b[i]);
//  }
//  puts("");
    LL sum = 0;
    if(n > 15) {
        for(LL i = 0; i < cnt; ++i) {
            if(dp[i] < n - 15) {
                sum++;
            }
            else break;
        }
        for(LL i = 1; i < p; ++i) { //卡了好久。。。
//          if(b[i] == 44) printf("%lld\n", i + n - 16);
//          if(b[i] == 47) printf("%lld\n", i + n - 16);
            if(judge(b[i]) && judge(i + n - 16)) {
                sum++;
            }
        }
    }
    else {
        for(LL i = 1; i < p; ++i) {
            if(judge(b[i]) && judge(i)) {
                sum++;
            }
        }
    }
    printf("%lld\n", sum);
    return 0;
}