LuoguP3413(数位dp)

链接：点击打开链接
题意：只有满足“存在长度至少为2的回文子串”的数是萌的——也就是说，101是萌的，因为101本身就是一个回文数；110是萌的，因为包含回文子串11；但是102不是萌的，1201也不是萌的。现在SOL想知道从l到r的所有整数中有多少个萌数。由于答案可能很大，所以只需要输出答案对1000000007（10^9+7）的余数。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const ll siz=1005;
const ll mod=1e9+7;

ll a[siz],dp[siz][15][15][2];
ll dfs(ll pos,ll pre,ll ppre,ll zero,ll lim){
    ll i,u,ans=0;
    if(pos==0)
    return 1;
    if(lim==0&&pre!=-1&&ppre!=-1&&dp[pos][pre][ppre][zero]!=-1)
    return dp[pos][pre][ppre][zero];
    u=lim?a[pos]:9;
    for(i=0;i<=u;i++){
        if(zero==1){
            if(i==0)
            ans=(ans+dfs(pos-1,-1,-1,1,lim&(i==u)))%mod;
            else
            ans=(ans+dfs(pos-1,i,-1,0,lim&(i==u)))%mod;
        }
        else{
            if(i==pre||i==ppre)
            continue;
            ans=(ans+dfs(pos-1,i,pre,0,lim&(i==u)))%mod;
        }
    }
    if(lim==0)
    dp[pos][pre][ppre][zero]=ans;
    return ans;
}
ll change(string s){
    ll i,x;
    x=0;
    for(i=0;s[i];i++)
    x=(x*10%mod+s[i]-'0')%mod;
    return x;
}
ll cal(string s){
    ll i,pos;
    pos=s.size();
    for(i=1;i<=pos;i++)
    a[i]=s[pos-i]-'0';
    return dfs(pos,-1,-1,1,1);
}                                               //记忆化搜素直接记录前两位，判断不含回文的
int main(){
    ll i,u,v,ans,sig;
    string ll,rr;
    cin>>ll>>rr;
    memset(dp,-1,sizeof(dp));
    u=cal(ll),v=cal(rr);
    sig=1;
    for(i=1;ll[i];i++){                         //判断一下最左端是否符合条件
        if(i-2>=0){
            if(ll[i]==ll[i-1]||ll[i]==ll[i-2]){
                sig=0;
                break;
            }
        }
        else{
            if(ll[i]==ll[i-1]||ll[i]==ll[i-2]){
                sig=0;
                break;
            }
        }
    }
    u=(u-sig+mod)%mod;
    ans=((change(rr)-change(ll)+1+mod)%mod-(v-u+mod)%mod+mod)%mod;
    printf("%lld\n",ans);
    return 0;
}
/*
1 1000000000000000000000000000000000
*/