[hdu 2089] 不要62 数位dp|dfs 入门

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2089

题意:求[n, m]区间内不含4和62的数字个数。

这题有两种思路,直接数位dp和dfs

数位dp:

  dp[i][j]表示i位数,首位是j的符合要求的数字个数。

  j = 4时    dp[i][j] = 0

  j != 4时

   [hdu 2089] 不要62  数位dp|dfs  入门

  例如求dp[3][2],2xx的个数,2已经确定了,2后面xx的个数即为2xx的个数,只用求出0x, 1x, 2x...9x的个数之和即可。同时要注意限制条件,dp[i][4]均为0,如果i位首位为6,i-1位首位为2的话也为0。这样我们首先预处理下,然后由此可以求区间内所符合要求的数字个数。

  以求[0, 365]为例,先求0xx, 1xx, 2xx, xx的个数即为每个的个数,当然如果是555的的话4xx是跳过的,或者前一位是6,那么2xx也要跳过。然后求[300, 365]的个数,已经确定首位为3,求3xx的个数,然后类似的确定xx的个数。如果遇到456这种情况,只用求0xx,1xx,2xx,3xx的个数,4xx就不用求了,因此最外层循环就可以停止了。类似的6223.. 没必要求[6200,6223]了。

代码:

#include <iostream>
#include <stdio.h>
#include <cstring>
#include <algorithm>
using namespace std; int dp[][];
int d[]; void init()
{
memset(dp, , sizeof(dp));
dp[][] = ;
for(int i = ; i <= ; i++)
for (int j = ; j <= ; j++)
for (int k = ; k <= ; k++)
if (j != && !(j== && k==))
dp[i][j] += dp[i-][k];
} int solve(int n)
{
int ans = , len = ;
while (n) {
d[++len] = n % ;
n /= ;
}
d[len+] = ;
for (int i = len; i >= ; i--) {
for (int j = ; j < d[i]; j++) {
if (d[i+] != || j != )
ans += dp[i][j];
}
if (d[i]== || (d[i+]== && d[i]==))
break;
}
return ans;
} int main()
{
freopen("1.txt", "r", stdin);
int n, m;
init();
while (~scanf("%d%d", &n, &m)) {
if (n + m == ) break;
printf("%d\n", solve(m+)-solve(n));
} return ;
}

DFS+记忆化搜索:

  dp[i][j]表示i位数,前一位数组是否为6的符合要求的个数。

  dfs的参数l是当前的位数,从最高位开始搜索。six是前一位是否为6,limit是最高位是否受限,如365,最高位就受限与0~3,然后开始搜索0xx, 1xx, 2xx, 3xx, 其中0xx,1xx,2xx中的xx都是不受限的,0~99均可取,而3xx中的xx要受65的限制,搜索下一位时继续设限。如果该位为4,或上一位为6,该位为2时就跳过不搜。另外搜索中有大量重复,所以采用记忆化搜索。如果受限的话就不能采用记忆化搜索的结果。

代码:

#include <iostream>
#include <stdio.h>
#include <cstring>
#include <algorithm>
using namespace std; int digit[], dp[][], v[][]; int dfs(int l, bool six, bool limit)
{
if (l == ) return ;
if (!limit && v[l][six]) return dp[l][six];
int len = limit ? digit[l] : ;
int nx = ;
for (int i = ; i <= len; i++) {
if ((i == ) || (six&&i==))
continue;
nx += dfs(l-, i==, limit&&(i==len));
}
if (!limit) {
v[l][six] = true;
dp[l][six] = nx;
}
return nx;
} int sum(int n)
{
memset(dp, , sizeof(dp));
memset(v, , sizeof(v));
int pos = ;
while (n) {
digit[++pos] = n % ;
n /= ;
}
int ans = dfs(pos, false, true);
return ans;
}
int main()
{
//freopen("1.txt", "r", stdin);
int n, m;
while (~scanf("%d%d", &n, &m)) {
if (n + m == ) break;
printf("%d\n", sum(m)-sum(n-));
} return ;
}

ps:注意两者在[n, m]时的处理。

  

  

上一篇:cf701E Connecting Universities


下一篇:hibernate课程 初探单表映射3-1 单一主键