10166. 「一本通 5.3 练习 1」数字游戏

(Updated: )
oi
,

题意

给定多组数据，每组数据给定三个数：$a,b,n$表示求在区间$[a,b]$内各位数之和模$n=0$的数的个数。

思路

这是一道数位$DP$的模板题。 设$f[i][S]$表示处理到第$i$位，$S$为和。 $f[i][S]=f[i-1][(S+i)%N](0<=k<=Dim[i] or 9)$ 其中$k$的取值范围根据上一位的状态来定。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
#include<algorithm>
#include<bitset>
#include<complex>
#include<deque>
#include<exception>
#include<fstream>
#include<functional>
#include<iomanip>
#include<ios>
#include<iosfwd>
#include<iostream>
#include<istream>
#include<iterator>
#include<limits>
#include<list>
#include<locale>
#include<map>
#include<memory>
#include<new>
#include<numeric>
#include<ostream>
#include<queue>
#include<set>
#include<sstream>
#include<stack>
#include<stdexcept>
#include<streambuf>
#include<string>
#include<typeinfo>
#include<utility>
#include<valarray>
#include<vector>
#include<cctype>
#include<cerrno>
#include<cfloat>
#include<ciso646>
#include<climits>
#include<clocale>
#include<cmath>
#include<csetjmp>
#include<csignal>
#include<cstdarg>
#include<cstddef>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<ctime>
using namespace std;
inline int read(){
    int res=0,f=1;char ch=getchar();
    while(ch<'0'ch>'9'){if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') res=res*10+ch-'0',ch=getchar();
    return res*f;
}
inline void write(int x){
    if(x<0) putchar('-'),x=-x;
    if(x<10) putchar(x+'0');
    else{
        write(x/10);
        putchar(x%10+'0');
    }
}
//queue<int> q;
//set<int> s;
//priority_queue<int> q1;
//priority_queue<int,vector<int>,greater<int> > q2;
//list<int> l;
//stack<int> s;
int a,b,N,Dim[25],f[555][110];
int dfs(int x,int sum,int bj){//分别表示第几位，和为多少，上一位是否已经大于
    if(x==0){
        if(sum%N==0) return 1;
        else return 0;
    }
    if(bj==0&&f[x][sum]!=-1) return f[x][sum];//记忆化优化
    int res=0,ret=0;
    if(bj==1) res=Dim[x];
    else res=9;//k的范围
    for(int i=0;i<=res;i++)
        ret+=dfs(x-1,(sum+i)%N,bj&&i==res);//DP转移
    if(bj==1) return ret;
    else return f[x][sum]=ret;
}
int Get(int x){
    Dim[0]=0;
    while(x){
        Dim[++Dim[0]]=x%10;
        x/=10;
    }//把该数字拆开
    return dfs(Dim[0],0,1);//数位dp记忆化
}
void reset(){
    for(int i=0;i<=200;i++){
        for(int j=1;j<=110;j++) f[i][j]=-1;//因为每一次的询问的n不同，所以清空
    }
}
int main(){
    while(cin>>a){
        reset();
        b=read();N=read();
        write(Get(b)-Get(a-1));putchar('\n');//数位dp的基本框架
    }
    return 0;
}