leetcode-416. 分割等和子集

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求和結果的一半。