acwing算法基础之动态规划--数位统计DP、状态压缩DP、树形DP和记忆化搜索

这篇具有很好参考价值的文章主要介绍了acwing算法基础之动态规划--数位统计DP、状态压缩DP、树形DP和记忆化搜索。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

1 基础知识

暂无。。。

2 模板

暂无。。。

3 工程化

题目1:求a~b中数字0、数字1、…、数字9出现的次数。

思路:先计算1~a中每位数字出现的次数,然后计算1~b-1中每位数字出现的次数,两个相减即是最终答案。

那么,如何计算1~a中每位数字出现的次数呢?

首先,将a的每一位存入向量num中,例如a=1234567,那么num为,
acwing算法基础之动态规划--数位统计DP、状态压缩DP、树形DP和记忆化搜索,Acwing,C++学习,算法,动态规划
考虑如下两个子问题,

  1. 1~a中数字0出现的次数。
  2. 1~a中数字5出现的次数。为啥选择数字5呢?因为1到9中的任意一个数都和5等价。

对于问题1:1~x中数字0出现的次数。

记num中有n位,从第0位不考虑,因为第0位不可能取到0,即数字首位不能为0,例如0123,这样的数字是不合法的,应该表示成123。所以i从1到n-1,即for (int i = 1; i < n; ++i)。考虑如下情况,

  1. 当num的第i位取0且其左侧部分不取L时,有多少个数?
    acwing算法基础之动态规划--数位统计DP、状态压缩DP、树形DP和记忆化搜索,Acwing,C++学习,算法,动态规划
    左边部分的方案数为:1~L-1,共L-1个数。
    右边部分的方案数为:0~ 1 0 n − 1 − i − 1 10^{n-1-i}-1 10n1i1,共 1 0 n − 1 − i 10^{n-1-i} 10n1i个数。
    故总方案数为: ( L − 1 ) ⋅ 1 0 n − 1 − i (L-1)\cdot 10^{n-1-i} (L1)10n1i

  2. 当num的第i位取0且其左侧部分取L时,
    2.1 如果第i位原先数字等于0,那么左边方案数=1,右边方案数=0~R,共R+1个,总方案数=1 * (R+1) = R+1。
    2.2 如果第i位原先数字大于0,那么左边方案数=1,右边方案数=0~ 1 0 n − 1 − i − 1 10^{n-1-i}-1 10n1i1,共 1 0 n − 1 − i 10^{n-1-i} 10n1i个数。

对于问题2:1~x中数字5出现的次数。

如果num的第0位等于5,有R+1个方案数。如果num的第0位大于5,有 1 0 n − 1 10^{n-1} 10n1个数。

i从1到n-1,即for (int i = 1; i < n; ++i)。考虑如下情况,

  1. 当num的第i位取5且其左侧部分不取L时,有多少个数?
    acwing算法基础之动态规划--数位统计DP、状态压缩DP、树形DP和记忆化搜索,Acwing,C++学习,算法,动态规划
    左边部分的方案数为:0~L-1,共L个数。
    右边部分的方案数为:0~ 1 0 n − 1 − i − 1 10^{n-1-i}-1 10n1i1,共 1 0 n − 1 − i 10^{n-1-i} 10n1i个数。
    故总方案数为: L ⋅ 1 0 n − 1 − i L\cdot 10^{n-1-i} L10n1i

  2. 当num的第i位取5且其左侧部分取L时,
    2.1 如果第i位原先数字等于5,那么左边方案数=1,右边方案数=0~R,共R+1个,总方案数=1 * (R+1) = R+1。
    2.2 如果第i位原先数字大于5,那么左边方案数=1,右边方案数=0~ 1 0 n − 1 − i − 1 10^{n-1-i}-1 10n1i1,共 1 0 n − 1 − i 10^{n-1-i} 10n1i个数。

综合上述,C++代码如下,

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

using namespace std;
int a, b;

int compute_R(const vector<int>& num, int i) {//计算第i位右侧数字大小
    int R = 0;
    for (int k = i + 1; k < num.size(); ++k) {
        R = R * 10 + num[k];
    }
    return R;
}

int compute_L(const vector<int>& num, int i) {//计算第i位左侧数字大小
    int L = 0;
    for (int k = 0; k < i; ++k) {
        L = L * 10 + num[k];
    }
    return L;
}

