[CQOI2016]手机号码 题解

· · 题解

是不是所有的数位DP都长得一样? (也许是我太菜,只做过最简单的?)

这题也一样,只不过记得东西变了变,直接记搜即可。

代码真的很短,注释在代码里。

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

typedef long long ll;
int num[12];
ll l,r,dp[11][11][11][2][2][2][2];

ll f(int p,int a,int b,bool c,bool d,bool _4,bool _8){
    /*
        a:p+1位是多少
        b:p+2位是多少,若是前导0的话是-1
        c:之前是否已经出现连续三个相同的数字
        d:之前是否已经保证x<n
        _4:是否出现过4
        _8:是否出现过8
    */
    if(_4&&_8) return 0;//又有4又有8 不合法 
    if(p<=0) return c;
    //这里是个偷懒,如果当前出现过连续三个相同的数字返回1,否则返回0 
    if (~dp[p][a][b][c][d][_4][_8]) return dp[p][a][b][c][d][_4][_8];//记忆化 
    ll res=0;int lim=!d?num[p]:9;//如果没有严格小于,只能填到num,如果已经严格小于,就可以随便填 
    for(int i=0;i<=lim;i++)
        res+=f(p-1,i,a,c||(i==b&&i==a),d||(i<lim),_4||(i==4),_8||(i==8));
        //下一位,
        //下一位的上一位(就是这一位),
        //下一位的上上位(就是现在的上一位),
        //以前就出现过||这次填数刚造出来的, 
    return dp[p][a][b][c][d][_4][_8]=res;
}

ll calc(ll x){
    if(x<1e10) return 0;//不是手机号 
    memset(dp,-1,sizeof(dp));
    //拆位 
    int len;
    for(len=0;x;x/=10) num[++len]=x%10;
    //第一位还要考虑前导0,不如直接手算 
    ll res=0;
    for(int i=1;i<=num[len];i++)
    res+=f(10,i,0,0,i<num[len],i==4,i==8);
    return res;
}

int main(){
    cin>>l>>r;
    cout<<calc(r)-calc(l-1);
}
/*
24014095542
96045812868
*/

无注释版

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int num[12],len;
ll l,r,dp[11][11][11][2][2][2][2];
ll f(int p,int a,int b,bool c,bool d,bool _4,bool _8){
    if(_4&&_8) return 0;
    if(p<=0) return c;
    if (~dp[p][a][b][c][d][_4][_8]) return dp[p][a][b][c][d][_4][_8];
    ll res=0;int lim=!d?num[p]:9;
    for(int i=0;i<=lim;i++)
        res+=f(p-1,i,a,c||(i==b&&i==a),d||(i<lim),_4||(i==4),_8||(i==8));
    return dp[p][a][b][c][d][_4][_8]=res;
}
ll calc(ll x){
    if(x<1e10) return 0;
    memset(dp,-1,sizeof(dp));
    for(len=0;x;x/=10) num[++len]=x%10;
    ll res=0;
    for(int i=1;i<=num[len];i++)
    res+=f(10,i,0,0,i<num[len],i==4,i==8);
    return res;
}
int main(){
    cin>>l>>r;
    cout<<calc(r)-calc(l-1);
}