[数位dp] Jzoj P3316 非回文数字
题解
- f[len][pre][pre2][zero1][zero2][0/1]表示第len位,第len-1位是pre,第len-2位是pre2,第len-1位是否是前导零,第len-2位是否是前导零,是否到达上界的方案数
- 然后记忆化深搜就可以过了
代码
1 #include <cstdio> 2 #include <iostream> 3 #include <cstring> 4 #define ll long long 5 using namespace std; 6 ll a,b,len,f[20][10][10][2][2][2],s[20]; 7 ll dp(int l,int fi,int se,bool bz1,bool bz2,bool bz) 8 { 9 if (l>len) return 1; 10 if (f[l][fi][se][bz1][bz2][bz]) return f[l][fi][se][bz1][bz2][bz]; 11 ll sum=0; 12 if (bz) 13 { 14 for (int i=0;i<=s[l];i++) if ((bz1||i!=fi)&&(bz2||i!=se)) sum+=dp(l+1,i,fi,bz1&&(i==0),bz1,bz&&i==s[l]); 15 return f[l][fi][se][bz1][bz2][bz]=sum; 16 } 17 else 18 { 19 for (int i=0;i<=9;i++) if ((bz1||i!=fi)&&(bz2||i!=se)) sum+=dp(l+1,i,fi,bz1&&(i==0),bz1,0); 20 return f[l][fi][se][bz1][bz2][bz]=sum; 21 } 22 } 23 ll calc(ll x) 24 { 25 memset(f,0,sizeof(f)),len=0; 26 if (x<0) return 0; 27 while (x) s[++len]=x%10,x/=10; 28 for (int i=1;i<=len/2;i++) swap(s[i],s[len-i+1]); 29 return dp(1,0,0,1,1,1); 30 } 31 int main() 32 { 33 scanf("%lld%lld",&a,&b),a--,printf("%lld",calc(b)-calc(a)); 34 }