poj3286 数位dp

最新推荐文章于 2019-10-03 15:50:13 发布

谢符宝

最新推荐文章于 2019-10-03 15:50:13 发布

阅读量759

点赞数

分类专栏：数位dp

本文链接：https://blog.csdn.net/xiefubao/article/details/20666317

版权

数位dp 专栏收录该内容

10 篇文章 0 订阅

订阅专栏

题意：求a-b之间的数0出现的总个数；

解法：compute(int k)是求0-k的0出现的个数；

dp[i][j][allZ][equ]表示第i位取j，allZ表示前边是否全部填充为0，equ表示前边是否与原数的高位全部相等（即决定此位的限制上界）；调了好久，完全是模仿dfs的模版乱套，还是不熟练，实在欠练！

代码：

/****************************************************
* author:xiefubao
*******************************************************/
#pragma comment(linker, "/STACK:102400000,102400000")
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <queue>
#include <vector>
#include <algorithm>
#include <cmath>
#include <map>
#include <set>
#include <stack>
#include <string.h>

using namespace std;

#define eps 1e-8
typedef long long LL;

LL num1[200];
char s[200];
LL dp[30][10][2][2];
LL m;
LL dfs(int pos,int num,bool allZ,bool equ)
{
    if(pos==0) return num==0;
    if(dp[pos][num][allZ][equ]!=-1) return dp[pos][num][allZ][equ];
    if(!equ&&!(allZ&&(num==0))) return num1[pos-1]*pos+(num==0? num1[pos] : 0);
    int end1=equ?s[pos-1]:9;
    LL ans=0;
    
    if(num==0&&equ&&!allZ) ans=m%num1[pos]+1;
    else if(num==0&&!equ&&!allZ) ans=num1[pos-1]*pos;
    
        for(int i=0;i<=end1;i++){
            ans+=dfs(pos-1,i,allZ&&(i==0),equ&&(i==end1));
        }
    return dp[pos][num][allZ][equ]=ans;
}
LL compute(LL a)
{
    if(a==0) return 1;
    if(a<0) return 0;
    m=a;
    memset(s,0,sizeof s);
    memset(dp,-1,sizeof dp);
    int p=0;
    while(a){
        s[p++]=a%10;a/=10;
    }
    LL ans=dfs(p-1,s[p-1],0,1);
    for(int i=0;i<s[p-1];i++){
        ans+=dfs(p-1,i,i==0,0);
    }
    return ans;
}

int main()
{
  //freopen ("in.txt" , "r" , stdin);
  LL a,b;
  num1[0]=1;
  for(int i=1;i<=16;i++) num1[i]=num1[i-1]*10;
  while(scanf("%I64d%I64d",&a,&b)!=EOF)
  {
      if(a==-1&&b==-1) break;
      if(a<b) swap(a,b);
      printf("%I64d\n",compute(a)-compute(b-1));
  }
   return 0;
}