首页 > 代码库 > [ACM] hdu 3555 Bomb (数位DP,统计1-N中含有“49”的总数)

[ACM] hdu 3555 Bomb (数位DP,统计1-N中含有“49”的总数)

Bomb

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 131072/65536 K (Java/Others)
Total Submission(s): 7187 Accepted Submission(s): 2512


Problem Description
The counter-terrorists found a time bomb in the dust. But this time the terrorists improve on the time bomb. The number sequence of the time bomb counts from 1 to N. If the current number sequence includes the sub-sequence "49", the power of the blast would add one point.
Now the counter-terrorist knows the number N. They want to know the final points of the power. Can you help them?

Input
The first line of input consists of an integer T (1 <= T <= 10000), indicating the number of test cases. For each test case, there will be an integer N (1 <= N <= 2^63-1) as the description.

The input terminates by end of file marker.

Output
For each test case, output an integer indicating the final points of the power.

Sample Input
3 1 50 500

Sample Output
0 1 15
Hint
From 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.

Author
fatboy_cw@WHU

Source
2010 ACM-ICPC Multi-University Training Contest(12)——Host by WHU

Recommend
zhouzeyong | We have carefully selected several similar problems for you:3554 3556 3557 3558 3559


解题思路:

题意就是求从整数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;
}