数位DP-ACWing(提高课思路整理)

1081(度的数量)

题意:找出[L,R]区间内部满足条件的数量,条件是由互不相等的B为底的K次幂数组成。
分析:由于是由互不相等的B为底的K次幂数组成,可以把一个数当作B为底的展开项,且每一项的系数非0即1,因为如果大于1则超过了条件(互不相等)的要求。而求区间内满足的数,实际也可以基于前缀和的思想,求[1-R] 内满足条件的数减去[1-L-1]区域内满足条件的结果
数位DP的关键在于枚举各位上的数的情况,然后进行分类讨论,通常采用树结构进行分类讨论。
在这里插入图片描述
对于左分支满足条件的数目统计:由于当前第i-1已经填0,剩下i-1位当中还需要填K个1,所以满足条件的总数 C i − 1 k C^k_{i-1} Ci1k
此时统计数目问题改解成求组合数问题,由于本题组合数的范围较小(不会超过底35)所以直接用递推公式去做: C j i = C j − 1 i + C j − 1 i − 1 C^i_{j}=C^{i}_{j-1}+C^{i-1}_{j-1} Cji=Cj1i+Cj1i1,提前预处理出来组合数组,便于直接求取。
对于右分支,则每一次都需要前面几位已经用了几个1,如果当前还可以使用1,则继续使用,如果当前位是最后一位,且设置为1之后K个1用完,则整个右侧树链(原数n)也是一个合法的的数,需要累加上1。

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>

using namespace std;

const int N = 35;

int x, y, k, b;
int f[N][N]; // 组合数存储

void init(){
    for(int i = 0; i <= N ; i++){
        for(int j = 0 ; j <= i; j++){
            if(!j) f[i][j] = 1;
            else
                f[i][j] =f[i - 1][j - 1] + f[i - 1][j]; 
        }
    }
}

int dp(int n){
    if (n == 0) return 0;
    // 存放n的b进制的每一位
    vector<int> nums;
    // 获取n的b进制的每一位
    while(n) nums.push_back(n % b), n /= b;
    int res = 0;
    // 设计一个last来存放之前已经有了几个1
    int last = 0;
    // 从高位向低位遍历,来统计数目
    for(int i = nums.size() - 1; i >= 0 ; i--){
        int x = nums[i];
        // 只有当前位的数值大于0时,才有树的划分
        if(x > 0){
            // 1 第一种状态:当前位为0, 种类数就是剩下的几位中选k - last位为1
            res += f[i][k - last];
            if( x > 1){
                // 2 第二种状态:当前位值大于1,此时只能选择x = 1这一种情况
                res += f[i][k - last - 1];
                // 3 其他情况已经违规了,不需要继续了
                break;
            }else{
                // 3 第三种状态:当前值 == 1
                last ++;
                if(last > k) break;
            }
        }
        // 如果当前数n也满足条件,res++
        if (i == 0 && last == k) res ++;
    }
    return res;
}

int main()
{
    init();
    cin >> x >> y >> k >> b;
    cout << dp(y) - dp(x - 1) << endl;
    return 0;
}

1082(数字游戏)

题意:基本与1081相同,只是条件发生了变更:数必须满足从高位到地位非递减。
分析:同样需要先将求区间转成前缀和求两个函数的差值;按照数字的n位进行划分。
在这里插入图片描述
对于左侧分支,想要统计满足条件的数目:即在确定了n-1位上是k时,符合的总数应该时第n-2位上数>=k的总数加和。因此可以预处理出来总共i位,首位是j的满足条件的总数f[i][j];
f [ i ] [ j ] = ∑ k = j 9 f [ i − 1 ] [ k ] f[i][j]=\sum_{k=j}^9 f[i-1][k] f[i][j]=k=j9f[i1][k]
对于右分支,当循环到第0位,且当前值大于等于前一位的值时,表示当前数n也满足条件,总数+1。

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>

using namespace std;

const int N = 32;

int a, b;
int f[N][10];

void init(){
    for(int i = 0; i <= 9; i++) f[1][i] = 1;
    for(int i = 2 ; i < N ; i++){
        for(int j = 0 ; j <= 9 ; j ++){
            for(int k = j; k <= 9; k++){
                f[i][j] += f[i - 1][k];
            }
        }
    }
}

