题目背景
本题由世界上最蒟蒻最辣鸡最撒比的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;
}