考虑到长度为2的回文串可能为aa,aba这两种情况,我们可以采取正难反解的方法,首先算出所有方案数字的和,然后减去不存在长度大于等于2的回文串的数字的个数,就是答案
实现统计不存在长度大于等于2的回文串的操作,在dfs时,要记录当前位前两位数分别是多少,然后再枚举这一位时,选择不能构成回文串(长度大于等于2)的数字,最后统计方案数
这道题由于输入很大,long long也不够,所以采取了直接读字符串,这样就无法进行-1的操作,所以要进行一下特判,如果那个数满足条件,就需要额外-1
代码
#include <bits/stdc++.h>
const int mod=1e9+7;
const long long maxn=1001;
using namespace std;
string l,r;
long long len[maxn];
long long f[maxn][10][10];
long long dfs(int pos,int pre,int gpre,bool lim,bool zero)
{
if(pos==0) return 1;
if(f[pos][pre][gpre]!=-1&&!lim&&!zero&&pre!=-1&&gpre!=-1) return f[pos][pre][gpre];
long long up=lim?len[pos]:9;
long long ans=0;
for(long long i=0;i<=up;i++)
{
if(i!=pre&&i!=gpre&&!zero)
ans=(ans+dfs(pos-1,i,pre,lim&&i==len[pos],0))%mod;
else if(zero)
ans=(ans+dfs(pos-1,(i==0&&zero)?-1:i,-1,lim&&i==len[pos],i==0&&zero))%mod;
}
if(!lim&&!zero&&pre!=-1&&gpre!=-1) return f[pos][pre][gpre]=ans;
return ans;
}
int main()
{
cin>>l>>r;
memset(f,-1,sizeof f);
long long ansa=0,ansb=0;
long long s1=l.length(),s2=r.length();
for(long long i=0;i<s1;i++)
ansa=(ansa*10%mod+(l[i]^48))%mod;
for(long long i=0;i<s2;i++)
ansb=(ansb*10%mod+(r[i]^48))%mod;
long long s=l.length();
for(long long i=0;i<s;i++)
len[s-i]=l[i]-'0';
long long ans1=dfs(s,-1,-1,1,1);
s=r.length();
for(long long i=0;i<s;i++)
len[s-i]=r[i]-'0';
long long ans2=dfs(s,-1,-1,1,1);
ans1--;
s=l.length();
for(long long i=2;i<=s;i++)
if(l[i]==l[i-1]||(l[i]==l[i-2]&&(i-2>=1)))
{
ans1++;
break;
}
printf("%lld",((ansb-ansa-ans2+ans1+1)%mod+mod)%mod);
return 0;
}