CodeForces-431D Random Task（二分答案+数位DP）

最新推荐文章于 2022-02-12 18:06:08 发布

原创最新推荐文章于 2022-02-12 18:06:08 发布 · 422 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

题目专栏收录该内容

140 篇文章

订阅专栏

探讨了在[n+1,2n]范围内，寻找恰好有k个1的二进制数，使得这类数的数量等于m的问题。通过证明单调性，采用二分查找结合数位动态规划的方法求解。

题意

给定 $m$ 和 $k$ ，求一个 $n$ 使得 $[n + 1, 2 n]$ 范围内的数中二进制恰好有 $k$ 个 $1$ 的数，恰有 $m$ 个。
$\leq m \leq 10^{18}$
$\leq k \leq 64$

思路

有一个“显然”的单调性， $n$ 越大， $[n + 1, 2 n]$ 中的数含有 $k$ 个 $1$ 的数单调不减，无论 $k$ 的取值。
其实也不难证，当 $n - 1$ 变为 $n$ 时，区间从 $[n, 2 n - 2]$ 挪到 $[n + 1, 2 n]$ 时，我们失去了 $n$ ，获得了 $2 n - 1, 2 n$ ，而 $2 n$ 和 $n$ 的二进制含有的 $1$ 数相同，而多的那个 $2 n - 1$ 又不会让含有 $k$ 个 $1$ 的数变少，由此得出单调性。
那就直接二分 $n$ ，用数位 $dp\text{dp}$ 验证答案即可。

代码

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define FOR(i,x,y) for(int i=(x),i##END=(y);i<=(y);++i)
#define DOR(i,x,y) for(int i=(x),i##END=(y);i>=(y);--i)
typedef long long LL;
using namespace std;
LL dp[70][70],m;
int num[70],K;

LL dfs(int k,int status,bool ismax)
{
	if(k==0)return status==K;
	if(~dp[k][status]&&!ismax)return dp[k][status];
	int maxer=ismax?num[k]:1;LL res=0;
	FOR(i,0,maxer)res+=dfs(k-1,status+i,ismax&&i==maxer);
	if(!ismax)dp[k][status]=res;
	return res;
}

LL solve(LL k)
{
	int n=0;
	while(k)
	{
		num[++n]=k%2;
		k>>=1;
	}
	return dfs(n,0,1);
}

int main()
{
	scanf("%lld%d",&m,&K);
	memset(dp,-1,sizeof(dp));
	LL L=1,R=1e18;
	while(L<R)
	{
		LL mid=L+R>>1;
		if(solve(mid*2)-solve(mid)>=m)
		{
			R=mid;
		}
		else L=mid+1;
	}
	printf("%lld\n",L);
    return 0;
}