Round Numbers(POJ - 3252)

题目：Round Numbers

题目链接：Round Numbers - POJ 3252 - Virtual Judge (vjudge.net)

题意：

如果一个正整数的二进制表示中，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;
}