P3413 SAC#1 - 萌数

题目

洛谷
数位动规用爆搜真好玩

做法

含有回文串实际我们仅需判断是否有\(2/3\)回文串
\(Dfs(now,num,pre,ly,lead,prel,top)\)：
在第\(now\)位
\(now-1\)位，\(now-2\)位
是否合法
\(now-1\)位是否为前导\(0\)，\(now-2\)为是否为前导\(0\)
是否受限

My complete code

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const LL maxn=1009,p=1e9+7;
LL ans;
LL a[maxn],f[maxn][10][10][2][2][2];
char s1[maxn],s2[maxn];
LL Dfs(LL now,LL num,LL pre,LL ly,LL lead,LL prel,LL top){
    if(!now) return ly;
    if(!top && f[now][num][pre][ly][lead][prel]!=-1) return f[now][num][pre][ly][lead][prel];
    LL ret(0);
    LL Up=top?a[now]:9;
    for(LL i=0;i<=Up;++i)
        (ret+=Dfs(now-1,i,num,ly||(!lead&&i==num)||(!prel&&i==pre),lead&&!i,lead,top&&(i==Up))%p)%=p;
    if(!top) f[now][num][pre][ly][lead][prel]=ret;
    return ret;
}
int main(){
    scanf(" %s %s",s1+1,s2+1);
    LL nl(strlen(s1+1)),nr(strlen(s2+1));
    LL tot(0);
    for(LL i=nr;i>=1;--i) a[++tot]=s2[i]-'0';
    memset(f,-1,sizeof(f));
    ans+=Dfs(tot,0,0,0,1,1,1);
    tot=0;
    for(LL i=nl;i>=1;--i) a[++tot]=s1[i]-'0';
    if(nl==1 && a[1]==0){
        cout<<ans;
        return 0;
    }
    --a[1];
    LL now(1);
    while(a[now]<0){
        a[now]=9;
        --a[++now];
    }
    while(a[tot]==0) --tot;
    ans=(ans-Dfs(tot,0,0,0,1,1,1)+p)%p;
    cout<<ans;
    return 0;
}