int compute_cnt(int a, int x) {//计算1~a中x出现的次数
    if (a <= 0) return 0;
    
    vector<int> num; //把a转化为num,高位在前
    int t = a;
    while (t) {
        num.emplace_back(t % 10);
        t /= 10;
    }
    reverse(num.begin(), num.end()); //保证高位在前
    
    int n = num.size(); //a总共有n位
    
    int res = 0; //存储1~a中x出现的次数
    
    //考虑第0位取x的情况
    if (x != 0) {
        if (num[0] == x) {
            //有R+1个方案
            int R = compute_R(num, 0);
            res += R + 1;
        } else if (num[0] > x) {
            //有10^(n-1)个方案数
            res += pow(10, n - 1);
        }
    }
    
    //考虑第i位取x的情况
    for (int i = 1; i < n; ++i) {
        //计算情况1中x出现的次数
        if (x == 0) {
            int L = compute_L(num, i);
            res += (L - 1) * pow(10, n - 1 - i);
        } else {
            int L = compute_L(num, i);
            res += L * pow(10, n - 1 - i);
        }
        
        //计算情况2中x出现的次数
        if (num[i] == x) {
            int R = compute_R(num, i);
            res += R + 1;
        } else if (num[i] > x) {
            res += pow(10, n - 1 - i);
        }
    }
    
    return res;//返回1~a中x出现的次数
}


int main() {
    
    while (cin >> a >> b, a || b) {
        if (a > b) swap(a, b); //保证b是大数
        
        for (int x = 0; x < 10; ++x) {
            int cnt1 = compute_cnt(b, x); //计算1~b中x出现的次数
            int cnt2 = compute_cnt(a - 1, x); //计算1~a中x出现的次数
            cout << cnt1 - cnt2 << " ";
        }
        cout << endl;
        
    }
    
    return 0;
}

题目2:在N * M的棋盘下,摆放1*2的矩形块,可以竖着摆,也可以横着摆,要求摆满棋盘,求有多少种摆放方案。

解题思路:状态压缩类DP。

规定横着摆放的1*2的矩形块,第一个小方格为起点方格,第二个小方格为终点方格,如下图所示,
acwing算法基础之动态规划--数位统计DP、状态压缩DP、树形DP和记忆化搜索,Acwing,C++学习,算法,动态规划
f[i][j]状态定义:前i-1列已经摆放完毕,第i列中终点方格的状态为j,的方案数。

其中终点方格的状态j是指,从第0行到第n-1行,有终点网格的记作1,没有终点网格的记作0。比如下图中第1列的终点方格的状态是 ( 1001 ) 2 (1001)_2 (1001)2,即十进制中的9。

acwing算法基础之动态规划--数位统计DP、状态压缩DP、树形DP和记忆化搜索,Acwing,C++学习,算法,动态规划
f[i][j]的状态转移:f[i-1][k],即前i-2列已经摆放完毕,第i-1列中终点网格的状态为k。k需要满足两个条件:

  1. 第i-1列中的终点网格和起点网格不能重合,其中终点网格的状态为k,起点网格的状态为j,也即k & j == 0
  2. 连续的空网格的数目不能是奇数,否则竖着摆放不下1*2的矩形块。即数k | j二进制表示中连续0的数目不能是奇数。

故,综合上述,状态转移方程为,
f [ i ] [ j ] = ∑ k f [ i − 1 ] [ k ] f[i][j] = \sum_kf[i-1][k] f[i][j]=kf[i1][k]

初始化f[0][0] = 1

最终答案f[M][0],也即第0列、第1列、…、第M-1列均摆放完毕,第M列中终点格子的状态为0的方案数。

C++代码如下,文章来源地址https://www.toymoban.com/news/detail-759751.html

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

using namespace std;

const int N = 12, M = 1 << N;
bool st[M];
vector<vector<int>> last(M);
long long f[N][M];
int n, m;