int dp(int n){
    // n == 0 默认是一种
    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 = last; j < x ; j++){
            res += f[i + 1][j];
        }
        if(x < last) break;
        last = x;
        if(!i) res ++;
    }
    return res;
}

int main()
{
    init();
    while(cin >> a >> b){
        cout << dp(b) - dp(a - 1) << endl;
    }
    return 0;
}

1083(windy数)

此题基本与上面两题差不多,就时改了个条件,不进一步分析

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
#include <cmath>

using namespace std;

const int N = 12;

int a, b;
int f[N][10];

void init(){
    for(int i = 0; i <= 9; i++) f[1][i] = 1;
    for(int i = 2; i < N ; i++){
        for(int j = 0 ; j <= 9; j ++){
            for(int k = 0; k <= 9; k ++){
                if(abs(j - k) >= 2){
                    f[i][j] += f[i - 1][k];
                }
            }
        }
    }
}

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

int main()
{
    init();
    cin >> a >> b;
    cout << dp(b) - dp(a - 1) << endl;
    return 0;
}

1084(数字游戏Ⅱ)

条件变更:数字必须满足各位数字之和 mod N 为 0。
分析:基于之前的学习,可以快速的构建出一个数位的树形结构,核心是如何快速的计算出左子树上满足条件的数目。如果第i位置上填上x,那么他可以从前所有其他位数之和+x mod N = 0的情况转移过来,进一步递推,设第i位之前各位之和是last,后面各位为rear,则(last + x + rear)mod N = 0,则 已知 (x + rear)mod N = -last mod N,所以第i位填上x的合法条件由所有余数为(-last mod N)的情况转移过来,所以可以设f[i][j][k]:表示共i位,首位填j,余数是k的总数,并事先预处理出所有的f。

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>

using namespace std;

const int N = 35, M = 110;

int a, b, p;
int f[N][10][M];

int mod(int x, int p){
    return (x % p + p) % p;
}

void init(){
    memset(f, 0, sizeof f);
    for(int i = 0; i <= 9; i++) f[1][i][i % p] ++;
    for(int i = 2; i < N; i ++){
        for(int j = 0 ; j <= 9 ; j++){
            for(int t = 0; t < p; t ++){
                for(int k = 0; k <= 9 ; k++){
                    f[i][j][t] += f[i-1][k][mod(t - j, p)];
                }
            }
        }
    }
}

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; // last存放前面已经确定位数的加和
    for(int i = nums.size() - 1; i >= 0; i--){
        int x = nums[i];
        for(int j = 0; j < x; j++){
            // (last + after)mod p = 0 -> after mod p = -last mod p
            res += f[i + 1][j][mod(-last, p)];
        }
        last += x;
        if(!i && last % p == 0) res ++;
    }
    return res;
}

/**
 * 条件变了,变成各位加和的取模N为0;
 * f[i][j][k]: 表示共i位,首位是j,模后余数是k的总数
 * f[i][j][k] += f[i-1][t][(k-j) % N]
 */
int main()
{
    while(cin >> a >> b >> p){
        init();
        cout << dp(b) - dp(a - 1) << endl;
    }
    return 0;
}```

# 1085(不要62)
一样的思路,不分析,条件也很简单,不需要kmp

```cpp
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>

using namespace std;

const int N = 35;
int f[N][10];
int a, b;

void init(){
    memset(f, 0, sizeof f);
    for(int i = 0; i <= 9; i++){
        if(i == 4) continue;
        f[1][i] = 1;
    }
    for(int i = 2; i < N; i ++){
        for(int j = 0 ; j <= 9; j++){
            if(j == 4) continue;
            for(int k = 0; k <=9; 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;
        last = x;
        if(!i) res ++;
    }
    return res;
}

/**
 * 条件变更:不包含4和连续的两位62的个数
 * f[i][j]: 表示共有i位,首位为j,且满足条件的数目
 */
int main()
{
    while(cin >> a >> b, a || b){
        init();
        cout << dp(b) - dp(a - 1) << endl;
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值