[Swust OJ 1097]--2014(数位dp)

题目链接：http://acm.swust.edu.cn/problem/1097/

 

Time limit(ms): 1000　　　　　　Memory limit(kb): 32768

 

今年是2014年，所以小明喜欢2014的每一位数字（即：2,0,1,4）,小明想知道在区间[l，r](包括l和r)中有多少个数中含有这4个数字（数字无前缀零）。

Description

多组数据。

每组数据输入2个数l，r（0<l<r<=10^9）

Input

输出占一行，即区间[l，r]（包括l和r）中包含的满足条件的数的个数

 

Output

1 2 3

1 10 100 1024

Sample Input

1 2 3

0 1

Sample Output

输出换行请使用\r\n

Hint

swust第10届校赛

 

 

解题思路：就一个简单的数位dp，直接套模板就是了~~~

（不会的可以戳戳这里：http://www.cnblogs.com/zyxStar/p/4563830.html）

 

代码如下：

#include<iostream>
#include<cstring>
using namespace std;
int dp[10][2][2][2][2];//返回各数状态
int bit[10];  //数位dp
int dfs(int pos, int s2, int s0, int s1, int s4, bool limit, bool fzero)
{
    //注意前导零的影响
    if (pos == -1) return s2&&s0&&s1&&s4;
    if (!limit&&!fzero&&~dp[pos][s2][s0][s1][s4])
        return dp[pos][s2][s0][s1][s4];//条件判断
    int end = limit ? bit[pos] : 9;
    int ans = 0, i;
    for (i = 0; i <= end; i++){
        int now2 = s2, now0 = s0, now1 = s1, now4 = s4;
        if (s2 == 0){
            if (i == 2)
                now2 = 1;
        }
        if (s0 == 0){
            if (!fzero&&i == 0)
                now0 = 1;
        }
        if (s1 == 0){
            if (i == 1) 
                now1 = 1;
        }
        if (s4 == 0){
            if (i == 4)
                now4 = 1;
        }
        ans += dfs(pos - 1, now2, now0, now1, now4, limit&&i == end, fzero&&!i);
    }
    return limit || fzero ? ans : dp[pos][s2][s0][s1][s4] = ans;
}
int calc(int n){
    int len = 0;
    while (n){
        bit[len++] = n % 10;
        n /= 10;
    }
    return dfs(len - 1, 0, 0, 0, 0, 1, 1);
}
int main(){
    int l, r;
    memset(dp, -1, sizeof(dp));
    while (cin >> l >> r)
        cout << calc(r) - calc(l - 1) << "\r\n";
    return 0;
}