题目链接:点击打开链接
题意:给你一个数字n, 求1~n所有数中包含子串49的个数。
思路:典型的数位DP水题, 做过一两道数位DP后这道题就很简单了。 把数字n当成字符串读进来, 用d[i][last][p][cur]表示当前到了第i位, 上一位是last,当前和n相等还是小于n, 是否已经包含49的方案数。 然后按照意义转移就行了。
细节参见代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
using namespace std;
typedef long long ll;
typedef long double ld;
const ld eps = 1e-9, PI = 3.1415926535897932384626433832795;
const int mod = 1000000000 + 7;
const int INF = 0x3f3f3f3f;
// & 0x7FFFFFFF
const int seed = 131;
const ll INF64 = ll(1e18);
const int maxn = 100;
int T,n,m,len,kase=0,vis[maxn][20][3][2];
ll d[maxn][20][3][2];
char s[100];
ll dp(int i, int last, int p, int cur) {
ll& ans = d[i][last][p][cur];
if(i == len) return cur;
if(vis[i][last][p][cur] == kase) return ans;
vis[i][last][p][cur] = kase;
ans = 0;
int v = s[i] - '0';
for(int j=0;j<=9;j++) {
if(p == 0) {
if(j == v) {
if(last == 4 && j == 9) ans += dp(i+1, j, 0, 1);
else ans += dp(i+1, j, 0, cur);
}
else if(j < v) {
if(last == 4 && j == 9) ans += dp(i+1, j, 1, 1);
else ans += dp(i+1, j, 1, cur);
}
}
else {
if(cur) ans += dp(i+1, j, 1, 1);
else if(last == 4 && j == 9) ans += dp(i+1, j, 1, 1);
else ans += dp(i+1, j, 1, 0);
}
}
return ans;
}
int main() {
scanf("%d",&T);
while(T--) {
scanf("%s",s);
len = strlen(s);
++kase;
printf("%I64d\n",dp(0, 0, 0, 0));
}
return 0;
}