【数位dp】P3413 SAC#1 - 萌数

考虑到长度为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;
}