CF-121c Lucky Permutation(组合数学)

CF-121c Lucky Permutation(组合数学)

题目大意

求1-n的字典序第k大的排列中有多少的幸运数其所在的位置也为幸运数，其中幸运数定义为完全由4和7构成的数

解题思路

首先预处理出所有的幸运数，预处理出小于等于n-13的正整数中有多少的幸运数，k小于$10^9$因此最多数字的最后13位数字发生变化，因此暴力找出后13位数并确定答案即可

AC代码

#include<bits/stdc++.h>
using namespace std;
map<int,bool> mp;
vector<int> v;
vector<int> num;
int fac[20];
const int maxn=1e9+7;
int chg()
{
	int s=1;
	int ans=0;
	for(auto x:num)
	{
		ans=ans+x*s;
		s=s*10;
	}
	return ans;
}
void init()
{
	int n;
	for(int i=1;i<=9;i++)
	{
		for(int j=0;j<(1<<i);j++)
		{
			num.clear();
			for(int k=0;k<i;k++)
			{
				if(j&(1<<k)) num.push_back(4);
				else num.push_back(7);
			}
			n=chg();
			v.push_back(n);
			mp[n]=true;
		}
	}
}

int arr[15];
int main()
{
	int n,k;
	scanf("%d%d",&n,&k);
	init();
	fac[0]=1;
	for(int i=1;i<=12;i++)
	fac[i]=fac[i-1]*i;
	if(n<=12&&fac[n]<k)
	{
		printf("-1\n");
		return 0;
	}
	for(int i=1;i<=13;i++) arr[i]=n-13+i; 
	int s;
	int ans=0;
	sort(v.begin(),v.end());
	for(auto x:v) {if(x<=n-13) ans++;else break;}
	bool flag=0;
	for(int i=12;i>=0;i--)
	{
		int ns=arr[1];
		if(k>fac[i])
		{
			ns=arr[k/fac[i]+(k%fac[i]!=0)];
			for(int j=k/fac[i]+(k%fac[i]!=0);j<=12;j++) arr[j]=arr[j+1];
			k=k%fac[i];
			if(k==0) k=fac[i];
		}
		else for(int j=1;j<=12;j++) arr[j]=arr[j+1];
		if(mp[n-i]&&mp[ns]) ans++;
	}
	printf("%d\n",ans);
}