代码:萌数

萌数
链接:https://www.luogu.com.cn/problem/P3413
来源:洛谷

题目描述:
辣鸡蒟蒻SOL是一个傻逼，他居然觉得数很萌！
好在在他眼里，并不是所有数都是萌的。只有满足“存在长度至少为2的回文子串”的数是萌的——也就是说，101是萌的，因为101本身就是一个回文数；110是萌的，因为包含回文子串11；但是102不是萌的，1201也不是萌的。
现在SOL想知道从l到r的所有整数中有多少个萌数。
由于答案可能很大，所以只需要输出答案对1000000007（10^9+7）的余数。

输入格式:
输入包含仅1行，包含两个整数：l、r。

输出格式:
输出仅1行，包含一个整数，即为答案。

#include <bits/stdc++.h>
using namespace std;
const int mod = 1e9 + 7;
long long dp[1002][10][2];
int num[1002];
long long dfs(int pos, int pre1, int pre2, int state, bool lead, bool limit)
//数位DP模板
{
    if (!pos)
        return state;
    if (pre1 != -1 && pre2 != -1 && !lead && !limit && dp[pos][pre1][state] != -1)
        return dp[pos][pre1][state];
    int up = limit ? num[pos] : 9;
    long long ans = 0;
    for (int i = 0; i <= up; i++)
    {
        int temp = (lead && i == 0) ? -1 : i;
        ans += dfs(pos - 1, temp, pre1, state || (temp != -1 && (temp == pre1 || temp == pre2)), lead && temp == -1, limit && (temp == num[pos])) % mod;
        ans %= mod;
    }
    if (pre1 != -1 && pre2 != -1 && !lead && !limit)
        dp[pos][pre1][state] = ans;
    return ans;
}
long long solve(string str, bool judge)
//高精度处理
{
    memset(dp, -1, sizeof(dp));
    int len;
    len = str.size();
    for (int i = 1; i <= len; i++)
        num[len - i + 1] = str[i - 1] - '0';
    if (judge)
    //对于L而言，我们需要用高精度算法求得L-1的值
    {
        int pos = 1;
        while (num[pos] == 0)
            num[pos++] = 9;
        num[pos]--;
        while (!num[len] && len > 0)
            len--;
    }
    return dfs(len, -1, -1, 0, true, true);
}
int main()
{
    string L, R;
    cin >> L >> R;
    cout << (solve(R, false) % mod - solve(L, true) % mod + mod) % mod << endl;
    //前缀和处理
    return 0;
}