CF431D Random Task

一、题目

点此看题

二、解法

首先证明$n$越大，这样的数越多。当$n$增加$1$时，范围变为$[n+2,2n+2]$，相比于原来$[n+1,2n]$增加了$2n+1,2n+2$，减少了$n+1$，而$2n+2$和$n+1$一的个数相同，所以具有单调性。

然后就可以二分啦，检查就写一个二进制情况下的数位$dp$就行了。

#include <cstdio>
#include <cstring>
#define int long long
const int M = 70;
int read()
{
	int x=0,f=1;char c;
	while((c=getchar())<'0' || c>'9') {if(c=='-') f=-1;}
	while(c>='0' && c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();}
	return x*f;
}
int n,m,k,ans,a[M],dp[M][M];
int dfs(int x,int r,int up)
{
	if(!x) return r==k;
	if(dp[x][r]!=-1 && !up) return dp[x][r];
	int ans=0;
	for(int i=0;i<=1;i++)
	{
		if(up && i>a[x]) break;
		ans+=dfs(x-1,r+(i==1),up&&(i==a[x]));
	}
	if(!up) dp[x][r]=ans;
	return ans;
}
int cal(int x)
{
	n=0;
	while(x) a[++n]=x%2,x/=2;
	return dfs(n,0,1);
}
void dich(int l,int r)
{
	if(l>r) return ;
	int mid=(l+r)>>1,t=cal(mid*2)-cal(mid);
	if(t>=m)
	{
		ans=mid;
		dich(l,mid-1);
	}
	else dich(mid+1,r);
}
signed main()
{
	memset(dp,-1,sizeof dp);
	m=read();k=read();
	dich(1,1e18);
	printf("%lld\n",ans);
}