int main() {
    while (cin >> n >> m, n || m) {
        //n*m的棋盘
        
        //步骤1:把状态0~1<<n-1中满足连续0个数为偶数的状态标识出来
        memset(st, 0, sizeof st); //多组测试数据,所以要初始化
        for (int i = 0; i < 1 << n; ++i) {
            //数i的二进制表示中,连续0的个数是不是都是偶数
            bool flag = true;
            int cnt = 0; //连续0的个数
            for (int j = 0; j < n; ++j) {
                if (i >> j & 1) {//第j位为1,前面的连续0之和为cnt
                    if (cnt & 1) {//如果cnt是奇数
                        flag = false;
                        break;
                    }
                    cnt = 0;
                } else {
                    cnt += 1;
                }
            }
            if (cnt & 1) flag = false;//判断最后一个连续0的数目
            
            st[i] = flag; //true表示状态i中连续0的个数都是偶数。
            //cout << "i = " << i << ", st[i] = " << st[i] << endl;
        }
        
        //步骤2:f[i][j]和f[i-1][k],存储j满足要求的k
        for (int j = 0; j < 1 << n; ++j) {
            last[j].clear(); //多组测试数据,所以要清空
            for (int k = 0; k < 1 << n; ++k) {
                if ((k & j) == 0 && st[k | j]) {
                    last[j].emplace_back(k);
                }
            }
        }
        
        //步骤3:正式dp
        memset(f, 0, sizeof f);
        f[0][0] = 1;
        for (int i = 1; i <= m; ++i) {
            for (int j = 0; j < 1 << n; ++j) {
                for (auto k : last[j]) {
                    f[i][j] += f[i-1][k];
                }
            }
        }
        
        cout << f[m][0] << endl;
    }
    return 0;
}

题目3:给定n*n的矩阵,表示n个结点两两之间的距离,求从0号结点出发到达第n-1号结点,经过每一个结点一次的路径的最小距离。

解题思路:状态压缩DP。

状态定义,f[i][j],即:所有从第0号结点走到第j号结点,且走过的结点是i的路径,的最小值。

其中i的二进制表示中第0位为1表示,经过第0号结点,否则不经过第0号结点。
i的二进制表示中第1位为1表示,经过第1号结点,否则不经过第1号结点。
i的二进制表示中第2位为1表示,经过第2号结点,否则不经过第2号结点。
……
i的二进制表示中第n-1位为1表示,经过第n-1号结点,否则不经过第n-1号结点。

注意,只有满足(i & 1) == 1 && (i & (1 << j)) == (1 << j)时,状态f[i][j]才合法。

状态转移,考虑最后一步咋走的,即f[last][k] + d[k][j],其中i包含结点k时才合法,即(i & (1 << k)) == (1 << k),且last = i - (1 << j)

故,综合上述,状态转移方程为,
f [ i ] [ j ] = m i n k { f [ l a s t ] [ k ] + d [ k ] [ j ] } f[i][j] = \underset {k}{min} \{ f[last][k] + d[k][j] \} f[i][j]=kmin{f[last][k]+d[k][j]}
l a s t = i − ( 1 < < k ) last = i - (1 << k) last=i(1<<k)

初始化,f[1][0] = 0,其余初始化为正无穷大。

最终答案,f[(1 << n) - 1][n-1]

C++代码如下,

#include <iostream>
#include <cstring>

using namespace std;

const int N = 21, M = 1 << N;
int f[M][N];
int n;
int d[N][N];

int main() {
    cin >> n;
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < n; ++j) {
            cin >> d[i][j];
        }
    }
    
    memset(f, 0x3f, sizeof f);
    f[1][0] = 0;
    
    for (int i = 0; i < (1 << n); ++i) {
        for (int j = 0; j < n; ++j) {
            //从0出发到达j,经过的结点是i
            if ((i & 1) == 1 && (i & (1 << j)) == (1 << j)) {
                //f[i][j]是合法的
                for (int k = 0; k < n; ++k) {
                    if ((i & (1 << k)) == (1 << k)) {
                        //i中有k
                        int last = i - (1 << j);
                        f[i][j] = min(f[i][j], f[last][k] + d[k][j]);
                    }
                }
            }
        }
    }
    
    cout << f[(1 << n) - 1][n-1] << endl;
    
    return 0;
}

题目4:没有上司的舞会。直接上司和员工不能同时选择,求最大值。

解题思路:树形DP,从根结点递归处理。

状态定义,有,

  1. f[i][0],所有以i为根的子树中选择,但不选择结点i的方案数中的最大值。
  2. f[i][1],所有以i为根的子树中选择,且选择结点i的方案数中的最大值。

状态转移,

