Leetcode: Matchsticks to Square

473. Matchsticks to Square

题意大致是给定一个整数组，判断是否可以分组成4组等和的。

先求和除四，算出每组的大小 Len，如果不能被4整除的就直接返回啦。然后问题变成了数组能不能分成和为 Len 的4组。

直接 brute force 的话需要做4遍检索，这里考虑用 pair 存每个数出现的次数，然后采用先降序排列再dfs，dfs时能取大数尽量取大数，这样组成4组Len就成功了。

尝试后发现逻辑有错误，比如 [5,5,5,5,4,4,4,4,3,3,3,3] ，尽量去大数会取到2个5，然后就返回 false 了。显然 dfs 需要增加一个分支，如果取大数不成功，则要放弃最大的尝试从后面的里面找。见代码的 17-19行。于是成功 ac 啦~

  1 class Solution {
 2 public:
 3 
 4     static bool compare(int a, int b) {
 5         return a > b;
 6     }
 7     vector<pair<int, int> > v;
 8 
 9     bool dfs(int len, int pos) {
10         for (int i=pos;i<v.size();++i) {
11             if (v[i].second == 0 || v[i].first > len) continue;
12             if (v[i].first == len) {
13                 v[i].second--;
14                 return true;
15             }
16             v[i].second--;
17             if (dfs(len-v[i].first, pos) == true) return true;
18             v[i].second++;
19             if (pos+1 < v.size()) return dfs(len, pos+1);
20         }
21         return false;
22     }
23 
24     bool makesquare(vector<int>& nums) {
25         if (nums.size() < 4) return false;
26         sort(nums.begin(), nums.end(), compare);
27         int sum = accumulate(nums.begin(), nums.end(), 0);
28         int len = sum / 4;
29         if (sum % 4 != 0 || nums[0] > len) return false;
30         for (int i = 0;i<nums.size();++i) {
31             int flag = 0;
32             for (int j=0;j<v.size();++j) {
33                 if (v[j].first == nums[i]){
34                     v[j].second++;
35                     flag = 1;
36                     break;
37                 }
38             }
39             if (flag == 0) {
40                 v.push_back(pair<int, int>(nums[i], 1));
41             }
42         }
43         for (int i = 0;i<4;++i) {
44             if (dfs(len, 0) == false) return false;
45         }
46         return true;
47     }
48 };

 

 

转载于:https://www.cnblogs.com/suwish/p/6554455.html