SPOJ BALNUM Balanced Numbers 入门数位dp

https://www.spoj.com/problems/BALNUM/en/

题意:如果一个数每个位上的奇数出现偶数次, 偶数出现奇数次, 那这个数是好数, 问区间内有多少个好数

思路: 数位dp

开20+3*10+2共12维空间, 20是数长度, 3*10表示0-9每个数出现的情况,0表示没出现过, 1表示出现奇数次, 2表示出现偶数次, 2表示是否还在前导零区域.

#include<bits/stdc++.h>
#define fuck(x) std::cout<<"["<<#x<<"->"<<x<<"]"<<endl;
using namespace std;
typedef long long ll;

const int M=2e5+5;
const int inf=1e9+5;
const int mod=1e9+7;
//memset(a,0x3f,sizeof(a));

int s[20];
ll dp[20][3][3][3][3][3][3][3][3][3][3][2];
//    pos 0  1  2  3  4  5  6  7  8  9 fs0
//fs0=1表示还处于前导零
ll dfs(int pos,int n0,int n1,int n2,int n3,int n4,int n5,
       int n6,int n7,int n8,int n9,int lim,int fs0) {
    if(pos<=0) {
        if(n0==2||n1==1||n2==2||n3==1||n4==2||n5==1||n6==2||n7==1||n8==2||n9==1)
            return 0;
        else
            return 1;
    }
    if(!lim&&dp[pos][n0][n1][n2][n3][n4][n5][n6][n7][n8][n9][fs0]!=-1) {
        return dp[pos][n0][n1][n2][n3][n4][n5][n6][n7][n8][n9][fs0];
    }
    int nn=lim?s[pos]:9;
    ll ans=0;
    for(int i=0; i<=nn; i++) {
        int _n0=n0,_n1=n1,_n2=n2,_n3=n3,_n4=n4,_n5=n5,
            _n6=n6,_n7=n7,_n8=n8,_n9=n9,_fs0;
        if(i!=0||fs0==0) _fs0=0;
        else _fs0=1;
        if(i==0&&_fs0==0) {
            if(n0==0||n0==2) _n0=1;
            else _n0=2;
        }
        if(i==1) {
            if(n1==0||n1==2) _n1=1;
            else _n1=2;
        }
        if(i==2) {
            if(n2==0||n2==2) _n2=1;
            else _n2=2;
        }
        if(i==3) {
            if(n3==0||n3==2) _n3=1;
            else _n3=2;
        }
        if(i==4) {
            if(n4==0||n4==2) _n4=1;
            else _n4=2;
        }
        if(i==5) {
            if(n5==0||n5==2) _n5=1;
            else _n5=2;
        }
        if(i==6) {
            if(n6==0||n6==2) _n6=1;
            else _n6=2;
        }
        if(i==7) {
            if(n7==0||n7==2) _n7=1;
            else _n7=2;
        }
        if(i==8) {
            if(n8==0||n8==2) _n8=1;
            else _n8=2;
        }
        if(i==9) {
            if(n9==0||n9==2)_n9=1;
            else _n9=2;
        }
        ans+=dfs(pos-1,_n0,_n1,_n2,_n3,_n4,_n5,_n6,_n7,_n8,_n9,lim&&i==s[pos],_fs0);
    }
    if(!lim)
        dp[pos][n0][n1][n2][n3][n4][n5][n6][n7][n8][n9][fs0]=ans;
    return ans;
}

ll solve(ll x) {
    int si=0;
    memset(s,0,sizeof(s));
    while(x>0) {
        s[++si]=x%10;
        x/=10;
    }
    return dfs(si,0,0,0,0,0,0,0,0,0,0,1,1);
}

int main() {
    int _;
    scanf("%d",&_);
    memset(dp,-1,sizeof(dp));
    while(_--) {
        ll n,m;
        scanf("%lld%lld",&n,&m);
        printf("%lld\n",solve(m)-solve(n-1));
    }
    return 0;
}