第9天,恩,忘记洗衣服了,写完就去洗。
今天的题目是Matchsticks to Square。
没想到是一道暴搜的题。。。
既然是暴搜,那思路就比较简单了:
我们先计算出数组和,然后除以4就是每条边的长度了,然后用暴力搜索的方式看是否能连续四次移除长度之和为边长的火柴。
暴搜时可以用一些小技巧,比如先排序来保证先用掉比较长的火柴,这样一旦发现,剩下的火柴比需要的长度还长就可以直接判定失败了,减少搜索次数。
而且标记是否使用掉该火柴的used
数组也可以省略掉,因为火柴长度一定大于0, 可以用其相反数来表示已经用掉了
1class Solution {2public:3 bool makesquare(vector<int>& nums) {4 int len = nums.size();5 if (len == 0) return false;6 sort(nums.rbegin(), nums.rend());7
8 int sum = 0;9 for(auto &i: nums) sum+=i;10
11 if (sum % 4) return false;12
13 for(int i = 0;i < 4; i++) {14 if (dfs(nums, sum/4) == false) return false;15 }16 collapsed lines
16 return true;17 }18 bool dfs(vector<int> &nums, int len) {19 if (len == 0) return true;20 else if (len < 0) return false;21
22 for(int i = 0;i < nums.size(); i++) {23 if (len < nums[i]) return false;24 if (nums[i] <= 0) continue;25 nums[i] = -nums[i];26 if (dfs(nums, len + nums[i])) return true;27 nums[i] = -nums[i];28 }29 return false;30 }31};