1. 解析
题目大意,将序列分成和相等的两个子序列 。
2. 分析
我想到的解法是,首先排除特殊的情况,因为题目要求划分成两个和相等的子序列,如果当前序列的总和为奇数或者序列当中存在某个元素大于和的一半,那么肯定无法划分成两个子序列;对于其他情况采用深度优先检索,看是否存在满足的情况,如果存在,直接返回true;反之,返回false即可。思想比较简单,但这种解法是无法OJ的,主要是因为递归要重复的扫描,导致时间复杂度过高,例如,如果序列全为1,即[1, 1, 1, 1, 1, ......., 96, 97],可以明显看到,要反复的检索。
class Solution {
public:
bool canPartition(vector<int>& nums) {
int target = 0;
for (int num : nums) target += num;
if (target % 2 != 0 || upper_bound(nums.begin(), nums.end(), target / 2) != nums.end()) //和为奇数或者存在某个元素超过和的一半,无法划分
return false;
bool res = false;
sort(nums.begin(), nums.end());
DFS(nums, target/2, 0, res);
return res;
}
void DFS(vector<int>& nums, int target, int pos, bool& res){
if (res) return;
if (target == 0){
res = true;
return;
}
for (int i = pos; i < nums.size(); ++i){
if (nums[i] > target) break;
DFS(nums, target - nums[i], i + 1, res);
}
}
};
3. DP解法
参考@Grandyang的思路,还是比较难想到的~~~dp[i]表示序列中的元素之和等于i,每次遍历当前元素,查看之前(target-num)是否可以用数组中的元素表示,如果可以,那么target也就可以表示。
例如 nums = [1, 5, 11, 5]
target = 11
检测1,dp[1] = true, 其他为false,即若目标为1,则可以用数组中的元素表示
检测5,dp[1] = true, dp[5] = true, dp[6] = true,即若目标为1, 5, 6,都可以用数组中的元素表示
.....
最后返回dp[target]即可
class Solution {
public:
bool canPartition(vector<int>& nums){
int sum = accumulate(nums.begin(), nums.end(), 0);
int target = sum >> 1;
if (sum & 1) return false; //若序列元素的总和为奇数
vector<bool> dp(target+1, false);
dp[0] = true;
for (int num : nums){
for (int i = target; i >= num; --i)
dp[i] = dp[i] || dp[i-num]; //如果之前的状态已经检测过,就赋给当前状态
}
return dp[target];
}
};