Matchsticks to Square 火柴拼正方形

还记得童话《卖火柴的小女孩》吗?现在,你知道小女孩有多少根火柴,请找出一种能使用所有火柴拼成一个正方形的方法。不能折断火柴,可以把火柴连接起来,并且每根火柴都要用到。

输入为小女孩拥有火柴的数目,每根火柴用其长度表示。输出即为是否能用所有的火柴拼成正方形。

示例 1:

输入: [1,1,2,2,2]
输出: true

解释: 能拼成一个边长为2的正方形,每边两根火柴。

示例 2:

输入: [3,3,3,3,4]
输出: false

解释: 不能用所有火柴拼成一个正方形。

注意:

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

思路:这道题和Partition to K Equal Sum Subsets 划分为k个相等的子集相同,只是把k换成了4,我们还是用之前的解法(DFS)

参考代码:

class Solution {
public:
bool makesquareCore(vector<int>& nums, vector<bool> &visited,int k,int start,int curSum,int target,int numbers) {
	if (curSum > target) return false;
	if (k == 0 && numbers == nums.size()) return true;
	if (curSum == target) return makesquareCore(nums, visited, k - 1, 0, 0, target,numbers);
	for (int i = start; i < nums.size(); i++) {
		if (!visited[i]) {
			visited[i] = true;
			if (makesquareCore(nums, visited, k, i + 1, curSum + nums[i], target,numbers+1)) return true;
			visited[i] = false;
		}
	}
	return false;
}
bool makesquare(vector<int>& nums) {
    if(nums.empty()) return false;
	int sum = 0;
	for (auto num : nums) sum += num;
	if (sum % 4 != 0) return false;
	sum = sum / 4;
	vector<bool> visited(nums.size(), false);
	return makesquareCore(nums,visited,4,0,0,sum,0);
}
};

猜你喜欢

转载自blog.csdn.net/qq_26410101/article/details/82811618
今日推荐