寒假：Day18

Day18

1083. Windy数 - AcWing题库

数位DP的题的预处理就含有一定的DP方式，预处理可以让你的个数更好的求出来，而且每一道题不同的条件有不同的预处理方式，这也是数位DP的一个难点

#include<bits/stdc++.h>
using namespace std;
const int N = 11;
int f[N][N]; // 剩余i位且最高位是j的Windy数的个数

void init()
{
    for (int i = 0; i < 10; i++) f[1][i] = 1; // 预处理一位数的情况
    
    for (int i = 2; i < N; i++) { // 递推出二位数以上的情况
        for (int j = 0; j < 10; j++) {
            for (int k = 0; k < 10; k++) {
                if (abs(j - k) >= 2) f[i][j] += f[i - 1][k];
            }
        }
    }
}

int dp(int n)
{
    if (!n) return 0; // 特判0
    
    vector<int> nums;
    while (n) nums.push_back(n % 10), n /= 10; // 拆数字
    
    int res = 0;
    int last = -2; // 让last置为-2是为了能让第一位数可以任意取值
    
    for (int i = nums.size() - 1; i >= 0; i--) { // 从高到低枚举每一位数
        int x = nums[i];
        // 如果是第1位数，那就只能从1开始，其他的可以从0开始
        for (int j = (i == nums.size() - 1); j < x; j++) { 
            if (abs(j - last) >= 2) res += f[i + 1][j];
        }
        if (abs(x - last) >= 2) last = x; // 如果当前数是合法的，那么就继续。反之break
        else break;
        
        if (!i) res ++; // 如果枚举到了最后一位，证明这个数字本身也是合法的，那就++
    }
    // 特殊枚举有前导0的数
    for (int i = 1; i < nums.size(); i++) {
        for (int j = 1; j < 10; j++) {
            res += f[i][j];
        }
    }
    return res;
}

int main(void)
{
    init();
    int l, r;
    cin >> l >> r;
    cout << dp(r) - dp(l - 1) << endl ;
    return 0;
}

1085. 不要62 - AcWing题库

简单题，而且之前做过，不过之前是暴力枚举，现在数据范围大了需要数位DP来解决

#include<bits/stdc++.h>
using namespace std;
const int N = 11;
int f[N][N];

void init()
{
    for (int i = 0; i < 10; i++) if (i != 4) f[1][i] = 1;
    
    for (int i = 2; i < N; i++) {
        for (int j = 0; j < 10; j++) {
            if(j == 4) continue;
            for (int k = 0; k < 10; k++) {
                if (k == 4 || j == 6 && k == 2) continue;
                f[i][j] += f[i - 1][k];
            }
        }
    }
}

int dp(int n)
{
    if (!n) return 1;
    
    vector<int> nums;
    while (n) nums.push_back(n % 10), n /= 10;
    
    int res = 0;
    int last = 0;
    
    for (int i = nums.size() - 1; i >= 0; i--) 
    {
        int x = nums[i];
        for (int j = 0; j < x; j++) {
            if (j == 4 || last == 6 && j == 2) continue;
            res += f[i + 1][j];
        }
        if (x == 4 || last == 6 && x == 2) break;
        else last = x;
        
        if (!i) res ++;
    }
    return res;
}

int main(void)
{
    init();
    int l, r;
    while (cin >> l >> r, l || r) cout << dp(r) - dp(l - 1) << endl ;
    return 0;
}