SPOJ - BALNUM Balanced Numbers 数位dp+状态压缩

1、状态压缩，用一个10位的3进制数字来表示0~9的状态，0表示未出现过，1表示出现奇数次，2表示出现偶数次。

2、输出时不要用I64d。

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <vector>
#include <cstdio>
#include <map>

using namespace std;

typedef long long int LL;
LL dp[20][60000];
int digit[20];

// 将pre转化成3进制存放在temp中
void change(int pre, int* temp)
{
int len = 0;
while (pre)
{
temp[len++] = pre % 3;
pre /= 3;
}
}

LL dfs(int pos, int pre, int lead, int limit)
{
if (!pos)
{
int temp[15] = {0};
change(pre, temp);
for (int i = 0; i < 10; i++)
if ((i%2==0 && temp[i]==2) || (i%2==1 && temp[i]==1))
return 0;
return 1;
}
if (!limit && dp[pos][pre] != -1)
return dp[pos][pre];
int up = limit ? digit[pos] : 9;
LL ans = 0;
for (int i = 0; i <= up; i++)
{
int temp[15] = {0};
change(pre, temp);
if (!temp[i])
temp[i] = 1;
else if (temp[i] == 1)
temp[i] = 2;
else
temp[i] = 1;
int num = 0;
for (int i = 9; i >= 0; i--)
num = num * 3 + temp[i];
ans += dfs(pos-1, (lead&&!i)?0:num, lead&&!i, limit&&i==up);
}
return limit ? ans : dp[pos][pre] = ans;
}

LL cal(LL n)
{
int len = 0;
while (n)
{
digit[++len] = n % 10;
n /= 10;
}
return dfs(len, 0, 1, 1);
}

int main()
{
//freopen("test.txt", "r", stdin);
memset(dp, -1, sizeof(dp));
int T;
scanf("%d", &T);
while (T--)
{
LL L, R;
cin >> L >> R;
cout << cal(R) - cal(L - 1) << endl;
}
return 0;
}