HDU-3555-数位DP-Bomb

题意
求1-N中含49的个数

Hint
From 1 to 500, the numbers that include the sub-sequence “49” are “49”,”149”,”249”,”349”,”449”,”490”,”491”,”492”,”493”,”494”,”495”,”496”,”497”,”498”,”499”, so the answer is 15.

思路
数位DP

// pos = 当前处理的位置（一般从高位到地位）
// pre = 上一个位的数字（更高的那一位）
// status = 要达到的状态 如果为1 则可以认为找了答案，到时候用来返回，
// 给计数器+1
//limit = 是否受限，也即当前处理这位能否随便取值。
//
ll dfs(int pos,int pre,int status,int limit){ 
    //如果搜索到尽头 返回是否找到了这个答案的状态
    if(pos < 1){
        return status ;
    }
    //DP里保存的时完整的，且不受限的答案，如果满足的话 可以直接返回
    if(!limit && dp[pos][pre][status]!=-1){
        return dp[pos][pre][status];
    }
    // 高位是否受限 返回结束标志
    int end = limit ?DIG[pos] : 9;
    ll ret = 0;
    //status 用||是因为如果前面找到答案那么后门还有没有答案就
    //无所谓了
    //limit 用&&是因为后面受限 必须由前面受限而且自身也受限推出
    for(int i = 0;i <=end ;i++){
        ret += dfs(pos -1 ,i ,status||(pre==4&&i ==9),limit && (i==end));
    }
    if(!limit){
        dp[pos][pre][status] = ret;
    }
    return ret;
}

完整代码

/************************************************************************* > File Name: mian.cpp > Author:Chazz > Created Time: 2015年11月03日 星期二 18时47分25秒 ************************************************************************/

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <string.h>
#include <algorithm>
#include <vector>
typedef long long ll;
using namespace std;
ll dp[20][20][2];
int DIG[20];
// pos = 当前处理的位置（一般从高位到地位）
// pre = 上一个位的数字（更高的那一位）
// status = 要达到的状态 如果为1 则可以认为找了答案，到时候用来返回，
// 给计数器+1
//limit = 是否受限，也即当前处理这位能否随便取值。 
ll dfs(int pos,int pre,int status,int limit){ 
    //如果搜索到尽头 返回是否找到了这个答案的状态
    if(pos < 1){
        return status ;
    }
    //DP里保存的时完整的，且不受限的答案，如果满足的话 可以直接返回
    if(!limit && dp[pos][pre][status]!=-1){
        return dp[pos][pre][status];
    }
    // 高位是否受限 返回结束标志
    int end = limit ?DIG[pos] : 9;
    ll ret = 0;
    //status 用||是因为如果前面找到答案那么后门还有没有答案就
    //无所谓了
    //limit 用&&是因为后面受限 必须由前面受限而且自身也受限推出
    for(int i = 0;i <=end ;i++){
        ret += dfs(pos -1 ,i ,status||(pre==4&&i ==9),limit && (i==end));
    }
    if(!limit){
        dp[pos][pre][status] = ret;
    }
    return ret;
}
ll solve(ll n){
    int len = 0;
    while(n){
        DIG[++len] = n % 10;
        n = n/10;
    }
  // printf("len : %d\n",len);
  // DIG[len+1] = 0;
    return dfs(len,0,0,1);
}
int main(void){
    int t;
    cin>>t;
    ll n;
    memset(dp,-1,sizeof(dp));
    while(t--){
        scanf("%lld",&n);
        printf("%lld\n",solve(n));
    }
    return 0 ;
}