POJ Round Numbers(数位DP)

题目大意：

Round Number: 将一个整数转化为二进制数字后，（不含前导0） 要是0的个数 大于等于1的个数 则是 Round Number

问从L-R之中有多少个Round Number

题目分析：

要转化为2进制数字，我们用10进制保存明显不好判断0和1的个数，所以选择用8进制来存储，这样的话每一次进位会多出 ”000“， 然后再加上8进制的尾数， 最后我们可以得出增加了几个 1 和 增加了 几个 0

需要注意的一点就是， 当前面的数字小于 8 的时候 我们要对 前导 0 进行特殊判断

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
typedef __int64 LL;
LL dp[20][163][133];//dp[位数][0的个数][1的个数]
int bit[50];
int binary[8][2] = { {3,0}, {2,1},{2,1},{1,2},{2,1},{1,2},{1,2},{0,3} };
LL dfs(int pos,int preCou0,int preCou1,int flag,int len)
{
if(pos == -1)
{
return  (preCou1 || preCou0) && preCou0 >= preCou1;//这里要对 0 进行特殊判断， 将0去掉
}

if( !flag && dp[pos][preCou0][preCou1] != -1)
return dp[pos][preCou0][preCou1];

LL ans = 0;
int end = flag?bit[pos]:7;

for(int i=0; i<= end; i++)
{
int nowCou0 = preCou0 + binary[i][0];
int nowCou1 = preCou1 + binary[i][1];
if(preCou0 == 0 && preCou1 == 0)//判断是否是第一位，若是第一位则需要进行特殊处理
{
if(i == 0 || i == 1)nowCou0 = 0;
if(i == 2)nowCou0 = 1;
if(i == 3)nowCou0 = 0;
}

ans += dfs(pos-1, nowCou0, nowCou1, flag && i == end, len);
}
if(!flag)
dp[pos][preCou0][preCou1] = ans;

return ans;
}

LL solve(LL n)
{
int len = 0;

while(n)
{
bit[len++] = n%8;
n /= 8;
}

return dfs(len-1, 0, 0, 1, len-1);
}

int main()
{
LL a, b;

memset(dp, -1 ,sizeof(dp));
while(scanf("%I64d%I64d", &a, &b) != EOF)
{
//    printf("%I64d\n", solve(a));
//    printf("%I64d\n", solve(b));
printf("%I64d\n", solve(b) - solve(a-1) );
}
return 0;
}

/*

0的个数 大于等于 1的个数
1    0001
2    0010   1
3    0011
4    0100   1
5    0101
6    0110
7    0111
8    1000   1
9    1001   1
10   1010   1
11   1011
12   1100   1
13   1101
14   1110
15   1111
16   10000  1

*/