首页 > 代码库 > [ACM] hdu 3555 Bomb (数位DP,统计1-N中含有“49”的总数)
[ACM] hdu 3555 Bomb (数位DP,统计1-N中含有“49”的总数)
Bomb
Now the counter-terrorist knows the number N. They want to know the final points of the power. Can you help them?
The input terminates by end of file marker.
3 1 50 500
0 1 15HintFrom 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.
题意就是求从整数1到N中有多少个含有“49”的数? 比如N=500,那么 "49","149","249","349","449","490","491","492","493","494","495","496","497","498","499",
so the answer is 15.
做这种数位DP的题,一开始除了暴力解决,没有一点数位DP的思路。。然后又去网上找解题报告,结果看了好多篇,完全都不知道在说什么。。。。。
后来用谷歌搜到了一篇不错的解题报告:http://www.cnblogs.com/liuxueyang/archive/2013/04/14/3020032.html
先说一下总体思路:
假设统计N=591时,那么按以下的顺序进行统计:
1~499 确定5这一位,统计的数比它小
500~589 确定9这一位 ,统计的数比它小
590 确定1这一位,统计的数比它小
最后判断一下自身是不是符合 即591
循环三次就把符合题意的数的总数全都求出来了,这就是本题的数位DP的奥妙之处.
再比如 1249
1~999
1000~1199
1200~1239
1240~1248
1249
dp[i][j] 表示长度为i的数(也就是有i位数)状态为j的数的总数有多少
本题状态有三种:
①dp[i][0]代表长度为i且不包含49的数有多少个
②dp[i][1]代表长度为i且不包含49且左边第一位(最高位)为9的数有多少个
③dp[i][2]代表长度为i且包含49的数有多少个
打表预处理,0<=i<=21(21位就够了),主要是处理状态的转移
dp[i][0]=dp[i-1][0]*10-dp[i-1][1]; //dp[i][0]高位随便加一个数字都可以,但是会出现49XXX的情况,要减去
dp[i][1]=dp[i-1][0]; //在不含49的情况下高位加9
dp[i][2]=dp[i-1][2]*10+dp[i-1][1]; //在含有49的情况下高位随便加一位或者不含49但高位是9,在前面最高位加上4就可以了
贴一下统计的代码:
for(int i=len;i>=1;i--)//每次确定一位 { ans+=dp[i-1][2]*bit[i];//低位中含有49,高位随便一个1,2,3....bit[i]都可以,bit[i]是代表有几个数字,比如bit[i]=5,那么代表有五个数字,0,1,2,3,4,比5小。 if(!has) { if(bit[i]>4) ans+=dp[i-1][1];//低位中高位是9,前面加上4就可以了 } else ans+=(dp[i-1][0])*bit[i];//如果有49,就随便选了,比如 495的时候,有490 491 492 493 494 //上面这句话困扰了我一天多的时间。为什么不写(d[i-1][0]+dp[i-1][2])*bit[i]呢,前面已经出现过49 //那么低位任意选择都可以,dp[i-1][0]是那些低位不出现49的,dp[i-1][2]是那些低位出现49的,按理说应该 //加上啊,BUT!!!清注意循环里面的第一句ans+=dp[i-1][2]*bit[i]; 前面已经加上了dp[i-1][2]低位有 //49的情况了,哎,欲哭无泪........ if(bit[i+1]==4&&bit[i]==9) has=true; } if(has) ans++;len为数的长度,bit[i]代表从右向左第几位(从1开始)的数字.
弄明白这个统计代码花费了太多时间
对于第i位,说长度为i比较好理解,先加上长度为i-1的所有数中包含49的数,即 ans+=dp[i-1][2]*bit[i]; 为什么乘以bit[i]呢? 比如 bit[i]=5, 那么小于5的数有5种选择,即0,1,2,3,4 ,对于每种选择长度为i-1的数中都有dp[i-1][2]个符合题意的数,所以有 dp[i-1][2]*bit[i] , 所以就求出了第i位比5小的所有数,对于590这个数来说,就是求出了
1~499这些数中一部分有多少符合题意的数。还有i-1位数中如果高位是9,那么第i位只要是4也可以符合题意,所以有了 if(bit[i] >4) ans += dp[i-1][1];
代码:
#include <iostream> #include <string.h> using namespace std; //dp[i][0]代表长度为i不含49的数 //dp[i][1]代表长度为i不含49且最高位为9的数 //dp[i][2]代表长度为i含有49的数 long long dp[22][3]; int bit[21]; long long n; void init() { dp[0][0]=1,dp[0][1]=0,dp[0][2]=0; for(int i=1;i<=21;i++) { dp[i][0]=dp[i-1][0]*10-dp[i-1][1];//dp[i][0]高位随便加一个数字都可以,但是会出现49XXX的情况,要减去 dp[i][1]=dp[i-1][0];//在不含49的情况下高位加9 dp[i][2]=dp[i-1][2]*10+dp[i-1][1];//在含有49的情况下高位随便一位或者不含49但高位是9,在前面最高位加上4就可以了 } } long long cal(long long n) { int len=0; while(n) { bit[++len]=n%10; n/=10; } long long ans=0; bit[len+1]=0; bool has=false; for(int i=len;i>=1;i--)//每次确定一位 { ans+=dp[i-1][2]*bit[i];//低位中含有49,高位随便一个1,2,3....bit[i]都可以,bit[i]是代表有几个数字,比如bit[i]=5,那么代表有五个数字,0,1,2,3,4,比5小。 if(!has) { if(bit[i]>4) ans+=dp[i-1][1];//低位中高位是9,前面加上4就可以了 } else ans+=(dp[i-1][0])*bit[i];//如果有49,就随便选了,比如 495的时候,有490 491 492 493 494 //上面这句话困扰了我一天多的时间。为什么不写(d[i-1][0]+dp[i-1][2])*bit[i]呢,前面已经出现过49 //那么低位任意选择都可以,dp[i-1][0]是那些低位不出现49的,dp[i-1][2]是那些低位出现49的,按理说应该 //加上啊,BUT!!!清注意循环里面的第一句ans+=dp[i-1][2]*bit[i]; 前面已经加上了dp[i-1][2]低位有 //49的情况了,哎,欲哭无泪........ if(bit[i+1]==4&&bit[i]==9) has=true; } if(has) ans++; return ans; } //以491为例,先求出所有比400小的数中有多少符合题意的,然后4这一位确定以后,再求所有比490小,再求出所有比491小 //i=3 求出数 049 149 249 349 //i=2 求出数 449 //i=1 求出数 490 //自身包含49 所以求出数491 int main() { init(); int t;cin>>t; while(t--) { cin>>n; cout<<cal(n)<<endl; } return 0; }