火柴拼正方形
题目描述
还记得童话《卖火柴的小女孩》吗?现在,你知道小女孩有多少根火柴,请找出一种能使用所有火柴拼成一个正方形的方法。不能折断火柴,可以把火柴连接起来,并且每根火柴都要用到。
输入为小女孩拥有火柴的数目,每根火柴用其长度表示。输出即为是否能用所有的火柴拼成正方形。
样例
输入: [1,1,2,2,2]
输出: true
解释: 能拼成一个边长为2的正方形,每边两根火柴。
输入: [3,3,3,3,4]
输出: false
解释: 不能用所有火柴拼成一个正方形。
注意:
给定的火柴长度和在 0 到 10^9之间。
火柴数组的长度不超过15。
笔记
经典剪枝优化问题:
所有火柴拼成4条等长的边: a=sum%4==0
依次构造正方形的每条边:
- 从大到小枚举所有边: 如 2 3 4 5 6 7
若剪 2:剪枝砍掉 3467 优先枚举分支少的,剪掉的更多
若剪 7:剪枝砍掉 2345 - 每条边内部的木棒长度规定从大到小
- 当前木棒拼接失败,跳过后面所有长度相同的木棒
- 如果当前木棒拼接失败,且是当前边的第一个,则直接剪掉当前分支
- 如果当前木棒拼接失败,且是当前边的最后一个,则直接剪掉当前分支
代码
class Solution {
public:
vector<int> used;
bool makesquare(vector<int>& nums) {
int sum=0;
for(auto x:nums) sum+=x