CodeForces-431D Random Task(二分答案+數位)
阿新 • • 發佈:2018-12-12
題意
給定 和 ,求一個 使得 範圍內的數中二進位制恰好有 個 的數,恰有 個。
思路
有一個“顯然”的單調性, 越大, 中的數含有 個 的數單調不減,無論 的取值。 其實也不難證,當 變為 時,區間從 挪到 時,我們失去了 ,獲得了 ,而 和 的二進位制含有的 數相同,而多的那個 又不會讓含有 個 的數變少,由此得出單調性。 那就直接二分 ,用數位驗證答案即可。
程式碼
#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;
}