数位dp入门

最新推荐文章于 2021-07-21 20:06:21 发布

denglao1906

最新推荐文章于 2021-07-21 20:06:21 发布

阅读量231

点赞数

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/01world/p/5651204.html

版权

数位dp，一般的题意是要求把某个区间的符合某种特征的数的个数求出来，一般的思路就是把最大数按位分解，然后dfs依次判断每一位相应的数是否满足要求。
把以前做过的几道简单数位dp的题总结一下，可以作为入门题做做。
1.
hdu 2089 不要62
题意：给定一个区间[a,b]，问在这个区间中的数字，不包含4和62的数有多少个？

分析：数据范围小于1e6，如果暴力的话，再加上分解因数，预处理的时间复杂度数量级最大1e7，查询O(1)。
用数位dp的做法：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int f[8][2],bit[8];
/*
f[i][j]:i是位数
j==0，尾位不是6
j==1，尾位是6
*/
int dfs(int pos,bool six,bool lim)//位数，尾位是否是6，是否是上限
{
    if(pos<=0)return 1;
    if(!lim&&f[pos][six]!=-1)return f[pos][six];
    int num=lim?bit[pos]:9; //假设该位是2,下一位是3，如果现在算到该位为1，那么lim=0，表示下一位是能取到9的，
                            //如果该位为2，那么lim=1，下一位只能取到3
    int ans=0;
    for(int i=0;i<=num;i++){
        if(i==4||six&&i==2)continue;//如果再加一位是4，或者尾位是6再加一位是2，那么都是不合法的状态，跳过。
        ans+=dfs(pos-1,i==6,lim&&i==num);
    }
    if(!lim)f[pos][six]=ans;
    return ans;
}
int solve(int n)
{
    int len=0;
    memset(f,-1,sizeof(f));
    while(n){
        bit[++len]=n%10;
        n/=10;
    }
    return dfs(len,0,1);
}
int main()
{
    int a,b;
    while(~scanf("%d%d",&a,&b)&&(a+b)){
        printf("%d\n",solve(b)-solve(a-1));
    }
    return 0;
}

2.
hdu 3652 B-number
题意：求小于n，是13的倍数且含有’13’的数的个数。n<1e9

分析：有两个要求，不含13且不是13的倍数，这样的话，可是可以用三维数组表示状态f[i][j][k]，i表示数位，然后增加上用j表示余数，k表示是否含有13。然后dfs，从高位开始以此判断每一位的状态。

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
using namespace std;
const int N=15;
int bit[N],f[N][N][3];
/*
dp[i][j][k] j表示余数
k==0 不包含13且不以1结尾
k==1 不包含13且以1结尾
k==2 包含13
*/
int dfs(int pos,int mod,int have,bool lim)//第几位数，余数，状态标号，是否是上限
{
    if(pos<=0)return mod==0&&have==2;
    if(!lim&&f[pos][mod][have]!=-1)return f[pos][mod][have];
    int num=lim?bit[pos]:9;
    int ans=0;
    for(int i=0;i<=num;i++){
        int mod_x=(mod*10+i)%13; //传到下一位的余数，模拟除法的过程
        int have_x=have;
        if(have==0&&i==1)have_x=1;//末尾不是1，再加一位是1，标记为1
        if(have==1&&i!=1)have_x=0;//末尾是1，再加一位不是1，标记为0
        if(have==1&&i==3)have_x=2;//末尾是1，再加一位是3，标记为2
        ans+=dfs(pos-1,mod_x,have_x,lim&&i==num);
    }
    if(!lim)f[pos][mod][have]=ans;
    return ans;
}
int main()
{
    int n,len;
    while(~scanf("%d",&n)){
        memset(f,-1,sizeof(f));
        len=0;
        while(n){
            bit[++len]=n%10;
            n/=10;
        }
        printf("%d\n",dfs(len,0,0,1));
    }
    return 0;
}

