BZOJ 4521 [CQOI2016]手机号码 - 数位DP

Description

在$[L, R]$找出有几个数满足两个条件 ：

1 ： 不同时含有$4$ 和 $8$

2 ： 至少有$3$个相邻的数相同

Solution

非常容易的数位DP，

$pos$ 为当前第几位， $ex$ 表示是否出现过$4$ 或 $8$， $pre$表示 前面的是几， $num$表示有几个相邻的数相同。

答案就是 $sum[pos][ex][pre][num] $了QuQ。 $DP$ 和 模板一样。

还需要注意如果$L = 1e10$ ， 查$L - 1$ 时必须返回$0$

Code

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<iostream>
 #define ll long long
 using namespace std;

 int a[];
 ll sum[][][][];

 ll dfs(int pos, int ex, int pre, int num, bool lim, bool lead) {
     if(!pos)
         return num >= ;
     if(!lim && !lead && sum[pos][ex][pre][num] != -)
         return sum[pos][ex][pre][num];
     int up = lim ? a[pos] : ;
     ll tmp = ;
     for(int i =  ; i <= up; ++i) {
         if(lead && i == ) continue;
         if(ex ==  && i == )    continue;
         if(ex ==  && i == )    continue;
         int nx;
         if(i == ) nx = ;
         else if(i == ) nx = ;
         else nx = ex;
         if(pre == i && num < )
             tmp += dfs(pos - , nx, i, num + , lim && a[pos] == i, false);
         else if(num >= )
             tmp += dfs(pos - , nx, pre, num, lim && a[pos] == i, false);
         else tmp += dfs(pos - , nx, i, , lim && a[pos] == i, false);
     }
     if(!lim && !lead) sum[pos][ex][pre][num] = tmp;
     return tmp;
 }

 ll work(ll x) {
     if(x < (ll)1e10)
         return ;
     int len = ;
     while(x) {
         a[++len] = x % ;
         x /= ;
     }
     return dfs(, , , , true, true);
 }

 int main()
 {
     ll l, r;
     scanf("%lld%lld", &l, &r);
     memset(sum, -, sizeof(sum));
     printf("%lld\n", work(r) - work(l - ));
 }