神奇的(逆)康托展开:求1到n的全排列中字典序第k小的排列
$k<=10^9<13!$,显然$k$最多只会影响后$13$位
前面一大串都是有序从小到大排列的,于是搞个数位dp
后面一小串用逆康托展开求出原串,枚举是否符合条件。
#include<iostream> #include<cstdio> #include<cstring> #include<vector> #include<algorithm> using namespace std; long long fac[14]={1,1,2,6,24,120,720,5040,40320,362880,3628800,39916800,479001600,6227020800}; int n,k,x,e[12],a[12]; vector <int> p1,p2; int dfs(int d,int w,int z){//普通的数位dp if(!d) return !z; if(!w&&!z&&e[d]>-1) return e[d]; int lim=w?a[d]:9,tot=0; for(int i=0;i<=lim;++i) if(i==4||i==7||(z&&!i)) tot+=dfs(d-1,w&&(i==lim),z&&!i); if(!w&&!z) e[d]=tot; return tot; } int solve1(int A){ int t=0; while(A) a[++t]=A%10,A/=10; return dfs(t,1,1); } bool is(int x){ for(;x;x/=10) if(x%10!=4&&x%10!=7) return 0; return 1; } int solve2(){ --k; for(int i=n-x+1;i<=n;++i) p1.push_back(i); for(int i=x,v;i>=1;--i){//逆康托展开求原串 v=k/fac[i-1]; k%=fac[i-1]; sort(p1.begin(),p1.end()); p2.push_back(p1[v]); p1.erase(p1.begin()+v); }int tot=0; for(int i=0;i<x;++i) if(is(n-x+i+1)&&is(p2[i])) ++tot; return tot; } int main(){ scanf("%d%d",&n,&k); if(n<13&&k>fac[n]){printf("-1");return 0;} while(k>fac[x])++x; memset(e,-1,sizeof(e)); printf("%d",solve1(n-x)+solve2()); return 0; }
原文地址:https://www.cnblogs.com/kafuuchino/p/10338589.html
时间: 2024-11-05 22:55:10