3.
hdu 3555 含有49的数：
题意：
找出2^63范围内含有49的数，注意用long long

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
const int N=20;
ll f[N][3];
int bit[N];
//f[pos][have]
//0 最后一位不是4且不含49
//1 最后一位是4
//2 有49
ll dfs(int pos,int have,bool lim)
{
    if(pos<=0)return have==2;
    if(!lim&&f[pos][have]!=-1)return f[pos][have];
    int num=lim?bit[pos]:9;
    ll ans=0;
    for(int i=0;i<=num;i++){
        int have_x=have;
        if(have==1&&i==9)have_x=2;
        else if(have==0&&i==4)have_x=1;
        else if(have==1&&i!=4)have_x=0;

        ans+=dfs(pos-1,have_x,lim&i==num);
    }
    if(!lim)f[pos][have]=ans;
    return ans;
}
ll solve(ll n)
{
    memset(f,-1,sizeof(f));
    int tot=0;
    while(n){
        bit[++tot]=n%10;
        n/=10;
    }
    return dfs(tot,0,1);
}
int main()
{
    int T;scanf("%d",&T);
    while(T--){
        ll n;cin>>n;
        printf("%lld\n",solve(n));
    }
    return 0;
}

这个也可以用递推解决：

dp[k][0]=dp[k-1][0]*9+dp[k-1][1]*8;         //dp[][0]表示不包含49并且以非4结尾的个数

dp[k][1]=dp[k-1][0]+dp[k-1][1];           //dp[][1]表示不包含49并且以4结尾的个数

dp[k][2]=dp[k-1][1]+dp[k-1][2]*10;        //dp[][2]表示包含49的个数

4.
数位dp的思想是处理一个序列，一位位的分析，上面的都是先把数分解成一个序列，有的题目就是直接问哟多少种满足要求的序列，处理也是一样的，对每一个位处理即可。
例：
hdu 5642 数位dp/ 递推
题意：数一个长度为 n的序列，并且序列中不能出现长度大于 3 的连续的相同的字符
分析：
这道题目是BC的一道题目，大部分人是用递推解决的，因为递推特别简单。

    ll f[N][4]; //依次表示末尾应经出现连续相等字母的数量
    f[1][1]=26;
    for(int i=2;i<N;i++){
        f[i][1]=(f[i-1][1]+f[i-1][2]+f[i-1][3])*25%mod;
        f[i][2]=f[i-1][1];
        f[i][3]=f[i-1][2];
    }

这题用数位dp的思想也是可以做的，d[i][j][k]为处理完 i个字符 , 结尾字符为 ′a′+j, 结尾部分已重复出现了 k次的方案数。

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
const int mod=1e9+7;
const int N=2009;
int f[N][26][4];
int dfs(int pos,int m,int k)
{
    if(pos<=1)return 1;
    if(f[pos][m][k]!=-1)return f[pos][m][k];
    int ans=0;
    for(int i=0;i<26;i++){
        if(i!=m)ans=(ans+dfs(pos-1,i,1))%mod;
        else if(i==m&&k<3)ans=(ans+dfs(pos-1,i,k+1))%mod;
    }
    return f[pos][m][k]=ans;
}
int main()
{
    int T;scanf("%d",&T);
    int n;
    memset(f,-1,sizeof(f));
    while(T--){
        scanf("%d",&n);
        for(int i=0;i<26;i++)dfs(n,i,1);
        int ans=0;
        for(int i=0;i<26;i++)ans=(ans+f[n][i][1])%mod;
        printf("%d\n",ans);
    }
    return 0;
}

