数位DP的实现

#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#define mod 13//这里这个mod的取值就要看题了,如果是要要求能整除13的数,那么mod就是13,最后的结果就是求和dp[len][0][0]+dp[len][0][1]
using namespace std;
int dp[105][mod][2];//dp[i][j][0]就代表的是总长度为len-i的所有数字中凡是所有模13为j的数字且前缀和小于前面的数字的所有数字的个数,0就表示还没有满,但是1就是表示已经满了
int mi[105];//保存的是10的i次方的幂对mod取模的结果
int cal(int x,int wei){
	return (x*mi[wei])%mod;//如x=3,wei=4  那么这个方程返回的值就是30000对mod的模
}
int solve(string s){
        memset(dp,0,sizeof(dp));
        dp[0][0][1]=1;//这个初始状态就是说前缀和是满的情况下给一个初始值为1,但是原理还是需要去研究一下
        int len=s.size();
        for(int i=0;i<len;i++)//注意这里是从高位向低位传递的
        for(int j=0;j<mod;j++)//列举所有的模后的情况,那么不管你这个数字有多大,你模mod的情况总是有限的,那么把每种mod后结果一样的且满足题意的结果数加起来那就是答案了
        for(int k=0;k<2;k++)
        if(dp[i][j][k]!=0){//自己已经是0了当然不用向后传递了
                int r=(k==0? 9:s[i]-'0');//给出右边界,没有满的时候当然所有的数字都可以列举的了,但是满了的情况下你的上限其实也是确定的
                for(int x=0;x<=r;x++){
                        dp[i+1][(j+cal(x,len-1-i))%mod][(k==1 && x==r)? 1:0] += dp[i][j][k];
                }
        }
        return dp[len][0][0]+dp[len][0][1];
}
int main(){
        string s;
        mi[0]=1;
        for(int i=1;i<12;i++)
                mi[i]=(mi[i-1]*10) % 13;
        while(cin >> s){
                cout<<solve(s)<<endl;
        }
        return 0;
}




下面还有别人的用DFS来写的数位DP,真的也很不错

//    pos    = 当前处理的位置(一般从高位到低位)
//    pre    = 上一个位的数字(更高的那一位)
//    status = 要达到的状态,如果为1则可以认为找到了答案,到时候用来返回,
//            给计数器+1。
//    limit  = 是否受限,也即当前处理这位能否随便取值。如567,当前处理6这位,
//            如果前面取的是4,则当前这位可以取0-9。如果前面取的5,那么当前
//            这位就不能随便取,不然会超出这个数的范围,所以如果前面取5的
//            话此时的limit=1,也就是说当前只可以取0-6。
//
//    用DP数组保存这三个状态是因为往后转移的时候会遇到很多重复的情况。
int    dfs(int pos,int pre,int status,int limit)
{
    //已结搜到尽头,返回"是否找到了答案"这个状态。
    if(pos < 1)
        return    status;

    //DP里保存的是完整的,也即不受限的答案,所以如果满足的话,可以直接返回。
    if(!limit && DP[pos][pre][status] != -1)
        return    DP[pos][pre][status];

    int    end = limit ? DIG[pos] : 9;
    int    ret = 0;
    
    //往下搜的状态表示的很巧妙,status用||是因为如果前面找到了答案那么后面
    //还有没有答案都无所谓了。而limti用&&是因为只有前面受限、当前受限才能
    //推出下一步也受限,比如567,如果是46X的情况,虽然6已经到尽头,但是后面的
    //个位仍然可以随便取,因为百位没受限,所以如果个位要受限,那么前面必须是56。
    //
    //这里用"不要49"一题来做例子。
    for(int i = 0;i <= end;i ++)
        ret += dfs(pos - 1,i,status || (pre == 4 && i == 9),limit && (i == end));

    //DP里保存完整的、取到尽头的数据
    if(!limit)
        DP[pos][pre][status] = ret;

    return    ret;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

门豪杰

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值