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

#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;
}