题目描述
1085 不要62
算法
数位dp
xswl,看提高课试听视频的时候,看y总写这题,last没更新,bug找了好久....
C++ 代码
#include<iostream>
#include<vector>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=11,M=110;
long long A,B,P;
int f[N][10];
void init(){
memset(f,0,sizeof f);
for(int i=0;i<=9;i++) f[1][i]++;
f[1][4]=0;
for(int i=2;i<N;i++){
for(int j=0;j<10;j++){
if(j==4) continue;
for(int k=0;k<10;k++){
if(k==4||(j==6&&k==2)) continue;
f[i][j]+=f[i-1][k];
}
}
}
}
int dp(int n){
if(!n) return 1;
vector<int> nums;
while(n) nums.push_back(n%10),n/=10;
int res=0,last=0;
for(int i=nums.size()-1;i>=0;i--){
int x=nums[i];
for(int j=0;j<x;j++){
if(j==4) continue;
if(last==6&&j==2) continue;
res+=f[i+1][j];
}
if(x==4) break;
if(last==6&&x==2) break;
last=x;
if(!i&&last!=4) res++;
}
return res;
}
int main(){
init();
while(cin>>A>>B,A||B){
cout<<dp(B)-dp(A-1)<<endl;
}
return 0;
}