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

题目背景
本题由世界上最蒟蒻最辣鸡最撒比的SOL提供。

寂月城网站是完美信息教室的官网。地址：http://191.101.11.174/mgzd 。

题目描述
辣鸡蒟蒻SOL是一个傻逼，他居然觉得数很萌！

好在在他眼里，并不是所有数都是萌的。只有满足“存在长度至少为2的回文子串”的数是萌的——也就是说，101是萌的，因为101本身就是一个回文数；110是萌的，因为包含回文子串11；但是102不是萌的，1201也不是萌的。

现在SOL想知道从l到r的所有整数中有多少个萌数。

由于答案可能很大，所以只需要输出答案对1000000007（10^9+7）的余数。

输入格式
输入包含仅1行，包含两个整数：l、r。

输出格式
输出仅1行，包含一个整数，即为答案。

输入输出样例
输入 #1 复制
1 100
输出 #1 复制
10
输入 #2 复制
100 1000
输出 #2 复制
253
说明/提示
记n为r在10进制下的位数。

对于10%的数据，n <= 3。

对于30%的数据，n <= 6。

对于60%的数据，n <= 9。

对于全部的数据，n <= 1000，l < r。

分析：

直接找是萌数的数字个数比较困难，所以找不是萌数的数字个数，判断是否存在长度大于等于2的回文子串只需要比较当前数字与前两位数字是否相等即可。
dp[i][j][k]:i表示数位，j表示当前数位前一位上的数字，k表示当前数位前两位上的数字
这题需要考虑前导零lead，初始将前两位数字都设置为-1，t1(前一位)，t2（前两位）。只有lead= =1&&i= =0（即当前位依然是前导零），t1=-1，否则t1=i。t2可以直接取t1.
发现题目的l和r非常大，所以用字符串输入。转化为位数的时候要倒着存入。
常规做法是solve（r）-solve（l-1），但是由于这题是字符串形式输入，为了方便，特判 l 是否是萌数，这样直接solve（r）-solve（l）
由于solve（）的结果是不是萌数的个数，所以需要转化成萌数的个数，即r-solve（r），但是r是字符串，则可以对r取模，算得的结果计算（ans-solve（r）+mod）%mod与（r-solve（r）+mod）%mod是一样的。
注意：只有当t1!=-1&&t2!=-1时，才可以用dp的值。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e3+10;
const int mod=1e9+7;
ll dp[N][10][10];
int a[N];
char l[N],r[N];
ll dfs(int pos,int lead,int t1,int t2,int flag)
{
	if(pos<1) return 1;
	if(!lead&&!flag&&t1!=-1&&t2!=-1&&dp[pos][t1][t2]!=-1) return dp[pos][t1][t2];
	int up=flag?a[pos]:9;
	ll res=0;
	for(int i=0;i<=up;i++)
	{
		if(t1==i||t2==i) continue;
		res=(res+dfs(pos-1,lead&&i==0,lead&&i==0?-1:i,t1,flag&&i==a[pos]))%mod; 
	}
	if(!flag&&!lead&&t1!=-1&&t2!=-1)
	{
		dp[pos][t1][t2]=res;
	}
	return res;
}
ll solve(char s[])
{
	int n=strlen(s);
	int k=n;
	for(int i=0;i<n;i++)
	{
		a[k]=s[i]-'0';
		k--;
	}
	return dfs(n,1,-1,-1,1);
}
int check(char s[])
{
	int n=strlen(s);
	for(int i=1;i<n;i++)
	{
		if(s[i]==s[i-1]) return 1;
		if(i>=2&&s[i]==s[i-2]) return 1;
	}
	return 0;
}
ll ji(char s[])
{
	int n=strlen(s);
	ll ans=0;
	for(int i=0;i<n;i++)
	{
		ans=(ans*10%mod+s[i]-'0')%mod;
	}
	return ans;
}
int main()
{
	memset(dp,-1,sizeof(dp));
	scanf("%s%s",l+1,r+1);
	ll ans=((ji(r+1)-solve(r+1)+mod)%mod-((ji(l+1)-solve(l+1)+mod)%mod)+mod)%mod;
	if(check(l+1)) ans=(ans+1)%mod;
	printf("%lld",ans); 
	return 0;
}