P3413 SAC#1 - 萌数 - ( 数位dp )

题目链接：点击进入

题目

思路

因为 l , r 范围在 1e1000，用字符串处理一下，对于 l - 1 特殊处理 ( 字符串没法直接-1 )
对于 dfs，state 状态设置为是否包含回文字串；
dp [ i ] [ 0 ] 表示 i 位后，不存在回文子串的数量
dp [ i ] [ 1 ] 表示 i 位后，存在回文子串的数量
同时题目要求回文子串必须长度为 2 ，这里记录一个 gpre 参数表示前一位的前一位是什么，用来判断字符串的长度是否超过 1 。
对于 state 状态的更新，考虑当前枚举位 i ,与前一位 pre ,跟前一位的前一位 gpre 的关系来判断
i == pre ，可以表示一个偶数长度的回文子串
i == gpre ，可以表示一个奇数长度的回文子串
（ 记得判断前导 0 ）

代码

//#pragma GCC optimize(3)//O3
//#pragma GCC optimize(2)//O2
#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
#define int long long
//#define double long double
//#define max(a,b) a>b?a:b
//#define min(a,b) a<b?a:b
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const int maxn=2e4+10;
const int Mod=998244353;
const int mod=1e9+7;
const double eps=1e-9;
const int N=1e6+10;
int a[1010];
int dp[1010][2];
int dfs(int pos,int pre,int gpre,int state,bool lead,bool limit)
{
	if(pos==-1) return state;
	if(!limit&&!lead&&dp[pos][state]!=-1) return dp[pos][state];
	int up=limit?a[pos]:9;
	int res=0;
	for(int i=0;i<=up;i++)
		res=(res+dfs(pos-1,i,(!lead)?pre:-1,state||(i==pre&&!lead)||(i==gpre&&!lead),lead&&i==0,limit&&i==a[pos]))%mod;
	if(!limit&&!lead&&gpre!=-1) dp[pos][state]=res;
	return res;
}
int solve(string s)
{
	int pos=0,len=s.size(),l=0;
	while(s[l]=='0'&&l<len) l++;
	if(l==len) l=len-1;
	while(pos<len&&len-pos-1>=l)
	{
		a[pos]=s[len-pos-1]-'0';
		pos++;
	}
	memset(dp,-1,sizeof(dp));
	return dfs(pos-1,-1,-1,0,1,1);
}
string getl(string s)
{
	int pos=s.size()-1;
	while(pos>0&&s[pos]=='0') s[pos]='9',pos--;
	if(s[pos]>'0') s[pos]-=1;
	return s;
}
signed main()																						
{	
//	ios::sync_with_stdio(false);
//	cin.tie(0);cout.tie(0);
	string l,r;
	cin>>l>>r;
	l=getl(l);
	printf("%lld\n",(solve(r)-solve(l)+mod)%mod);
    return 0;
}