5.
hdu 4347
题意：给一个数A (十进制表示形式为AnAn-1An-2 … A2A1，定义函数 F(x) = An * 2n-1 + An-1 * 2n-2 + … + A2 * 2 + A1 * 1，给一个B,求B以内的i,满足F(i)<=F(A)

分析：f[i][j]表示i位比j小的数有多少，先计算出F(a)，然后dfs，找出比fa小的B以内的数，找B以内的数，这就是简单的数位dp了，先把B按位存储，然后从高位dfs一位位比较就好了。

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
const int mod=1e9+7;
const int N=2009;
int f[12][4600],w[12];
int bit[12];
int dfs(int pos,int sum,bool lim)
{
    if(pos<=0)return 1;
    if(!lim&&f[pos][sum]!=-1)return f[pos][sum];
    int num=lim?bit[pos]:9;
    int ans=0;
    for(int i=0;i<=num;i++){
        if(sum-i*w[pos]>=0)ans+=dfs(pos-1,sum-i*w[pos],lim&&i==num);
    }
    if(!lim)f[pos][sum]=ans;
    return ans;
}
int main()
{
    for(int i=0;i<11;i++)w[i+1]=1<<i;
    int T;scanf("%d",&T);
    memset(f,-1,sizeof(f));
    for(int cas=1;cas<=T;cas++){
        int a,b;
        scanf("%d%d",&a,&b);
        int fa=0,i=0;
        while(a){
            fa+=(a%10)*w[++i];
            a/=10;
        }
        i=0;
        while(b){
            bit[++i]=b%10;
            b/=10;
        }
        printf("Case #%d: %d\n",cas,dfs(i,fa,1));
    }
    return 0;
}

6.
题意：
d magic number（0<=d<9）的意思就是一个数，从最高位开始奇数位不是d,偶数位是d
题目问，给a,b,m,d(a<=b,m<2000)问，a,b之间有多少个数满足既是d magic number，又可以被m整除
分析：
跟上面第二题有些类似，需要注意位数最大可能是2000。

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
const int mod=1e9+7;
const int N=2009;
int f[N][N],bit[N];
char a[N],b[N];
int m,d,len;
int dfs(int pos,int pre,bool lim)
{
    if(pos==len+1)return pre==0;
    if(!lim&&f[pos][pre]!=-1)return f[pos][pre];
    int num=lim?bit[pos]:9;
    int ans=0;
    for(int i=0;i<=num;i++){
        if((pos&1)&&i==d)continue;
        if(!(pos&1)&&i!=d)continue;
        ans=(ans+dfs(pos+1,(pre*10+i)%m,lim&&i==num))%mod;
    }
    if(!lim)f[pos][pre]=ans;
    return ans;
}
int solve(const char *s)
{
    len=strlen(s);
    for(int i=0;i<len;i++)bit[i+1]=s[i]-'0';
    return dfs(1,0,1);
}
int judge(const char *s)
{
    len-strlen(s);
    int pre=0;
    for(int i=0;i<len;i++){
        pre=(pre*10+s[i]-'0')%m;
        if(i&1&&(s[i]-'0'!=d))return 0;
        if(!(i&1)&&s[i]-'0'==d)return 0;
    }
    return pre==0;
}
int main()
{
    //freopen("f.txt","r",stdin);
    memset(f,-1,sizeof(f));
    scanf("%d%d",&m,&d);
    scanf("%s%s",a,b);
    printf("%d\n",(solve(b)-solve(a)+judge(a)+mod)%mod);
    return 0;
}

上面的6道数位dp题可以看出形式差不多是一样的，前几道是直接搬得我以前写的题解，注释写的比较详细，希望可以帮助需要的朋友！

转载于:https://www.cnblogs.com/01world/p/5651204.html

denglao1906

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
数位dp入门

数位dp，一般的题意是要求把某个区间的符合某种特征的数的个数求出来，一般的思路就是把最大数按位分解，然后dfs依次判断每一位相应的数是否满足要求。把以前做过的几道简单数位dp的题总结一下，可以作为入门题做做。 1. hdu 2089 不要62 题意：给定一个区间[a,b]，问在这个区间中的数字，不包含4和62的数有多少个？分析：数据范围小于1e6，如果暴...
复制链接

扫一扫