LeetCode动态规划 分割等和子集
Given a non-empty array nums containing only positive integers, find if the array can be partitioned into two subsets such that the sum of elements in both subsets is equal.
思路
這道題假如能看出是一道0-1背包問題的話,就很好解決了。題目說需要劃分成兩個和相等的子集,那么需要滿足:
1.每個元素必須被使用且只能被使用一次
2.兩個子集的和相等,子集和相加必為偶數
根據子集和相加必為偶數這一點,當nums數組元素和sum是奇數時,直接return false。那么接下來問題就可以轉化成能否選取幾個物品使其重量和為sum/2的問題。
狀態轉移方程
dp[i][j] = true, when dp[i-1][j] == true
dp[i][j] = true, when j == nums[i]
dp[i][j] = dp[i-1][j-nums[i]], when j > nums[i]
邊界條件
若nums[0] <= sum/2,dp[0][0] = true
注明
為什么此處的初始化不像一般的背包問題那樣是個循環呢,我想了一下一般的背包問題是求最大價值和的,所以需要循環一般dp[0][j]記錄各自中間值。而本題中不需要記錄中間值,只需要判斷是true還是false,因此僅需判斷dp[0][0]是否為true即可。
代碼
class Solution { public:bool canPartition(vector<int>& nums) {int sum = 0;int len = nums.size();for(auto num : nums)sum += num;if(sum % 2) return false;int bag_size = sum / 2;bool dp[205][10005] = {false};if(nums[0] <= bag_size)dp[0][nums[0]] = true;for(int i = 1; i < len; i++)for(int j = 0; j <= bag_size; j++){if(dp[i-1][j] == true)dp[i][j] = true;else if(j == nums[i])dp[i][j] = true;else if(j > nums[i])dp[i][j] = dp[i-1][j-nums[i]];}return dp[len-1][bag_size];} };總結
以上是生活随笔為你收集整理的LeetCode动态规划 分割等和子集的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: LeetCode动态规划 杨辉三角
- 下一篇: sklearn决策树概述