首页 > 编程语言 >「代码随想录算法训练营」第三十天 | 动态规划 part3

「代码随想录算法训练营」第三十天 | 动态规划 part3

时间:2024-08-06 11:10:42浏览次数:10  
标签:vector target weight int 训练营 随想录 part3 数组 dp

46. 携带研究材料(0-1背包问题)

题目链接:https://kamacoder.com/problempage.php?pid=1046
文章讲解:https://programmercarl.com/背包理论基础01背包-1.html
视频讲解:https://www.bilibili.com/video/BV1cg411g7Y6/
题目状态:看题解过

思路:

创建一个二维的dp数组,用来进行动态规划,其中dp[i][j]表示在背包大小为j、物品为i的情况下,最多能装入多大价值的东西。
首先当j=0时,也就是背包容量为0时,价值肯定是0,因此dp[i][0]肯定都是0。
关键是如何动态规划:

  • dp[i][j]肯定是在dp[i-1][j]的基础上进行的,要判断当前背包的大小能否装下i的物品,
  • 若不能,则继承dp[i-1][j]的最大价值;
  • 若能,则需要判断是继承dp[i-1][j]的价值大还是继承dp[i-1][j-weight[i]]+value[i]的价值大。
  • 其中dp[i-1][j-weight[i]]是为了给背包的容量中留出i体积大小的空余,以便能装下i。

dp二维数组可视化:

代码:

//二维dp数组实现
#include <bits/stdc++.h>
using namespace std;

int n, bagweight;// bagweight代表行李箱空间
void solve() {
    vector<int> weight(n, 0); // 存储每件物品所占空间
    vector<int> value(n, 0);  // 存储每件物品价值
    for(int i = 0; i < n; ++i) {
        cin >> weight[i];
    }
    for(int j = 0; j < n; ++j) {
        cin >> value[j];
    }
    // dp数组, dp[i][j]代表行李箱空间为j的情况下,从下标为[0, i]的物品里面任意取,能达到的最大价值
    vector<vector<int>> dp(weight.size(), vector<int>(bagweight + 1, 0));

    // 初始化, 因为需要用到dp[i - 1]的值
    // j < weight[0]已在上方被初始化为0
    // j >= weight[0]的值就初始化为value[0]
    for (int j = weight[0]; j <= bagweight; j++) {
        dp[0][j] = value[0];
    }

    for(int i = 1; i < weight.size(); i++) { // 遍历科研物品
        for(int j = 0; j <= bagweight; j++) { // 遍历行李箱容量
            // 如果装不下这个物品,那么就继承dp[i - 1][j]的值
            if (j < weight[i]) dp[i][j] = dp[i - 1][j];
            // 如果能装下,就将值更新为 不装这个物品的最大值 和 装这个物品的最大值 中的 最大值
            // 装这个物品的最大值由容量为j - weight[i]的包任意放入序号为[0, i - 1]的最大值 + 该物品的价值构成
            else dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - weight[i]] + value[i]);
        }
    }
    cout << dp[weight.size() - 1][bagweight] << endl;
}

int main() {
    while(cin >> n >> bagweight) {
        solve();
    }
    return 0;
}

这道题也可以使用一维的dp数组来实现,因为每次遍历的时候都发现当前的dp[i][j]要么继承自dp[i - 1][j],要么继承自dp[i - 1][j - weight[i]] + value[i],因此i一直是被i-1覆盖的,因此i这一维没有必要维护,可转化为一维数组,如下图

代码如下:

// 一维dp数组实现
#include <iostream>
#include <vector>
using namespace std;

int main() {
    // 读取 M 和 N
    int M, N;
    cin >> M >> N;

    vector<int> costs(M);
    vector<int> values(M);

    for (int i = 0; i < M; i++) {
        cin >> costs[i];
    }
    for (int j = 0; j < M; j++) {
        cin >> values[j];
    }

    // 创建一个动态规划数组dp,初始值为0
    vector<int> dp(N + 1, 0);

    // 外层循环遍历每个类型的研究材料
    for (int i = 0; i < M; ++i) {
        // 内层循环从 N 空间逐渐减少到当前研究材料所占空间
        for (int j = N; j >= costs[i]; --j) {
            // 考虑当前研究材料选择和不选择的情况,选择最大值
            dp[j] = max(dp[j], dp[j - costs[i]] + values[i]);
        }
    }

    // 输出dp[N],即在给定 N 行李空间可以携带的研究材料最大价值
    cout << dp[N] << endl;

    return 0;
}

416. 分割等和子集

题目链接:https://leetcode.cn/problems/partition-equal-subset-sum/
文章讲解:https://programmercarl.com/0416.分割等和子集.html
题目难度:中等
视频讲解:https://www.bilibili.com/video/BV1rt4y1N7jE/
题目状态:被测试用例误导,看题解过