f [ i ] [ 0 ] = ∑ j m a x { f [ j ] [ 0 ] , f [ j ] [ 1 ] } f[i][0] = \sum_j max\{ f[j][0], f[j][1] \} f[i][0]=jmax{f[j][0],f[j][1]}
其中结点j为结点i的子结点。
f [ i ] [ 1 ] = w [ i ] + ∑ j f [ j ] [ 0 ] f[i][1] = w[i] + \sum_j f[j][0] f[i][1]=w[i]+jf[j][0]
其中结点j为结点i的子结点。

初始化,二维数组初始化为0。

记根结点为start,那么最终答案res = max(f[start][0], f[start][1])

C++代码为,

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

using namespace std;

const int N = 6010;

int n, start;
bool has_father[N];
int happy[N];
int f[N][2];
vector<vector<int>> sons(N);

void dfs(int i) {
    f[i][1] = happy[i];
    
    for (auto j : sons[i]) {
        dfs(j);
        f[i][1] += f[j][0];
        f[i][0] += max(f[j][0], f[j][1]);
    }
    return;
}

int main() {
    cin >> n;
    for (int i = 1; i <= n; ++i) cin >> happy[i];
    
    for (int i = 1; i < n; ++i) {
        int a, b;
        cin >> a >> b;
        has_father[a] = true;
        sons[b].emplace_back(a);
    }
    
    //找到根结点
    start = -1;
    for (int i = 1; i <= n; ++i) {
        if (has_father[i] == false) {
            start = i;
            break;
        }
    }
    
    dfs(start);
    
    cout << max(f[start][0], f[start][1]) << endl;
    
    return 0;
}

题目5:滑雪。n*m的网格,可以往上下左右方向尝试,只能从高处滑到低处。求路径最长值。

思路:记忆化搜索,状态先全部初始化为-1,然后递归每一个位置,如果f[i][j] != -1,则说明它被计算过了,直接返回。

状态表示,f[i][j]:从(i,j)开始滑的最长路径值。

状态转移,有:

  1. 如果能往上滑,即(i-1,j)在网格内且h[i][j] > h[i-1][j],有f[i-1][j] + 1
  2. 如果能往下滑,即(i+1,j)在网格内且h[i][j] > h[i+1][j],有f[i+1][j] + 1
  3. 如果能往左滑,即(i-1,j)在网格内且h[i][j] > h[i-1][j],有f[i-1][j] + 1
  4. 如果能往右滑,即(i+1,j)在网格内且h[i][j] > h[i+1][j],有f[i+1][j] + 1

初始化,所有状态均设置为-1。

答案,即所有状态的最大值。

C++代码如下,

#include <iostream>
#include <cstring>

using namespace std;

const int N = 310;
int n, m;
int f[N][N], h[N][N];

int dirs[4][2] = {{1,0}, {-1,0}, {0,1}, {0,-1}};

int dfs(int i, int j) {
    if (f[i][j] != -1) return f[i][j];
    
    f[i][j] = 1;
    for (int k = 0; k < 4; ++k) {
        int x = i + dirs[k][0], y = j + dirs[k][1];
        if (x >= 1 && x <= n && y >= 1 && y <= m && h[i][j] > h[x][y]) {
            f[i][j] = max(f[i][j], dfs(x, y) + 1);
        }
    }
    
    return f[i][j];
}

int main() {
    cin >> n >> m;
    for (int i = 1; i <= n; ++i) {
        for (int j = 1; j <= m; ++j) {
            cin >> h[i][j];
        }
    }
    
    memset(f, -1, sizeof f);
    
    int res = 0;
    for (int i = 1; i <= n; ++i) {
        for (int j = 1; j <= m; ++j) {
            res = max(res, dfs(i, j));
        }
    }
    
    cout << res << endl;
    
    return 0;
}

