还记得童话《卖火柴的小女孩》吗?现在,你知道小女孩有多少根火柴,请找出一种能使用所有火柴拼成一个正方形的方法。不能折断火柴,可以把火柴连接起来,并且每根火柴都要用到。
输入为小女孩拥有火柴的数目,每根火柴用其长度表示。输出即为是否能用所有的火柴拼成正方形。
示例 1:
输入: [1,1,2,2,2] 输出: true 解释: 能拼成一个边长为2的正方形,每边两根火柴。
示例 2:
输入: [3,3,3,3,4] 输出: false 解释: 不能用所有火柴拼成一个正方形。
注意:
- 给定的火柴长度和在
0
到10^9
之间。 - 火柴数组的长度不超过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);
}
};