不要62 HDU - 2089(数位DP)

最新推荐文章于 2021-03-21 17:41:40 发布

肘子zhouzi

最新推荐文章于 2021-03-21 17:41:40 发布

阅读量148

点赞数 3

文章标签：数位DP

本文链接：https://blog.csdn.net/zhouzi2018/article/details/90111548

版权

跟着最最厉害的老王学了一波
数位DP：求出在给定区间[A,B]内，符合条件p(i)的数的个数，条件p(i)一般与数的大小无关，而只与数的构成有关，A,B一般都很大，需要将区间处理成求解[x,B]-[x-(A-1)]d的个数(x根据问题确定，一般为0)
题意：传送门
题解：首先这个题可以根据打表得出答案

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int INF=0x3f3f3f3f;
const int maxn=1e6+5;
inline int read()
{
    char ch=getchar();
    int f=1,x=0;
    while(!(ch>='0'&&ch<='9')){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-48;ch=getchar();}
    return x*f;
}
bool judge(int n)
{
    while(n){
        if(n%10==4)return false;
        if(n/10%10==6&&n%10==2)return false;
        n/=10;
    }
    return true;
}
int n,m,f[maxn];
int main()
{
    for(int i=0;i<maxn;i++){
        if(judge(i))f[i]=f[i-1]+1;
        else f[i]=f[i-1];
    }
    while(233){
        n=read();m=read();
        if(n==0&&m==0)break;
        printf("%d\n",f[m]-f[n-1]);
    }
    return 0;
}

考虑用数位DP来做，那么就先按照数位来遍历数字，但是这种却无法DP，引入数位DP中按数位遍历数字的方法，用到dfs，当固定前面的数位后，状态就可以传递了，也就是用字典树的思想来处理这个问题，实际上这个问题就是在字典树上进行搜索。

int dfs(int len,bool flag)
{
    if(len==-1)return 1;
    int end=flag?p[len]:9;
    int ans=0;
    for(int i=0;i<=end;i++){
        ans+=dfs(len-1,flag&&i==end);
    }
    return ans;
}
int solve(int n)
{
    int len=0;
    while(n){
        p[len++]=n%10;
        n/=10;
    }
    return dfs(len-1,true);
}

把老王的图附上：

然后在这个字典树上进行统计出符合条件的数即可。

int dfs(int len,bool six,bool flag)
{
    if(len==-1)return 1;
    int end=flag?p[len]:9;
    int ans=0;
    for(int i=0;i<=end;i++){
        if(i==4||(six&&i==2))continue;
        ans+=dfs(len-1,i==6,flag&&i==end);
    }
    return ans;
}

这样的话这个题虽然不会T，但是实际上却是非常耗时的，然后考虑使用记忆化来进行优化，首先进行状态压缩，用dp[len][2]，用这个数组来保存在len长度下，且上一位是不是6的状态的答案个数(还是蛮不好想的)。当flag为true时不用记忆化，是因为在边界上，没有搜索到全部情况，记忆化搜索以后，不仅解决了子问题重叠的问题，多次查询，还大大减少了求解的状态数O(10*2)，对于非常大的数，根据题目进行状态压缩以后，需要求解的状态数也不会很大。
最终代码：

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int INF=0x3f3f3f3f;
inline int read()
{
    char ch=getchar();
    int f=1,x=0;
    while(!(ch>='0'&&ch<='9')){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-48;ch=getchar();}
    return x*f;
}
int n,m,p[10],dp[10][2];
int dfs(int len,bool six,bool flag)
{
    if(len==-1)return 1;
    if(!flag&&dp[len][six])return dp[len][six];
    int end=flag?p[len]:9;
    int ans=0;
    for(int i=0;i<=end;i++){
        if(i==4||(six&&i==2))continue;
        ans+=dfs(len-1,i==6,flag&&i==end);
    }
    if(!flag)dp[len][six]=ans;
    return ans;
}
int solve(int n)
{
    int len=0;
    while(n){
        p[len++]=n%10;
        n/=10;
    }
    return dfs(len-1,false,true);
}
int main()
{
    while(233){
        n=read();m=read();
        if(n==0&&m==0)break;
        printf("%d\n",solve(m)-solve(n-1));
    }
    return 0;
}