到了这里,关于acwing算法基础之动态规划--数位统计DP、状态压缩DP、树形DP和记忆化搜索的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • 【数位dp】【动态规划】C++算法:233.数字 1 的个数

    视频算法专题 动态规划汇总 给定一个整数 n,计算所有小于等于 n 的非负整数中数字 1 出现的个数。 示例 1: 输入:n = 13 输出:6 示例 2: 输入:n = 0 输出:0 提示: 0 = n = 10 9 本题比较简单,主要讲封装类。m_vPre记录上一位所有状态,程序结束时,记录的是最后一位的所有

    2024年01月16日
    浏览(42)
  • 动态规划——数位DP 学习笔记

    引入 数位 DP 往往都是这样的题型:给定一个区间 ([l, r]) ,求这个区间中满足某种条件的数的总数。 简单的暴力代码如下: 而当数据规模过大,暴力枚举就 (mathbb T) 飞了,因此引入数位 DP: 概念 数位(digit):对于十进制,即把一个数字按照个位、十位、百位等,一位

    2024年02月08日
    浏览(47)
  • 算法沉淀 —— 动态规划篇(简单多状态dp问题下)

    几乎所有的动态规划问题大致可分为以下5个步骤,后续所有问题分析都将基于此 1.、状态表示:通常状态表示分为基本分为以下两种,其中更是以第一种为甚。 以i为结尾 ,dp[i] 表示什么,通常为代求问题(具体依题目而定) 以i为开始 ,dp[i]表示什么,通常为代求问题(具

    2024年04月16日
    浏览(46)
  • 【算法优选】 动态规划之简单多状态dp问题——贰

    动态规划相关题目都可以参考以下五个步骤进行解答: 状态表示 状态转移⽅程 初始化 填表顺序 返回值 后面题的解答思路也将按照这五个步骤进行讲解。 给定一个整数数组prices,其中第 prices[i] 表示第 i 天的股票价格 设计一个算法计算出最大利润。在满足以下约束条件下,

    2024年04月12日
    浏览(35)
  • 算法沉淀 —— 动态规划篇(简单多状态dp问题上)

    几乎所有的动态规划问题大致可分为以下5个步骤,后续所有问题分析都将基于此 1.、状态表示:通常状态表示分为以下两种,其中更是第一种为主。 以i为结尾 ,dp[i] 表示什么,通常为代求问题(具体依题目而定) 以i为开始 ,dp[i]表示什么,通常为代求问题(具体依题目而

    2024年04月11日
    浏览(43)
  • 【动态规划】【 数位dp】2827. 范围中美丽整数的数目

    数位dp 动态规划汇总 给你正整数 low ,high 和 k 。 如果一个数满足以下两个条件,那么它是 美丽的 : 偶数数位的数目与奇数数位的数目相同。 这个整数可以被 k 整除。 请你返回范围 [low, high] 中美丽整数的数目。 示例 1: 输入:low = 10, high = 20, k = 3 输出:2 解释:给定范围

    2024年04月12日
    浏览(41)
  • acwing算法基础之动态规划--背包问题

    (零) 背包问题描述:有 N N N 个物品,每个物品的体积是 v i v_i v i ​ ,价值是 w i w_i w i ​ ,现有容量是 V V V 的背包,求这个背包能装下的物品的最大价值。 01背包问题:每个物品只有1个。 完全背包问题:每个物品有无穷多个。 多重背包问题:第 i i i 个物品有 s i s_i s

    2024年02月05日
    浏览(47)
  • Acwing-基础算法课笔记之动态规划(背包问题)

    01背包中的0和1指的是放与不放,而且不能出现放多个的情况,背包只能放相同物品中的一个; 首先是对 d [ i ] [ j ] d[i][j] d [ i ] [ j ] 数组的解释,该数组表示的是只看前 i i i 个物品,总体积是 j j j 的情况下,总价值最大是多少; 不选第 i i i 个物品,只考虑前 i − 1 i-1 i −

    2024年04月17日
    浏览(49)
  • [leetcode~数位动态规划] 2719. 统计整数数目 hard

    给你两个数字字符串 num1 和 num2 ,以及两个整数 max_sum 和 min_sum 。如果一个整数 x 满足以下条件,我们称它是一个好整数: num1 = x = num2 min_sum = digit_sum(x) = max_sum. 请你返回好整数的数目。答案可能很大,请返回答案对 109 + 7 取余后的结果。 注意,digit_sum(x) 表示 x 各位数字之

    2024年01月18日
    浏览(43)
  • 【AcWing算法基础课】第五章 动态规划(未完待续)

    本专栏文章为本人AcWing算法基础课的学习笔记,课程地址在这。如有侵权,立即删除。 dp问题的优化 :在基本形式dp上作等价变形。 dp问题的解题方法 : 1)状态表示 集合 属性:最大值/最小值/数量。 2)状态计算 集合划分(不重不漏) 题目链接: 2. 01背包问题 - AcWing题库

    2024年02月12日
    浏览(72)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包