POJ 3252.Round Numbers

题目:http://poj.org/problem?id=3252

AC代码(C++):

#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <vector>
#include <queue>
#include <math.h>
#include <string>
#include <string.h>
#include <bitset>

#define INF 0xfffffff
#define MAXN 100105

using namespace std;

int n,m;
int c[33][33] = {0};
int bin[35];

void play_table(void){
    for(int i=0;i<=32;i++)
        for(int j=0;j<=i;j++)
            if(!j || i==j)
                c[i][j]=1;
            else
                c[i][j]=c[i-1][j-1]+c[i-1][j];
    return;
}

int f(int x){
	int ans = 0;
	int len = 0;
	for(; x != 0; len++){
		bin[len] = x%2;
		x/=2;
	}
	int zeros = 0;
	int ones = 1;
	for(int i = len-2; i >= 0; i--){
		if(bin[i]==0){
			zeros++;
			continue;
		}
		zeros++;
		for(int j = i; j >= 0; j--){//j zeros
			if(ones+i-j>zeros+j)break;
			ans += c[i][j];
		}
		zeros--;
		ones++;
	}
	for(int i = len-1; i > 0; i--){
		for(int j = i-1; j >= i-j; j--){
			ans += c[i-1][j];
		}
	}
	return ans;
}

int main(){
	play_table();
	cin>>n>>m;
	cout<<f(m+1)-f(n);
}

总结: 这题几个月前做过, 由于一直WA放到现在才重新捡起来做. 首先n到m的Round number数(以下简称RN), 就等于从1到m的RN减去从1到n的RN, 故只需写一个求1到n的RN的函数f就行了. 函数分2部分, 下面逐一讲解:

第一部分, 求与这个数长度一样(二进制长度)但小于这个数的数中, 有多少个RN. 因为第一位一定是1, 所以从第二位开始, 遇到0跳过, 遇到1则看做是把这个1置为0, 然后让这位数往后的数位任意组合并计算RN数, 这样就保证了求得的RN都小于这个数. 至于遇到1时具体怎么求RN数, 将后面数位任意组合, 即求组合数c[i][j]的和, 其中i是往后的位数, j是需要的0的数量, 加一个0数量大于等于1数量的条件就行.

第二部分, 求比这个数长度短的数中有多少RN. 这个很简单, 也是求组合数, 因为第一位一定是1所以只要求长度减一往后的组合数就行.

最后把两部分加上就得到了比这个数小的数中有多少RN. 输出f(m+1)-f(n)即为所求.

前面交了两三次WA一直不知道哪里有问题, 后来把组合数函数改成打表就AC了.