题目是说回文子串长度>2就是一个萌数,那么只要判断两种情况就好:aa,aba。
状态:dp[i,j,k,state=0~1]表示枚举到第i位数,前一位(即i+1)数字为j,前两位(即i+2)为k时,萌数(1)与非萌数(0)的个数。
边界:state==1? return 1: return 2;
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<iostream>
using namespace std;
typedef long long ll;
const int maxn=1e3+5;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
ll dp[maxn][11][11][2];
int digit[maxn];
char s1[maxn],s2[maxn];
ll dfs(int pos,int pre,int ppre,int state,bool limit,bool lead)
{
if(pos==-1)
{
if(state) return 1;
return 0;
}
if(!limit&&!lead&&dp[pos][ppre][pre][state]!=-1) return dp[pos][ppre][pre][state]%mod;
ll ret=0;
int up= limit? digit[pos]:9;
for(int i=0;i<=up;++i)
{
int p=i;
if(!i&&lead) p=-inf;
ret+=dfs(pos-1,p==-inf? 10:p,pre,state||pre==p||p==ppre,limit&&i==digit[pos],p==-inf)%mod; //注意dfs中pre的变化,已经state的状态变化。
}
if(!limit&&!lead) dp[pos][ppre][pre][state]=ret%mod;
return ret%mod;
}
ll solve(char *s)
{
int pos=0,len=strlen(s);
for(int i=len-1;i>=0;--i) //要反向记录
digit[pos++]=s[i]-'0';
return dfs(--pos,10,10,0,1,1)%mod;
}
int main()
{
cin>>s1>>s2;
if(s1[0]!='0')
{
int len=strlen(s1)-1;
s1[len]--;
while(s1[len]<'0')
{
if(!len) break;
s1[len]='9';
s1[(--len)]--;
}
}
memset(dp,-1,sizeof(dp));
ll ans1=solve(s1),ans2=solve(s2);
if(ans2<ans1) //这里有一个地方要注意,就是因为取模的原因,s2算出来的数ans2可能比s1的小
//这时如果输出就是一个负数,会WA,因此此时要将ans2再加mod,再减ans1(不会影响正确答案的)
{
ans2+=mod;
cout<<(ans2-ans1)%mod<<endl;
}
else cout<<(ans2-ans1)%mod<<endl;
return 0;
}
本题有两个要点:
1.数字要反向记录;
2.最终结果ans2-ans1可能为负数,可将ans2+mod后减ans1.