timus-1507 Amount of Degrees 按位DP

http://acm.timus.ru/problem.aspx?space=1&num=1057

其实这题可以算是一个组合数的题目了,主要是将基于其他进制转化为基于2进制的算法,对于某一位不为1的话,那么取其他位的话是一定不满足题意的,所以要找到一个数的最高位大于1,将这一位以及后面的每一位都赋值为1,然后就是一个按位DP的过程了,dp[len][statu]表示长度剩余量为len,要求1的个数为statu个时,并且对后面的位没有要求的情况下,所有可能的解。

代码如下:

#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
using namespace std;

typedef unsigned int Int;

Int x, y, k, b, bit[35];

int len, dp[35][35];

Int dfs(int pos, int statu, int limit)
{ 
    if (pos == -1) {
        return statu == 0;
    }
    if (!limit && dp[pos][statu] != -1) return dp[pos][statu];
    Int sum = 0;
    int s, end = limit ? bit[pos] : 1;
    for (int i = 0; i <= end; ++i) {
        s = statu;
        if (i == 1) s = statu - 1;
        if (s < 0) continue;
        sum += dfs(pos-1, s, limit && i==end);
    }
    if (!limit) {
        dp[pos][statu] = sum; // 如果有限制的话,那么end的取值是并没有取全的
    }
    return sum;
}

Int Cal(Int x)
{
    len = -1;
    while (x != 0) { 
        bit[++len] = x % b;
        x /= b;
    }
    if (b != 2) {
    // 对于不是2进制的数,我们需要将其转化为2进制进行求解
    // 若一个数被表示为 a1*b^e1 + a2*b^e2 + a3*b^e3 次方,那么这个数要能成为题目中要求的数
    // 就要在把这个数缩小的情况下不对解带来影响,所以我们选择从高位开始的第一个系数既
    // 不是0又不是1的位来进行处理,并把从这一位开始的后面的所有的位都变成1,这样就是舍弃了
    // 一部分一定不会成为解的数,但是有会所有解空间
        for (int i = len; i >= 0; --i) {
            if (bit[i] > 1) {
                for (int j = i; j >= 0; --j) {
                    bit[j] = 1;
                }
                break;
            }
        }
    }
    return dfs(len, k, 1);
}

int main()
{    memset(dp, 0xff, sizeof (dp));
    while (scanf("%u %u %u %u", &x, &y, &k, &b) == 4) {
        x -= 1; 
        printf("%u\n", Cal(y) - Cal(x));
    }
    return 0;    
} 
posted @ 2012-08-13 16:01  沐阳  阅读(284)  评论(0编辑  收藏  举报