spoj 10606 Balanced Numbers 数位dp

题目链接

一个数称为平衡数， 满足他各个数位里面的数， 奇数出现偶数次， 偶数出现奇数次， 求一个范围内的平衡数个数。

用三进制压缩， 一个数没有出现用0表示， 出现奇数次用1表示， 出现偶数次用2表示， 这样只需要开一个20*60000的数组。

 #include<bits/stdc++.h>
 using namespace std;
 #define pb(x) push_back(x)
 #define ll long long
 #define mk(x, y) make_pair(x, y)
 #define lson l, m, rt<<1
 #define mem(a) memset(a, 0, sizeof(a))
 #define rson m+1, r, rt<<1|1
 #define mem1(a) memset(a, -1, sizeof(a))
 #define mem2(a) memset(a, 0x3f, sizeof(a))
 #define rep(i, a, n) for(int i = a; i<n; i++)
 #define ull unsigned long long
 typedef pair<int, int> pll;
 const double PI = acos(-1.0);
 const double eps = 1e-;
 const int mod = 1e9+;
 const int inf = ;
 const int dir[][] = { {-, }, {, }, {, -}, {, } };
 int digit[], a[];
 ll dp[][];
 int judge(int num) {
     int cnt = ;
     for(int i = ; i<; i++) {
         a[i] = num%;
         num/=;
     }
     for(int i = ; i<; i++) {
         if(i%==&&a[i]==)
             return ;
         if(i%==&&a[i]==)
             return ;
     }
     return ;
 }
 int cal(int num, int tmp) {
     int cnt = ;
     for(int i = ; i<; i++) {
         a[i] = num%;
         num/=;
     }
     a[tmp]++;
     if(a[tmp]==)
         a[tmp]=;
     for(int i = ; i>=; i--) {
         num = num*+a[i];
     }
     return num;
 }
 ll dfs(int len, int num, int fp, bool first) {
     if(!len) {
         return judge(num);
     }
     if(!fp&&dp[len][num]!=-) {
         return dp[len][num];
     }
     ll ret = ;
     int maxx = fp?digit[len]:;
     for (int i = ; i<=maxx; i++) {
         ret += dfs(len-, (first&&i==)?:cal(num, i), fp&&i==maxx, i==&&first);
     }
     if(!fp)
         return dp[len][num] = ret;
     return ret;
 }
 ll cal(ll n) {
     int len = ;
     while(n) {
         digit[++len] = n%;
         n/=;
     }
     return dfs(len, , , true);
 }
 int main()
 {
     mem1(dp);
     int t;
     ll a, b;
     cin>>t;
     while(t--) {
         scanf("%lld%lld", &a, &b);    //I64d会超时......
         printf("%lld\n", cal(b)-cal(a-));
     }
 }