Day43 Golang (背包问题) 1049.最后一块石头的重量II 494.目标和 474.一和零

风的余温zl

已于 2024-02-26 20:31:13 修改

阅读量380

点赞数 7

分类专栏：代码随想录60天刷题文章标签：算法 golang

于 2024-02-26 20:30:07 首次发布

本文链接：https://blog.csdn.net/fdyw_gzl/article/details/136304830

版权

代码随想录60天刷题专栏收录该内容

53 篇文章 0 订阅

订阅专栏

文章介绍了如何通过动态规划算法解决三个不同场景的背包问题：石头重量问题、元素和与目标和的组合以及字符串中的字符组合。主要展示了如何计算满足条件的元素数量或价值。

摘要由CSDN通过智能技术生成

func lastStoneWeightII(stones []int) int {
	//转换为背包问题,计算重量之和并取一半
	sum := 0
	for _, v := range stones {
		sum += v
	}
	target := sum / 2
	//dp[j]表示背包重量为j的最大价值，这里重量=价值
	dp := make([]int, 1501)
	//初始化dp数组
	dp[0] = 0
	//遍历顺序为先遍历物品再倒序遍历背包
	for i := 0; i < len(stones); i++ {
		for j := target; j >= stones[i]; j-- {
			dp[j] = max(dp[j], dp[j-stones[i]]+stones[i])
		}
	}
    return sum-2*dp[target]
}
func max(a,b int) int {
	if a > b {
		return a
	}
	return b
}

func findTargetSumWays(nums []int, target int) int {
	//求元素之和
	sum := 0
	for _, v := range nums {
		sum += v
	}
	//如果target>sum或者(sum+target)/2为奇数则返回0
	if abs(target) > sum {
		return 0
	}
	if (sum+target)%2 == 1 {
		return 0
	}
	//加法集合 left=（sum+target）/2
	left := (sum + target) / 2
	//dp[j]表示装满加法集合left有多少种方法
	dp := make([]int, left+1)
	//dp[0]初始化为1，其余为0
	dp[0] = 1
	for i := 0; i < len(nums); i++ {
		for j := left; j >= nums[i]; j-- {
			dp[j] += dp[j-nums[i]]
		}
	}
	return dp[left]
}
func abs(a int) int {
	return int(math.Abs(float64(a)))
}

func findMaxForm(strs []string, m int, n int) int {
	// 转换为背包问题，每个物品的重量为m个0，n个1
	// dp[i][j]表示背包容量为 m个0，n个1 所能装的最大元素个数
	dp := make([][]int, m+1)
	for i, _ := range dp {
		dp[i] = make([]int, n+1)
	}
	//dp初始化
	dp[0][0] = 0
	//遍历顺序为先遍历物品再遍历背包
	for _, str := range strs { //遍历物品
		x, y := 0, 0 //统计0和1的数量
		for _, c := range str {
			if c == '0' {
				x++
			} else {
				y++
			}
		}
		for i := m; i >= x; i-- { //再遍历背包，x、y顺序可以交换
			for j := n; j >= y; j-- {
				dp[i][j] = max(dp[i][j], dp[i-x][j-y]+1)
			}
		}
	}
    return dp[m][n]
}