洛谷P3413 SAC#1 - 萌数【数位DP】

时空限制 1000ms / 128MB

题目描述

辣鸡蒟蒻SOL是一个傻逼,他居然觉得数很萌!

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

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

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

输入格式:

输入包含仅1行,包含两个整数:l、r。

输出格式:

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

说明

记n为r在10进制下的位数。
对于10%的数据,n <= 3。
对于30%的数据,n <= 6。
对于60%的数据,n <= 9。
对于全部的数据,n <= 1000,l < r


题目分析

复习了一下总是换汤不换药的数位DP
d p [ l e n ] [ 0 / 1 ] [ l a s t 1 ] [ l a s t 2 ] dp[len][0/1][last1][last2] dp[len][0/1][last1][last2]表示在len数位,当前是否已出现回文串,上一位和前一位分别为last1,last2时的合法数字
按数位DP套路记搜即可
处理字符串的-1比DP还麻烦

#include<iostream>
#include<cmath>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long lt;
typedef double dd;
 
int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}
 
const lt mod=1e9+7;
const int maxn=1010;
char L[maxn],R[maxn];
lt dp[maxn][2][10][10];
int dig[maxn];

lt DP(int len,bool pre,int judge,int d,int last1,int last2)
{
    if(len==0) return d==1;
    if(!pre&&!judge&&dp[len][d][last1][last2]!=-1)
    return dp[len][d][last1][last2];
    
    int res=0,mx=pre?dig[len]:9;
    for(int i=0;i<=mx;i++) 
    {	
        int nxtd=d||(!judge&&i==last1)||(!judge&&i==last2);
        res+=DP(len-1,pre&&i==mx,judge&&i==0,nxtd,i,judge?-1:last1);
        res%=mod;
    }
    if(!pre&&!judge&&last2!=-1) dp[len][d][last1][last2]=res;
    return res;
}

int getd(char* num,int d)
{
    //int len=0,lim=strlen(num);
    int len=strlen(num);
    for(int i=0;i<len;++i)
    dig[i+1]=num[len-i-1]-'0';
    
    if(d==1){
        if(dig[1]!=0) dig[1]--;
        else{
            int k=1;
            while(dig[k]==0) dig[k++]=9;
            dig[k]--;
        }
    }
    return len;
}

lt solve(char* ss,int d)
{
    int len=getd(ss,d);
    memset(dp,-1,sizeof(dp));
    return DP(len,1,1,0,-1,-1);
}

int main()
{
    scanf("%s%s",L,R);
    lt ans=solve(R,0)-solve(L,1);
    printf("%lld",(ans%mod+mod)%mod);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值