SPOJ - BALNUM 数位DP + 状态压缩

SPOJ - BALNUM

题意

一个正整数 $x$ 是平衡 的当且仅当：$x$ 的奇数数位出现了偶数次，偶数数位出现了奇数次

求区间 $[l,r]$ 的平衡数数量

思路

考虑数位 $DP$ + 状态压缩来表示当前限制下每一位的状态。
$pos$ 个全变化位，$sta$ 为当前 $0$ ~$9$ 每个数位的出现次数（将其转成二进制看每一位的值，$1$ 表示这一位出现了奇数次，反之则出现偶数次），$oc$ 表示 $0$ ~ $9$ 每个数位是否出现（也是一个十进制数，将其看成二进制来每一位判断，如果是 $1$ 则代表出现了）

那么 $dp[pos][sta][oc]$ 的值就是当前限制条件下平衡数的数量。

转移的话，我们可以初始化 $sta$ 为 $341$ ，$oc$ 为 $0$。这是因为我们最终要让奇数数位出现偶数次，偶数数位出现奇数次，所以初始化奇数数位的那些地方为 $1$，如果它们出现了奇数次，那么这一位的结果最终就是 $0$，偶数为也是类似。那么 $sta=0101010101_2$，二进制转十进制就是 $341$。

通过这个巧妙的构造，我们最底层的判断是否合法的条件就可以写为：$sta$ & $oc=0$。也就是出现了的那些位必须要符合限制条件。

转移的话要注意一下前导 $0$。

也可以用三进制来表示状态压缩，$1$ 表示这个位出现了奇数次，$2$ 表示这个位出现了偶数次，$0$ 表示没有出现。复杂度是更优的，因为：$2^{20}>3^{10}$

#include<bits/stdc++.h>
#define fore(i,l,r)	for(int i=(int)(l);i<(int)(r);++i)
#define fi first
#define se second
#define endl '\n'
#define ull unsigned long long
#define ALL(v) v.begin(), v.end()
#define Debug(x, ed) std::cerr << #x << " = " << x << ed;

const int INF=0x3f3f3f3e;
const long long INFLL=1e18;

typedef long long ll;

ull dp[20][1030][1030];
int num[20];

ull dfs(int pos, int sta, int oc, bool lead, bool limit){
    if(!pos) return !(sta & oc);
    if(!lead && !limit && ~dp[pos][sta][oc]) return dp[pos][sta][oc];
    ull res = 0;
    int up = (limit ? num[pos] : 9);
    fore(i, 0, up + 1){
        if(lead && !i) res += dfs(pos - 1, sta, oc, true, limit && i == up);
        else res += dfs(pos - 1, sta ^ (1 << i), oc | (1 << i), false, limit && i == up);
    }
    if(!lead && !limit) dp[pos][sta][oc] = res;
    return res;
}

ull solve(ull x){
    int len = 0;
    while(x){
        num[++len] = x % 10;
        x /= 10;
    }
    return dfs(len, 341, 0, true, true);
}

int main(){
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    std::cout.tie(nullptr);
    int t;
    std::cin >> t;
    memset(dp, -1, sizeof(dp));
    while(t--){
        ull l, r;
        std::cin >> l >> r;
        std::cout << solve(r) - solve(l - 1) << endl;
    }
    return 0;
}