LeetCode 火柴拼正方形（深度优先搜搜）

还记得童话《卖火柴的小女孩》吗？现在，你知道小女孩有多少根火柴，请找出一种能使用所有火柴拼成一个正方形的方法。不能折断火柴，可以把火柴连接起来，并且每根火柴都要用到。
输入为小女孩拥有火柴的数目，每根火柴用其长度表示。输出即为是否能用所有的火柴拼成正方形。
示例 1:

输入: [1,1,2,2,2]
输出: true
解释: 能拼成一个边长为2的正方形，每边两根火柴。

示例 2:

输入: [3,3,3,3,4]
输出: false
解释: 不能用所有火柴拼成一个正方形。

注意:

给定的火柴长度和在 0 到 10^9之间。
火柴数组的长度不超过15。

思路分析：首先我们可以对所有火柴进行求和，如果sum不能整除4，说明无论如何都无法拼凑出正方形。然后我们尝试将火柴分配到四个边长，如果最后四边都相等，则说明可以拼凑。

class Solution {
public:
	bool makesquare(vector<int>& nums) {
		if (nums.size() == 0) {
			return false;
		}
		int sum = 0;//求和
		for (auto num : nums) {
			sum += num;
		}
		if (sum % 4 != 0) {//剪枝
			return false;
		}
		int target = sum / 4;
		return myDFS(nums, 0, target, target, target, target);
	}
	//oneNeed，twoNeed，threeNeed，fourNeed是四个边剩余需要的长度，index是nums数组的扫描下标
	bool myDFS(vector<int> &nums, int index, int oneNeed, int twoNeed, int threeNeed, int fourNeed) {
		if (index == nums.size()) {//判断是否成功
			return oneNeed == twoNeed && twoNeed == threeNeed && && threeNeed == fourNeed;
		}
		//剪枝2，不能超过需要的长度
		if (oneNeed < 0 || twoNeed < 0 || threeNeed < 0 || fourNeed < 0) {
			return false;
		}
		//尝试分配到四条边
		return myDFS(nums, index + 1, oneNeed - nums[index], twoNeed, threeNeed, fourNeed) ||
			myDFS(nums, index + 1, oneNeed, twoNeed - nums[index], threeNeed, fourNeed) ||
			myDFS(nums, index + 1, oneNeed, twoNeed, threeNeed - nums[index], fourNeed) ||
			myDFS(nums, index + 1, oneNeed, twoNeed, threeNeed, fourNeed - nums[index]);
	}
};

这种算法就是暴搜，其中剪枝2的效果并不明显，我们将nums进行降序排序，这在尝试分配到四条边的时候就能快速进行剪枝2.。

class Solution {
public:
	bool makesquare(vector<int>& nums) {
		if (nums.size() == 0) {
			return false;
		}
		int sum = accumulate(nums.begin(), nums.end(), 0);//累计求和
		if (sum % 4 != 0) {//剪枝
			return false;
		}
		int target = sum / 4;
        sort(nums.rbegin(), nums.rend());//降序排序
		return myDFS(nums, 0, target, target, target, target);
	}
	//oneNeed，twoNeed，threeNeed，fourNeed是四个边剩余需要的长度，index是nums数组的扫描下标
	bool myDFS(vector<int> &nums, int index, int oneNeed, int twoNeed, int threeNeed, int fourNeed) {
		if (index == nums.size()) {//判断是否成功
			return oneNeed == twoNeed && twoNeed == threeNeed && threeNeed == fourNeed;
		}
		//剪枝，不能超过需要的长度
		if (oneNeed < 0 || twoNeed < 0 || threeNeed < 0 || fourNeed < 0) {
			return false;
		}
		//尝试分配到四条边
		return myDFS(nums, index + 1, oneNeed - nums[index], twoNeed, threeNeed, fourNeed) ||
			myDFS(nums, index + 1, oneNeed, twoNeed - nums[index], threeNeed, fourNeed) ||
			myDFS(nums, index + 1, oneNeed, twoNeed, threeNeed - nums[index], fourNeed) ||
			myDFS(nums, index + 1, oneNeed, twoNeed, threeNeed, fourNeed - nums[index]);
	}
};