思路:

和背包思路一样,不过这次的dp数组用来存的是最大和,dp[i]表示数组中的某些值加起来小于i的最大值,而当i为target时,dp[i]正好也等于target时,就是找到了我们要找的等和子集,其中target是题目给的数组中所有元素值之和的一半。
可视化:

代码:

class Solution {
public:
    bool canPartition(vector<int>& nums) {
        int sum = 0;
        vector<int> dp(10001, 0);
        for(auto &num : nums) {
            sum += num;
        }
        if(sum % 2 == 1) return false;
        int target = sum / 2;
        for(int i = 1; i < nums.size(); ++i) {
            for(int j = target; j >= nums[i]; --j) {
                dp[j] = max(dp[j], dp[j - nums[i]] + nums[i]);
            }
        }
        if(dp[target] == target) return true;
        return false;
    }
};

标签:vector,target,weight,int,训练营,随想录,part3,数组,dp
From: https://www.cnblogs.com/frontpen/p/18344724

相关文章

  • 排序Part3——归并排序
    续上篇,我们继续讲述排序的内容:归并排序首先我们先考虑两个已经排好序的列表,现在要使他们合成一个列表并且保证合成的列表也有序,那应该怎么做呢?其实呢,以上过程我们成为一次合并。一次合并其实算法很简单,我们先分别设置两个指针,一个指向表1,一个指向表2,同时按顺序历遍:第......
  • 代码随想录二刷栈与队列
    代码随想录二刷栈与队列栈模拟队列具体思路如下:程序如下:classMyQueue:def__init__(self):self.stack_in=[]self.stack_out=[]defpush(self,x:int)->None:self.stack_in.append(x)defpop(self)->int:if......
  • Studying-代码随想录训练营day59| dijkstra(堆优化版)精讲、Bellman_ford 算法精讲
    第59天,dijkstra算法的优化版本,以及Bellman_ford算法......
  • 代码随想录Day5
    242.有效的字母异位词给定两个字符串s和t,编写一个函数来判断t是否是s的字母异位词。注意:若s和t中每个字符出现的次数都相同,则称s和t互为字母异位词。示例1:输入:s="anagram",t="nagaram"输出:true示例2:输入:s="rat",t="car"输出:false......
  • 代码随想录算法训练营day04之字符串
    题目及链接:344.反转字符串541.反转字符串||卡码网54.替换数字151.翻转字符串里的单词卡码网55.右旋字符串28.找出字符串中第一个匹配项的下标459.重复的子字符串344.反转字符串太简单就不写了541.反转字符串||题意:给定一个字符串s和一个整数k,从字符串开头算起,每......
  • 代码随想录day20 || 235 二叉搜索树最近公共祖先,701 二叉搜索树插入,450,二叉搜索树删除
    235二叉搜索树最近公共祖先unclowestCommonAncestor(root,p,q*TreeNode)*TreeNode{ //本题相较于普通二叉树寻找最近公共祖先加了题设条件二叉搜索树,所以使用二叉搜索树特性 //如果root大于两个目标节点,那么目标都在root左子树 //如果root小于两个目标节点,那么目......
  • 代码随想录Day4
    24.两两交换链表中的节点给你一个链表,两两交换其中相邻的节点,并返回交换后链表的头节点。你必须在不修改节点内部的值的情况下完成本题(即,只能进行节点交换)。示例1:输入:head=[1,2,3,4]输出:[2,1,4,3]示例2:输入:head=[]输出:[]示例3:输入:head=[1]输出:[1]提示:链表......
  • 代码随想录二刷字符串
    代码随想录二刷字符串看leetcode这样一道题目:这道题若是用python库函数直接就秒了。但是那这道题就失去了本身的意义。题目注意事项中也说了输入字符串S可能存在前导空格、尾随空格或者单词间的多个空格。所以首先是对字符串处理。去除其中的空格。这与之前去除数组中去除特定......
  • 2024牛客暑期多校训练营5
    目录写在前面ELBHKGJ写在最后写在前面比赛地址:https://ac.nowcoder.com/acm/contest/81600。以下按个人难度向排序。妈的坐牢场啊前期除了日常战犯环节嗯吃三发之外顺的一批,后面4h一直在J上坐牢最后样例都没过呃呃呃呃,还剩1.5hdztlb大神说会K了但是以为J能调出来没......
  • 代码随想录算法训练营day03|203.移除链表元素,707.设计链表,206.反转链表
    203.移除链表元素题目链接:https://leetcode.cn/problems/remove-linked-list-elements/description/我的代码(分头节点和中间节点两种情况操作):/***Definitionforsingly-linkedlist.*structListNode{*intval;*ListNode*next;*ListNode():val......