代码随想录算法训练营|day43

果木26

已于 2024-02-26 14:46:02 修改

阅读量1.1k

点赞数 34

分类专栏：代码随想录练习文章标签：算法 go

于 2024-02-26 14:33:55 首次发布

本文链接：https://blog.csdn.net/weixin_43785190/article/details/136289561

版权

代码随想录练习专栏收录该内容

43 篇文章 0 订阅

订阅专栏

本文详细解析了三个编程题目，分别是使用动态规划解决的背包问题：1049.最后一块石头的重量II、494.目标和和474.一和零。重点介绍了如何通过递推公式计算背包容量和满足条件的方案数。

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

1049.最后一块石头的重量II

每次选出的两块石头重量越接近，最后剩余的重量越轻
将石头分成两堆，寻找重量轻的那一堆石头的最大值maxWeight，则另一堆石头重量为sum-maxWeight，最终结果为sum-2*maxWeight，因此定义背包容量target = sum(nums)/2
dp[i]表示背包容量为i时，能装到的最大石头重量。

func lastStoneWeightII(stones []int) int {
	sum := 0
	for _, stone := range stones {
		sum += stone
	}
	target := sum / 2
	dp := make([]int, target+1)
	for _, stone := range stones {
		for i := target; i >= stone; i-- {
			dp[i] = max(dp[i], dp[i-stone]+stone)
		}
	}
	return sum - 2*dp[target]
}

494.目标和

假设组成的等式中被减数为left，减数为right，和为sum，题意为寻找满足left-right=target的个数，改写成left-(sum-left)=target，即为寻找数组中满足left=(sum+target)/2的个数。
dp[i]表示背包容量为i时，有dp[i]种方法能装满背包。若abs(target)<abs(sum)或被减数为小数，无解

func findTargetSumWays(nums []int, target int) int {
	sum := 0
	for _, num := range nums {
		sum += num
	}
	if abs(sum) < abs(target) || (sum+target)%2 == 1 {
		return 0
	}
	left := (sum + target) / 2
	dp := make([]int, left+1)
	dp[0] = 1
	for _, num := range nums {
		for i := left; i >= num; i-- {
			dp[i] += dp[i-num]
		}
	}
	return dp[left]
}

func abs(x int) int {
	return int(math.Abs(float64(x)))
}

474.一和零

dp[i][j]表示背包容量为i，j时，能用str装满背包的个数，即strs子集的个数

func findMaxForm(strs []string, m int, n int) int {
	dp := make([][]int, m + 1)
	for i := 0; i < m + 1; i++ {
		dp[i] = make([]int, n + 1)
	}
	for _, str := range strs {
		zeros, ones := countZerosAndOnes(str)[0], countZerosAndOnes(str)[1]
		for i := m; i >= zeros; i-- {
			for j := n; j >= ones; j-- {
				dp[i][j] = max(dp[i][j], dp[i-zeros][j-ones]+1)
			}
		}
	}
	return dp[m][n]
}

func countZerosAndOnes(str string) []int {
	count := make([]int, 2)
	for _, c := range str {
		count[c-'0']++
	}
	return count
}