【HDU3555】Bomb-数位DP入门题

Maxwei_wzj

于 2017-02-23 23:26:29 发布

阅读量301

点赞数

分类专栏：动态规划-数位DP

本文链接：https://blog.csdn.net/Maxwei_wzj/article/details/56689311

版权

动态规划-数位DP 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

测试地址：Bomb

题目大意：求区间[0,n]中包含“49”的数字个数。

做法：参照这位的代码写的，做法也写在里面了，这里就不赘述了：Bomb，WA了好久，结果发现bit数组没初始化......囧得要死，本质上就是一道数位DP的入门题，不是很难，理解了很简单。

以下是本人代码：

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#define ll unsigned long long
using namespace std;
int T,bit[25];
ll a,f[25][3]={0};

void init()
{
  f[0][0]=1;
  for(int i=1;i<=20;i++)
  {
	f[i][0]=f[i-1][0]*10-f[i-1][1];
	f[i][1]=f[i-1][0];
	f[i][2]=f[i-1][2]*10+f[i-1][1];
  }
}

int main()
{
  scanf("%d",&T);
  
  init();
  for(int i=1;i<=T;i++)
  {
    scanf("%I64d",&a);
	memset(bit,0,sizeof(bit));
	a++;
	int len=0;
    while(a)
    {
      bit[++len]=a%10;
	  a/=10;
    }
    ll ans=0;
    bool flag=0;
    for(int i=len;i>=1;i--)
    {
      ans+=f[i-1][2]*bit[i];
	  if (flag) ans+=f[i-1][0]*bit[i];
	  else if (bit[i]>4) ans+=f[i-1][1];
	  if (bit[i+1]==4&&bit[i]==9) flag=1;
    }
	printf("%I64d\n",ans);
  }

  return 0;
}