牛客寒假算法基础集训营3_G处女座和小姐姐（三）（数位dp）

题目链接：https://ac.nowcoder.com/acm/contest/329/G

题目大意：输入两个数l，r，求l到r之前所有不含6的数有几个。

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int a[20];
ll dp[20][20];
 
ll dfs(int pos, int pre, bool limit){
  if(pos == -1) return 1;
  if(!limit && dp[pos][pre] != -1) return dp[pos][pre];
  int up = limit ? a[pos] : 9;
  ll ans = 0;
  for(int i=0;i<=up;i++){
    if(i == 6)continue;
    ans += dfs(pos - 1, i, limit && i == a[pos]);
  }
  return limit ? ans : dp[pos][pre] = ans;
}
 
ll solve(ll x){
  int pos = 0;
  while(x){
    a[pos++] = x % 10;
    x /= 10;
  }
  return dfs(pos - 1, 0, true);
}
 
int main()
{
  ll l, r;
  scanf("%lld%lld",&l, &r);
  memset(dp,-1,sizeof(dp));
  ll xx = r - l + 1;
  ll yy = solve(r) - solve(l - 1);
  printf("%lld\n",xx - yy);
  return 0;
}

 

---

 

另一道题，如果是不要4和62。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[20];
int dp[20][2];
int dfs(int pos,int pre,int sta,bool limit)
{
    if(pos==-1) return 1;
    if(!limit && dp[pos][sta]!=-1) return dp[pos][sta];
    int up=limit ? a[pos] : 9;
    int tmp=0;
    for(int i=0;i<=up;i++){
        if(pre==6 && i==2)continue;
        if(i==4) continue;
        tmp+=dfs(pos-1,i,i==6,limit && i==a[pos]);
    }
    if(!limit) dp[pos][sta]=tmp;
    return tmp;
}
int solve(int x)
{
    int pos=0;
    while(x){
        a[pos++]=x%10;
        x/=10;
    }
    return dfs(pos-1,-1,0,true);
}
int main()
{
    int le,ri;
    while(~scanf("%d%d",&le,&ri) && le+ri){
        memset(dp,-1,sizeof dp);
        printf("%d\n",solve(ri)-solve(le-1));
    }
    return 0;
}