如果直接使用暴力的做法,从1-n直接去遍历,然后再去做判断。过不了全部的数据。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll n,k,ans=0;
bool judge(ll op)
{
int now=0;
while(op>=1)
{
if(now>k)return false;
if(op%2==1)
{
now++;
}
op/=2;
}
if(now==k)
{
return true;
}
return false;
}
int main()
{
scanf("%d %d",&n,&k);
for(ll i=1;i<=n;i++)
{
if(judge(i))
{
ans++;
}
}
printf("%d\n",ans);
return 0;
}
看了看大佬的题解才发现这个是数位dp的问题,但是对于数位dp是什么还没有了解过,所以先去了解一下什么是数位dp。
数位dp的特性:需要求解的内容与数的组成有关而与数的大小无关。
//第一步:把数按位拆分
//第二部:进入记忆化搜索
//状态数组的设定:
//对于此题:dp[i][j][k] 代表的是第i个数 此时已经有了1的个数 此时的数是0还是1.
//搜索的内容:
在搜索当前的位置时,如果当前位取得了最高位,那么他的下一位的最大值就是它本身
如果当前位没有取得最大值,那么他的下一位可以取得 0,1随便取。当它的下一位取零的时候,没有达到最高位。他的下下一位也可以任意取,当它的下一位取1的时候,也达到了最高位,他的下下一位的最大值就是下下一位本身。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll n,k;
ll dp[110][100][2];
int data[110];
ll dfs(int len,int sum,bool flag)
{
if(len==0)return sum==k;
//if(sum>k)return 0;
if(dp[len][sum][flag]!=-1)return dp[len][sum][flag];
int max1=flag?data[len]:1; //这个是取当前能取到的最大值
ll ans=0;
for(int i=0;i<=max1;i++)
{
ans+=dfs(len-1,sum+(i==1),flag&(i==max1));
}
return dp[len][sum][flag]=ans;
}
ll chaifen(ll nn)
{
int len=0;
while(nn>=1)
{
data[++len]=nn%2;
nn/=2;
}
memset(dp,-1,sizeof(dp));
return dfs(len,0,1);
}
int main()
{
scanf("%lld %lld",&n,&k);
cout<<1ll*chaifen(n)<<endl;
return 0;
}