文章目录
- 1. 题目
- 2. 解题
1. 题目
还记得童话《卖火柴的小女孩》吗?现在,你知道小女孩有多少根火柴,请找出一种能使用所有火柴拼成一个正方形的方法。
不能折断火柴,可以把火柴连接起来,并且每根火柴都要用到。
输入为小女孩拥有火柴的数目,每根火柴用其长度表示。输出即为是否能用所有的火柴拼成正方形。
示例 1:
输入: [1,1,2,2,2]
输出: true
解释: 能拼成一个边长为2的正方形,每边两根火柴。示例 2:
输入: [3,3,3,3,4]
输出: false解释: 不能用所有火柴拼成一个正方形。
注意:
给定的火柴长度和在 0 到 10^9之间。
火柴数组的长度不超过15。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/matchsticks-to-square
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 解题
- 注意先从大到小排序,不排序会超时,优先拿大的,省掉很多无效的操作
class Solution {bool found = false;int s1_4;vector<int> parts;
public:bool makesquare(vector<int>& nums) {sort(nums.rbegin(), nums.rend());int sum = accumulate(nums.begin(), nums.end(), 0);s1_4 = sum/4;parts = vector<int> (4,0);if(sum%4 || nums.size() < 4 || nums[0] > s1_4) return false;if(nums[0] < s1_4 && nums[0]+nums.back() > s1_4)//都小于1/4,且最大+最小 > 1/4 return false;bt(nums, 0);return found;}void bt(vector<int>& nums, int idx){if(found) return;if(parts[0] > s1_4 || parts[1] > s1_4|| parts[2] > s1_4 || parts[3] > s1_4)return;if(idx==nums.size() && parts[0]==parts[1] && parts[0]==parts[2] && parts[0]==parts[3]){found = true;return;}for(int i = 0; i < 4; ++i){parts[i] += nums[idx];bt(nums, idx+1);parts[i] -= nums[idx];}}
};
204 ms 9.9 MB
我的CSDN博客地址 https://michael.blog.csdn.net/
长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!