Leetcode 416. Partition Equal Subset Sum

来源:互联网 发布:淘宝如何批量上传宝贝 编辑:程序博客网 时间:2024/05/01 09:01

最近刚好学了动规思想,所以在leetcode找了两道dp的题目来学习并复习。这一道题难度是medium,使用的是一维的dp算法。
-题目-
Given a non-empty array 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.
Note:
Each of the array element will not exceed 100.
The array size will not exceed 200.
-思路-
首先,因为问题是只分成两个子集,所以每个集合的和为总数的一半。我们将数组的所有元素求和,如果和为奇数,由于sum的一半不是整数,所以此时将数列拆分成两个和相等的集合是无解的。
若和为偶数,我们将这个问题定义为:
F[k]表示为是否有能组成和为k的数列的子集存在
则函数的返回值为F[sum]
然后我们的状态转移方程为:
F[k] = F[k] || F[k-nums[i]]
因为若k-nums[i]这个数是可得到的,那么k也是可得到的
因此我们就解决了这道经典的动规问题
-代码-

class Solution {public:    bool canPartition(vector<int>& nums) {        int sum = 0;         for(int i = 0; i < nums.size(); i++) {            sum += nums[i];         }        if(sum % 2) return false;         sum /= 2;         int result[sum+1] = {0};         result[0] = 1;         for(int i = 0; i < nums.size(); i++) {            for(int j = sum; j >= nums[i]; j--) {                result[j] = result[j] || result[j-nums[i]];             }        }        return result[sum];     }};
1 0
原创粉丝点击