首页 > 代码库 > [数位dp] spoj 10606 Balanced Numbers

[数位dp] spoj 10606 Balanced Numbers

题意:

对于一个数的每个位上的数。

对于每个奇数,如果出现必须出现偶数次。

对于每个偶数,如果出现必须出现奇数次。

思路:

用三进制存储每个数出现的状态,0没出现,1出现奇数次,2出现偶数次。

然后其他和普通数位dp就一样了。

代码:

#include"cstdlib"
#include"cstdio"
#include"cstring"
#include"cmath"
#include"queue"
#include"algorithm"
#include"iostream"
using namespace std;
//2014年9月25日10:03:08
long long dp[22][180000];
int num[22];
int ok(int x)
{
    int v[50],cnt=0;
    while(x)
    {
        v[cnt++]=x%3;
        x/=3;
    }
    for(int i=0;i<cnt;i++)
    {
        if(!v[i]) continue;
        if(i%2==0&&v[i]!=1) return 0;
        if(i%2!=0&&v[i]!=2) return 0;
    }
    return 1;
}
int js(int n,int x)
{
    int v[50],cnt=0,tep=0;
    memset(v,0,sizeof(v));
    while(n)
    {
        v[cnt++]=n%3;
        n/=3;
    }
    if(v[x]<2) v[x]++;
    else v[x]=1;
    for(int i=9;i>=0;i--) tep=tep*3+v[i];
    return tep;
}
long long dfs(int site,int n,int zero,int f)
{
    if(site==0)
    {
        if(zero) return 0;
        return ok(n);
    }
    if(!f&&!zero&&~dp[site][n]) return dp[site][n];
    int len=f?num[site]:9;
    long long ans=0;
    for(int i=0; i<=len; i++)
    {
        if(zero)
        {
            if(i==0) ans+=dfs(site-1,n,zero&&i==0,f&&i==len);
            else ans+=dfs(site-1,js(n,i),zero&&i==0,f&&i==len);
        }
        else ans+=dfs(site-1,js(n,i),zero&&i==0,f&&i==len);
    }
    if(!f&&!zero) dp[site][n]=ans;
    return ans;
}
long long solve(long long x)
{
    int cnt=0;
    while(x)
    {
        num[++cnt]=x%10;
        x/=10;
    }
    return dfs(cnt,0,1,1);
}
int main()
{
    int t;
    cin>>t;
    memset(dp,-1,sizeof(dp));
    while(t--)
    {
        long long x,y;
        scanf("%lld%lld",&x,&y);
        printf("%lld\n",solve(y)-solve(x-1));
    }
    return 0;
}
//2014年9月25日10:39:37


[数位dp] spoj 10606 Balanced Numbers