数位dp不要62

来源:互联网 发布:python sum axis 1 编辑:程序博客网 时间:2024/06/05 21:08
大意:给定区间[n,m],求在n到m中没有“62“或“4“的数的个数。如62315包含62,88914包含4,这两个数都是不合法的。

思路就两种,一个暴力一个数位dp,开始给的样例里的解题思路那个数位dp没看懂
思路:做这道题我们就要用到数思路:打表 ,数据量不大
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 1000000;
int n, m;
int f[maxn+5];
int check(int num) {
while (num) {
if (num % 10 == 4)
return 0;
if (num % 100 == 62)
return 0;
num /= 10;
}
return 1;
}


int cal(int num) {
int ans = 0;
for (int i = 1; i <= num; i++)
if (check(i))
f[i] = f[i-1] + 1;
else f[i] = f[i-1];
return ans;
}


int main() {
memset(f, 0, sizeof(f));
cal(maxn);
while (scanf("%d%d", &n, &m) != EOF && n+m) {
printf("%d\n", f[m]-f[n-1]);
}
return 0;
}


思路:数位DP,注意最后的flag的推断。这个是为了将数字本身算进去


#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
const int maxn = 1000;


/*
* dp[i][0],表示长度为i。不存在不吉利数字
* dp[i][1],表示长度为i,不存在不吉利数字。且最高位为2
* dp[i][2],表示长度为i,存在不吉利数字
*/


int n, m;
int dp[maxn][3];


void init() {
memset(dp, 0, sizeof(dp));
dp[0][0] = 1;
dp[0][1] = dp[0][2] = 0;
for (int i = 1; i < maxn; i++) {
dp[i][0] = dp[i-1][0]*9 - dp[i-1][1];
dp[i][1] = dp[i-1][0];
dp[i][2] = dp[i-1][2]*10 + dp[i-1][1] + dp[i-1][0];
}
}


int cal(int num) {
int len = 0;
int tmp = num;
int bit[maxn];
while (num) {
bit[++len] = num%10;
num /= 10;
}
bit[len+1] = 0;
int ans = 0;
int flag = 0;
for (int i = len; i >= 1; i--) {
ans += dp[i-1][2]*bit[i];
if (flag)
ans += dp[i-1][0]*bit[i];
if (!flag && bit[i] > 4)
ans += dp[i-1][0];
if (!flag && bit[i+1] == 6 && bit[i] > 2)
ans += dp[i][1];
if (!flag && bit[i] > 6)
ans += dp[i-1][1];
if (bit[i] == 4 || (bit[i+1] == 6 && bit[i] == 2))
flag = 1;
}
if (flag)
ans++;
return tmp - ans;
}


int main() {
init();
while (scanf("%d%d", &n, &m) != EOF && n+m) {
printf("%d\n", cal(m)-cal(n-1));
}
return 0;

}

这个数位dp的话写的还行,起码能看懂,要加快速度了,一个数位dp看这么长时间还不算入门。

原创粉丝点击