class Solution {
bool search(vector<int>& nums, int idx, int target,
vector<vector<pair<int, bool>>> &dp) {
if (idx >= nums.size() || target < 0)
return false;
if (target == 0)
return true;
if (dp[idx][target].first != -1) {//计算过时
return dp[idx][target].second;//直接返回
}
bool ret = search(nums, idx + 1, target - nums[idx], dp) || search(nums, idx + 1, target, dp);
//修改pair的第一个值,标记为已访问
dp[idx][target].first = 0;//改成-1之外的任意值都可以
//修改pair的第二个值,保存本次的计算结果
dp[idx][target].second = ret;
return ret;
}
public:
bool canPartition(vector<int>& nums) {
if (nums.empty())
return false;
int sum = 0;
for (int i = 0;i < nums.size();++i)
sum += nums[i];
if (sum % 2 != 0)
return false;
int target = sum / 2;
vector<vector<pair<int, bool>>> dp(nums.size(), vector <pair<int, bool>>(target + 1, { -1,false }));
return search(nums, 0, target, dp);
}
};
思路:此题和01背包问题非常相似,先对数组nums求和,如果和为奇数,那么这个数组不可能被分为2个部分。如果是偶数,那么,从头到尾遍历数组,看能否凑出nums求和结果的一半。