BZOJ4521: [Cqoi2016]手机号码 （数位DP）

就裸数位DP了...比较奇怪的是之前只写了pos和sta两位状态一直不对 于是把dfs的变量都加成状态就莫名过了 不明白呀

#include <stdio.h>
#include <algorithm>
#include <iostream>
#include <string.h>
using namespace std;
typedef long long ll;
int q[20];
ll dp[20][3][11][11][2];

ll dfs(int pos, int sta, int q1, int q2, bool sj, bool f) //数位 4和8的状态 前2位 前1位 上界 连续出现三个数
{
    if(pos == 0) return f;
    if(!sj && dp[pos][sta][q1][q2][f] != -1) return dp[pos][sta][q1][q2][f];

    int up = sj ? q[pos] : 9;
    ll sum = 0;
    for(int i = 0; i <= up; i++)
    {
        if(sta == 1 && i == 4) continue;
        if(sta == 2 && i == 8) continue;

        int s = 0;
        if(sta == 0)
        {
            if(i == 8) s = 1;
            if(i == 4) s = 2;
        }
        else
            s = sta;

        bool ff = false;
        if(i == q1 && i == q2) ff = true;
        sum += dfs(pos - 1, s, q2, i, sj && i == up, f || ff);
    }

    if(!sj) dp[pos][sta][q1][q2][f] = sum;
    return sum;
}

ll solve(ll x)
{
    memset(q, 0, sizeof(q));
    int cnt = 0;
    while(x)
    {
        cnt++;
        q[cnt] = x % 10;
        x /= 10;
    }

    return dfs(11, 0, -1, -1, 1, 0);
}


int main()
{
    ll l, r;
    scanf("%lld%lld", &l, &r);
    memset(dp, -1, sizeof(dp));
    printf("%lld\n", solve(r) - solve(l - 1));
    return 0;
}