1.解析
题目大意,数组中的所有元素是否可以组成正方形?
2. 解析
这道题本质上和Partition Equal Subset Sum是一样的,这里是求解数组元素组成4个相等的数,首先,如果数组所有元素的和不能整除4,那肯定不满足条件。这里要注意的一点是,让数组从大到小进行排列,主要是为了避免重复扫描小的元素。不然无法OJ。
class Solution {
public:
bool makesquare(vector<int>& nums){
if (nums.empty() || nums.size() < 4) return false;
int sum = accumulate(nums.begin(), nums.end(), 0);
if (sum % 4 != 0) return false; //和不满足4的倍数,则不可能分解成4个相等的数
vector<int> sides(4, 0);
sort(nums.rbegin(), nums.rend()); //从 大->小 排列,防止递归重复扫描
return DFS(nums, 0, sides, sum / 4);
}
bool DFS(vector<int>& nums, int pos, vector<int> sides, int target){
if (pos >= nums.size()){
return sides[0] == target && sides[1] == target && sides[2] == target;
}
for (int i = 0; i < 4; ++i){
if (sides[i] + nums[pos] > target) continue; //如若当前数的值大于目标值(总和的1/4)
sides[i] += nums[pos];
if (DFS(nums, pos + 1, sides, target)) return true;
sides[i] -= nums[pos];
}
return false;
}
};
类似的题目 :