题目:Round Numbers
题意:
如果一个正整数的二进制表示中,0的数目不小于 1的数目,它就被称为「圆数」。 例如,9 的二进制表示为1001,其中有2个0与2个1。因此,9 是一个「圆数」。 请你计算,区间[l,r] 中有多少个「圆数」。
思路:
数位dp,先取出2进制上的每一位
四维:
这道题是必须要考虑前导0的影响,所以可以通过1的计数来舍弃前导0的影响
#include <iostream>
#include <bitset>
#include <algorithm>
#include <cstdio>
#include <cstring>
const int N = 35;
int f[N][2][N][N];
int a[N];
int dfs(int pos, int flag, int cnt0, int cnt1, bool limit){
if(!pos) return cnt0 >= cnt1;
if(!limit && ~f[pos][flag][cnt0][cnt1]) return f[pos][flag][cnt0][cnt1];
int ans = 0, end = limit ? a[pos] : 1;
for(int i = 0; i <= end; i++)
ans += dfs(pos - 1, i, cnt0 + (cnt1 ? (i ^ 1) : 0), cnt1 + i, limit && i == end);
if(!limit) f[pos][flag][cnt0][cnt1] = ans;
return ans;
}
int dp(int n){
int t = 0;
while(n) a[++t] = n % 2, n >>= 1;
return dfs(t, 0, 0, 0, 1);
}
signed main(){
memset(f, -1, sizeof f);
int l, r;
scanf("%d%d", &l, &r);
printf("%d\n", dp(r) - dp(l - 1));
return 0;
}
三维:
通过first标记是否在第一位以此来排除前导0的影响
#include <iostream>
#include <bitset>
#include <algorithm>
#include <cstdio>
#include <cstring>
const int N = 50;
int f[N][N][N];
int bit[N];
int dfs(int pos, int cnt0, int cnt1, int first, bool limit){
if(!pos) return cnt0 >= cnt1;
if(!limit && f[pos][cnt0][cnt1] != -1) return f[pos][cnt0][cnt1];
int ans = 0, end = limit ? bit[pos] : 1;
for(int i = 0; i <= end; i++){
int j = first && (i == 0);
ans += dfs(pos - 1, j ? 0 : cnt0 + (i == 0), j ? 0 : cnt1 + (i == 1), j, limit && i == end);
}
if(!limit) f[pos][cnt0][cnt1] = ans;
return ans;
}
int dp(int n){
int t = 0;
while(n) bit[++t] = n % 2, n >>= 1;
return dfs(t, 0, 0, 1, 1);
}
signed main(){
memset(f, -1, sizeof f);
int l, r;
scanf("%d%d", &l, &r);
printf("%d\n", dp(r) - dp(l - 1));
return 0;
}