题目描述
https://vjudge.net/problem/HDU-2089
思路分析
数位dp第一题,方法是打表+按位求ans。
首先题目有点说不清楚,他求的应该是[m,n]闭区间的数字中符合条件的。
先说打表,我们定义一个dp矩阵,dp[i][j]表示i位数,以j为开头数字的符合要求的数的个数。比如:我们以求[0,324]中的数字个数为例子。我们需要处理一下信息:
对于三位数的,ans+=dp[3][0]+dp[3][1]+dp[3][2],我们这里计算的就包括了000~099,100199,200299这些范围内的符合要求的数字。
对于两位数,ans+=dp[2][0]+dp[2][1],这里求的是,在00~09,10~19内符合要求的,但实际上,我们这里求的是建立在假设百位上的3已经固定下来的时候,求后面两位的总情况数,也就是对应实际的300~309,310~319情况的个数。
个位以此类推。
在这个过程中,我们需要排除掉含4和6、2连续的数字,具体的就看代码吧。
void init(){
dp[0][0]=1;
for(int i=1;i<LEN;i++){
for(int j=0;j<10;j++){
if(j==4) continue;
for(int k=0;k<10;k++){
if(j==6&&k==2) continue;
dp[i][j]+=dp[i-1][k];
}
}
}
}
计算ans的时候,我们需要注意:当当前位置变为4的时候,或者两位连续6、2的时候break掉。
ll caculate(ll len,ll*digit){
ll ans=0;
for(ll i=len;i>=1;i--){
for(int j=0;j<digit[i];j++){
//这里有个细节,j!=digit[i],因此,我们输入的右端点的数并没有被计算进去,所以,我们在处理数字的时候,需要先对m+1.
if(digit[i+1]!=6||j!=2){
ans+=dp[i][j];
}
}
if(digit[i]==4||(digit[i+1]==6&&digit[i]==2)){
break;
}
}
return ans;
}
完整代码
#include <iostream>
using namespace std;
typedef long long ll;
const int LEN=12;
ll dp[LEN][10];
void init(){
dp[0][0]=1;
for(int i=1;i<LEN;i++){
for(int j=0;j<10;j++){
if(j==4) continue;
for(int k=0;k<10;k++){
if(j==6&&k==2) continue;
dp[i][j]+=dp[i-1][k];
}
}
}
}
ll caculate(ll len,ll*digit){
ll ans=0;
for(ll i=len;i>=1;i--){
for(int j=0;j<digit[i];j++){
if(digit[i+1]!=6||j!=2){
ans+=dp[i][j];
}
}
if(digit[i]==4||(digit[i+1]==6&&digit[i]==2)){
break;
}
}
return ans;
}
int main()
{
init();
ll n,m;
while(~scanf("%lld %lld",&n,&m)&&m+n){
ll digit1[LEN]={0};
ll digit2[LEN]={0};
ll len1=0,len2=0;
m+=1;
while(n){
digit1[++len1]=n%10;
n/=10;
}
while(m){
digit2[++len2]=m%10;
m/=10;
}
ll ans1=caculate(len1,digit1);
ll ans2=caculate(len2,digit2);
cout<<ans2-ans1<<endl;
}
return 0;
}