Google Code Jam 2014 Round 1B Problem B

二进制数位DP，涉及到数字的按位与操作。

查看官方解题报告

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

#define MAX_LEN 50

long long A, B, K;
int a[MAX_LEN], b[MAX_LEN], k[MAX_LEN];
long long memoize[MAX_LEN][][][];

void input()
{
    scanf("%lld%lld%lld", &A, &B, &K);
}

int convert(long long A, int a[])
{
    int i = ;
    while (A)
    {
        a[i] = A & ;
        A >>= ;
        i++;
    }
    return i;
}

long long dfs(int current_bit, bool less_a, bool less_b, bool less_k)
{
    if (current_bit == -)
    {
        if (less_a && less_b && less_k)
        {
            return ;
        }
        return ;
    }
    if (memoize[current_bit][less_a][less_b][less_k] != -)
        return memoize[current_bit][less_a][less_b][less_k];
    bool one_a = less_a || a[current_bit] == ;
    bool one_b = less_b || b[current_bit] == ;
    bool one_k = less_k || k[current_bit] == ;
    // a0 b0
    long long ret = dfs(current_bit - , one_a, one_b, one_k);
    // a1 b0
    if (one_a)
    {
        ret += dfs(current_bit - , less_a, one_b, one_k);
    }
    // a0 b1
    if (one_b)
    {
        ret += dfs(current_bit - , one_a, less_b, one_k);
    }
    // a1 b1
    if (one_a && one_b && one_k)
    {
        ret += dfs(current_bit - , less_a, less_b, less_k);
    }
    return memoize[current_bit][less_a][less_b][less_k] = ret;
}

int main()
{
    int t;
    scanf("%d", &t);
    for (int i = ; i < t; i++)
    {
        printf("Case #%d: ", i + );
        input();
        memset(a, , sizeof(a));
        memset(b, , sizeof(b));
        memset(k, , sizeof(k));
        convert(A, a);
        convert(B, b);
        convert(K, k);
        memset(memoize, -, sizeof(memoize));
        long long ans = dfs(, false, false, false);
        printf("%lld\n", ans);
    }
    return ;
}