数位DP

Posted czsharecode

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了数位DP相关的知识,希望对你有一定的参考价值。

数位DP的算法思想

  • 数位DP放弃了原来数要从1连续枚举到N逐个检查合法性的思想,而是专注于数本身的字符串构成特性
  • 对于每一种限制条件,dp数组都设置了一维表示。比如dp[pos][s1][s2]...[sn].表示在pos位置下,之前限制条件s1 s2 ... sn的状态,当枚举当前pos处的数字i时,状态s1 s2 ... sn 以及 i是否会失去合法性。若该状态不合法,及时剪枝。

  • 采用记忆化搜索的方法,从高位向低位搜索,每搜索一步更新当前字串的状态(即与限制条件的关系),直到搜索完最后一位,根据当前状态是否与限制条件相矛盾来决定是否贡献答案。

  • 若某种状态已经计算出来,并且是合法的,则无需再搜索,直接返回计算结果

  • 算法步骤
    typedef long long ll;
    int a[20];
    ll dp[20][state];//不同题目状态不同
    ll dfs(int pos,/*state变量*/,bool lead/*前导零*/,bool limit/*数位上界变量*/)//不是每个题都要判断前导零
    {
        //递归边界,既然是按位枚举,最低位是0,那么pos==-1说明这个数我枚举完了
        if(pos==-1) return 1;/*这里一般返回1,表示你枚举的这个数是合法的,那么这里就需要你在枚举时必须每一位都要满足题目条件,也就是说当前枚举到pos位,一定要保证前面已经枚举的数位是合法的。不过具体题目不同或者写法不同的话不一定要返回1 */
        //第二个就是记忆化(在此前可能不同题目还能有一些剪枝)
        if(!limit && !lead && dp[pos][state]!=-1) return dp[pos][state];
        /*常规写法都是在没有限制的条件记忆化,这里与下面记录状态是对应,具体为什么是有条件的记忆化后面会讲*/
        int up=limit?a[pos]:9;//根据limit判断枚举的上界up;这个的例子前面用213讲过了
        ll ans=0;
        //开始计数
        for(int i=0;i<=up;i++)//枚举,然后把不同情况的个数加到ans就可以了
        {
            if() ...
            else if()...
            ans+=dfs(pos-1,/*状态转移*/,lead && i==0,limit && i==a[pos]) //最后两个变量传参都是这样写的
    
        }
        //计算完,记录状态
        if(!limit && !lead) dp[pos][state]=ans;
        /*这里对应上面的记忆化,在一定条件下时记录,保证一致性,当然如果约束条件不需要考虑lead,这里就是lead就完全不用考虑了*/
        return ans;
    }
    ll solve(ll x)
    {
        int pos=0;
        while(x)//把数位都分解出来
        {
            a[pos++]=x%10;
            x/=10;
        }
        return dfs(pos-1/*从最高位开始枚举*/,/*一系列状态 */,true,true);//刚开始最高位都是有限制并且有前导零的,显然比最高位还要高的一位视为0而且是有限制的
    }
    int main()
    {
        ll le,ri;
        while(~scanf("%lld%lld",&le,&ri))
        {
            //初始化dp数组为-1,这里还有更加优美的优化,后面讲
            printf("%lld
",solve(ri)-solve(le-1));
        }
    }

入门例题

  • HDU 2089 求区间[l,r]内不要62也不要4的数的个数
#include<iostream>
#include<queue>
#include<stack>
#include<list>
#include<vector>
#include<cstring>
#include<set>
using namespace std;
typedef long long ll;
const int maxn = 1e5+10;
ll a[30]; //分解数位
ll dp[30][2]; //dp[pos][status]表示到pos位置,当前状态为status,合法的数有多少个

//记忆化DFS
ll dfs(int pos , bool limit, bool lead, bool status){
    //如果枚举完最后一个数都不出错,证明这个数字是合法的,返回1贡献给答案
    if(pos == -1) return 1; 
    //如果没有限制 没有前导0 并且当前DP[pos][status]的值已经被计算过了,则可以直接用不需要再搜索了
    if(!limit && !lead && dp[pos][status] != -1) return dp[pos][status];
    //当前位的上限是否有限制
    int up = limit ? a[pos] : 9;
    //累计合法数到ans
    ll ans = 0;
    //枚举该位所有可能的数字
    for(int i=0; i<=up; i++){
        if(i == 4) continue; //不合条件的统统continue
        if(status && i==2) continue; //之前status为TRUE则表明前的是6,所以这一位不能为2
        ans += dfs(pos-1 , limit && i==a[pos] , lead && i==0, i==6);//向前搜索
    }
    //如果没有限制并且没有前导0 则可以赋值 以便用于记忆化搜索
    if(!limit && !lead) dp[pos][status] = ans;
    return ans;
}

//SOLVE 表示[0,x]区间内合法的数字  事实上如果对前导0有要求那么就是[1,x]
ll solve(int x){
    int pos = -1;
    while(x){
        pos++;
        a[pos] = x % 10;
        x /= 10;
    }
    return dfs(pos , 1 , 1, 0);
}
ll l,r;
int main(){
    while(cin>>l>>r){
        if(l == 0 && r==0) break;
        //初始DP记为-1表示没有计算出来结果
        for(int i=0; i<=20; i++){
            dp[i][0] = -1;
            dp[i][1] = -1;
        }
        cout<<solve(r) - solve(l-1)<<endl;
    }
}

数位DP中limit标记

  • 在数位dp中,limit的作用主要有两个。

  • 控制枚举的界限,倘若没有界限,每一位的枚举范围都是0-9. 但如果有界限,那么可能不能取到9. 例如求1到311范围内不含有连续两个1的个数。当百位枚举到了3,那么如果你的十位只能枚举0-2.

  • 控制剪枝的合法性: 很明显,相比于暴力搜索,数位dp唯一的优势就是剪枝。但剪枝往往会带来一些错误。在数位dp中,我们进行剪枝的一个前提是limit != 0,也就是当前位枚举没有限制时才能进行剪枝。
    • 为什么?请先看这个例子:题目依然是求1到321中,不含连续两个1的个数。当百位枚举为0,十位枚举为1时,当枚举个位时,除了个位枚举1不满足题意外,其他9个数都满足,所以dp[0][1]=9。而当百位枚举到3时,十位枚举1,此时我们已经记忆了dp[0][1]=9,但很明显,此时只有310,满足题意,如果还采用之前的9则发生错误。
    • 为什么会出现这样的情况? 其实一个明显的原因就是:我们记忆化时的状态,仅仅只记忆了limit = 1的情况,也就是没有限制时的状态。该算法认为,有限制条件的数是很少的,所以对于有限制条件的数,我们都是直接暴力搜索的。



  • Balanced Number HDU - 3709
    • 一个数是平衡数,当且仅当能找到一个枢轴使得两边力矩平衡
    • 技巧
      • 枚举枢轴
      • 左边力矩和为正,右边为负,最终为0则合法
#include<cstdio>
#include <iostream>
#include <cstdio>
#include<queue>
#include<algorithm>
#include<stack>
#include<list>
#include<vector>
#include<cstring>
#include<string>
#include<set>
#include<map>
using namespace std;
typedef long long ll;

ll dp[20][20][2100];//dp[i][j][k]表示当前处于第i位,支点是j,力矩和为k  对答案的贡献
int a[30];

ll dfs(int pos , int pivot, int sum, bool limit, bool lead){
    if(pos <= 0) return sum == 0;

    if(sum < 0) return 0;

    if(!limit && !lead && dp[pos][pivot][sum] != -1) return dp[pos][pivot][sum];

    ll ans = 0;

    int up = limit ? a[pos] : 9;

    for(int i=0; i<=up; i++){
        int t = sum + (pos - pivot) * i;
        ans += dfs(pos-1, pivot, t, limit && i==a[pos], lead && i==0);
    }

    if(!lead && !limit) dp[pos][pivot][sum] = ans;
    return ans; 

}

ll solve(ll x){
    ll ans = 0;
    int pos = 0;
    while(x){
        pos++;
        a[pos] = x % 10;
        x /= 10;
        //cout<<"???"<<endl;
    }
    //cout<<"pos = "<<pos<<endl;
    for(int i=1; i<=pos; i++){
        //枚举支点 从第1位到第pos位
        ans += dfs(pos , i, 0, 1, 1)-1;
    }
    return ans ;
}


int main(){
    int t;
    cin>>t;
    ll x,y;
    while(t--){
        for(int i=0; i<20; i++){
            for(int j=0; j<20; j++){
                for(int k=0; k<2100; k++){
                    dp[i][j][k] = -1;   
                }
            }
        }
        dp[0][0][0] = 1;
        cin>>x>>y;
        cout<<solve(y) - solve(x-1)<<endl;
    }
    return 0;
}






  • B-number HDU
    • 求[1,n]有多少个能被13整除且包含13的数字
    • 维护一个余数mod,模拟整数除法,最后为0合法
    • 必须包含13,维护一个has标记,在status为1时,若当前位置为3则has生效
    • has = 1 且 mod = 0才能贡献答案
#include<iostream>
#include<queue>
#include<stack>
#include<list>
#include<vector>
#include<cstring>
#include<set>
using namespace std;
typedef long long ll;
const int maxn = 1e5+10;
ll a[30];
ll dp[30][15][2][2];//dp[pos][mod][status][has]
//mod表示当前的余数  status 表示前面一位是不是1  has表示到当前位有没有13
ll dfs(int pos, bool limit, bool lead, int mod, bool status, bool has){

    if(pos == -1) return mod == 0 && has;

    if(!limit && !lead && dp[pos][mod][status][has] != -1) return dp[pos][mod][status][has];

    int up = limit ? a[pos] : 9;

    ll ans = 0;
    
    for(int i=0; i<=up; i++){
        //注意这里一定要考虑前导0,前导0不算0的个数
       ans += dfs(pos-1, limit&&i == a[pos], lead && i==0, (mod*10+i)%13, i==1, has || (status && i==3) );
    }
    
    if(!limit && !lead) dp[pos][mod][status][has] = ans;
    return ans;
}

ll solve(ll x){
    int pos = -1;
    while(x){
        pos++;
        a[pos] = x % 10;
        x = x / 10;
    }
    return dfs(pos, 1, 1, 0, 0, 0);
}
ll l,r;
int main(){
    while(cin>>r){
        for(int i=0; i<30; i++){
            for(int j=0; j<15; j++){
                for(int k=0; k<2; k++){
                    for(int x=0; x<2; x++)
                        dp[i][j][k][x] = -1;
                } 
            } 
        }
        cout<<solve(r)<<endl;
    }
}

以上是关于数位DP的主要内容,如果未能解决你的问题,请参考以下文章

数位dp小练

动态规划_计数类dp_数位统计dp_状态压缩dp_树形dp_记忆化搜索

数位DP

数位dp

HDU 2089 数位dp入门

数位DP