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 啦~

class Solution {
public:

    static bool compare(int a, int b) {
        return a > b;
    }
    vector<pair<int, int> > v;

    bool dfs(int len, int pos) {
        for (int i=pos;i<v.size();++i) {
            if (v[i].second == 0 || v[i].first > len) continue;
            if (v[i].first == len) {
                v[i].second--;
                return true;
            }
            v[i].second--;
            if (dfs(len-v[i].first, pos) == true) return true;
            v[i].second++;
            if (pos+1 < v.size()) return dfs(len, pos+1);
        }
        return false;
    }

    bool makesquare(vector<int>& nums) {
        if (nums.size() < 4) return false;
        sort(nums.begin(), nums.end(), compare);
        int sum = accumulate(nums.begin(), nums.end(), 0);
        int len = sum / 4;
        if (sum % 4 != 0 || nums[0] > len) return false;
        for (int i = 0;i<nums.size();++i) {
            int flag = 0;
            for (int j=0;j<v.size();++j) {
                if (v[j].first == nums[i]){
                    v[j].second++;
                    flag = 1;
                    break;
                }
            }
            if (flag == 0) {
                v.push_back(pair<int, int>(nums[i], 1));
            }
        }
        for (int i = 0;i<4;++i) {
            if (dfs(len, 0) == false) return false;
        }
        return true;
    }
};