1. 程式人生 > >不要62(數位dp普通寫法和模板寫法)

不要62(數位dp普通寫法和模板寫法)

多組資料,每次給定區間[n,m],求在n到m中沒有“62“或“4“的數的個數。

如62315包含62,88914包含4,這兩個數都是不合法的。0<n<=m<1000000

假設我們現在把x分成了a1,a2,...,aL這樣一個數組,長度為L,aL是最高位。

那麼結果實際上就是這樣:長度為L,最高位取[0,aL-1]的所有的符合題意數的和;再加上長度為L-1,最高位取aL,次高位取[0,aL-1-1]的所有符合題意數的和;再加上……;一直到第一位。

#include <iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define inf 0x3f3f3f3f
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
const int maxn=10;
long long f[maxn][10];

void getdp()
{
    f[0][0]=1;
    for (int i=1;i<10;i++)
    {
        for (int j=0;j<10;j++)
        {
            if (j==4) f[i][j]=0;
            else if (j==6)
            {
                for (int k=0;k<10;k++)
                    f[i][j]+=f[i-1][k];
                f[i][j]-=f[i-1][2];
            }
            else
            {
                for (int k=0;k<10;k++)
                    f[i][j]+=f[i-1][k];
            }
        }
    }
}

int a[maxn];
long long solve(int n)
{
    a[0]=0;
    while (n)
    {
        a[++a[0]]=n%10;//a1-an
        n/=10;//求出各個位上的數
    }
    a[a[0]+1]=0;//最高位補0
    long long ans=0;
    for (int i=a[0];i>=1;i--)//a[0]為位數,代表一共幾位;
    {
        for (int j=0;j<a[i];j++)//j代表哪一位;
            if (j!=4 && !(a[i+1]==6 && j==2))
                ans+=f[i][j];//i位數,最高位為j
        if (a[i]==4) break;
        if (a[i+1]==6 && a[i]==2) break;
    }
    return ans;
}

int main()
{
    int n,m;
    getdp();
    while (scanf("%d %d",&n,&m)==2 && (n||m))
    {
        long long k1=solve(m+1);
        long long k2=solve(n);
        //printf("::%d,%d::",k1,k2);
        printf("%I64d\n",k1-k2);
    }
    return 0;
}

模板寫法

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
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;
    //memset(dp,-1,sizeof dp);可優化
    while(~scanf("%d%d",&le,&ri) && le+ri)
    {
        memset(dp,-1,sizeof dp);
        printf("%d\n",solve(ri)-solve(le-1));
    }
    return 0;
}