洛谷P3413 SAC#1 - 萌数（数位DP）

题目是说回文子串长度>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.