hdu 3555 Bomb（数位dp）

传送门

题意：问1~n的范围内有多少个数包含连续数位“49”。

题解：数位dp，重点考虑哪些状态是可以直接记下来的。如果定义f[pos][pre]来表示搜到第pos位，上一位是pre时之后的填法，那么可以此时不能确定49是否出现过，所以会WA。如果定义f[pos][pre][found]表示搜到第pos位，上一位是pre，当前出现过49（found == true）或者没出现过的填法，那么记忆化效率不高，会T。因为如果出现过49，那之后填什么效果都一样，可以压成一个状态直接return。所以可以定义f[pos][st]：

st == 0：还没出现过49且上一位不是4

st == 1：还没出现过49但上一位是4

st == 2：出现过49

这种状态设计可以将出现过49后的状态充分压缩，提高搜索效率。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
int l,r;
int a[20],tot;
ll f[20][3];
inline ll dfs(int pos,int st,bool lim) {
	if (pos<0) return st==2;
	if (!lim&&~f[pos][st]) return f[pos][st];
	ll ret=0;
	int up=lim?a[pos]:9;
	for (int i=0;i<=up;++i) {
		if (st==2||(st==1&&i==9)) ret+=dfs(pos-1,2,lim&&i==a[pos]);
		else if (i==4) ret+=dfs(pos-1,1,lim&&i==a[pos]);
		else ret+=dfs(pos-1,0,lim&&i==a[pos]);
	}
	if (!lim) f[pos][st]=ret;
	return ret;
}
inline void init() {
	tot=0;
}
inline ll calc(ll x) {
	init();
	while (x) {
		a[tot++]=x%10;
		x/=10;
	}
	return dfs(tot-1,0,true);
}
int main() {
	memset(f,-1,sizeof(f));
	int kase;
	scanf("%d",&kase);
	while (kase--) {
		ll x;
		scanf("%lld",&x);
		printf("%lld\n",calc(x));
	}
	return 0;
}