不要62(數位dp普通寫法和模板寫法)
阿新 • • 發佈:2019-02-07
多組資料,每次給定區間[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; }