数位DP小结

数位DP的模式一般是从L到R有多少满足条件的数。
写记忆化搜索不错,模式性很强而且好写。省略了不少无效状态。
比较难想的是状态的优化和表示。
SPOJ - BALNUM
Balanced Numbers
给出区间[L,R]问在区间中有多少数字满足其某一位是奇数的位数有偶数个,某一位是偶数的位数有奇数个。
考虑状态压缩,0-9这9个数,每个数3个状态,0代表没使用,1代表用了奇数个,2代表用了偶数个。然后就是个3^10的状态。f[i][j][0/1]表示dp到第i位,状态是j,dp过的前i位和R的大小比较。
然后每次encode/decode存取状态dp即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
int cas,bit[20];
ll l,r,f[20][60000];
bool check(int s)
{
    int tmp[10];
    for(int i=0;i<10;++i)
    {
        tmp[i]=s%3;
        s/=3;
    }
    for(int i=0;i<10;++i)
        if(tmp[i])
        {
            if(i%2==0&&tmp[i]==2)return 0;
            if(i%2==1&&tmp[i]==1)return 0;
        }
    return 1;
}
int getnews(int pos,int s)
{
    int tmp[10];
    for(int i=0;i<10;++i)
    {
        tmp[i]=s%3;
        s/=3;
    }
    if(tmp[pos]==0)tmp[pos]=1;
    else tmp[pos]=3-tmp[pos];
    int news=0;
    for(int i=9;i>=0;--i)
    {
        news*=3;
        news+=tmp[i];
    }
    return news;
}
ll dfs(int pos,int s,bool flag,bool z)
{
    if(pos==-1)return check(s);
    if(!flag&&f[pos][s]!=-1)return f[pos][s];
    ll res=0;
    int end=flag?bit[pos]:9;
    for(int i=0;i<=end;++i)res+=dfs(pos-1,(z&&i==0)?0:getnews(i,s),flag&&i==end,z&&i==0);
    if(!flag)f[pos][s]=res;
    return res;
}
ll cal(ll x)
{
    int tot=0;
    while(x)
    {
        bit[tot++]=x%10;
        x/=10;
    }
    return dfs(tot-1,0,1,1);
}
int main()
{
    scanf("%d",&cas);
    memset(f,-1,sizeof f);
    while(cas--)
    {
        scanf("%lld%lld",&l,&r);
        printf("%lld\n",cal(r)-cal(l-1));
    }
}

CodeForces 55D
Beautiful numbers
给出区间[L,R]问在区间中有多少数字满足这个数是它的所有数位的数的倍数。
易得lcm(0,1,2,3,4,5,6,7,8,9)=2520,那么只要这个数是2520的倍数,这个数一定是0-9中任意一个子集lcm的倍数。所以当前数字的状态得到压缩了。f[i][j][k]表示dp到前i位,这个数mod2520为j,这个数各位数字的lcm为k时满足条件的个数。然后dp就好了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
int cas,cnt,bit[20],inx[2525];
ll l,r,f[20][2520][50];
ll gcd(ll x,ll y)
{
    return y?gcd(y,x%y):x;
}
ll lcm(ll x,ll y)
{
    return x/gcd(x,y)*y;
}
ll dfs(int pos,int presum,int prelcm,bool flag)
{
    if(pos==-1)return presum%prelcm==0;
    if(!flag&&f[pos][presum][inx[prelcm]]!=-1)return f[pos][presum][inx[prelcm]];
    ll res=0;
    int end=flag?bit[pos]:9;
    for(int i=0;i<=end;++i)res+=dfs(pos-1,(presum*10+i)%2520,i?lcm(prelcm,i):prelcm,flag&&i==end);
    if(!flag)f[pos][presum][inx[prelcm]]=res;
    return res;
}
ll cal(ll x)
{
    int tot=0;
    while(x)
    {
        bit[tot++]=x%10;
        x/=10;
    }
    return dfs(tot-1,0,1,1);
}
int main()
{
    scanf("%d",&cas);
    memset(f,-1,sizeof f);
    for(int i=1;i<=2520;++i)if(2520%i==0)inx[i]=cnt++;
    while(cas--)
    {
        scanf("%I64d%I64d",&l,&r);
        printf("%I64d\n",cal(r)-cal(l-1));
    }
}

你可能感兴趣的:(数位dp)