20190425-Leetcode-473.火柴拼正方形

Leetcode-473.火柴拼真方形

还记得童话《卖火柴的小女孩》吗？现在，你知道小女孩有多少根火柴，请找出一种能使用所有火柴拼成一个正方形的方法。不能折断火柴，可以把火柴连接起来，并且每根火柴都要用到。
输入为小女孩拥有火柴的数目，每根火柴用其长度表示。输出即为是否能用所有的火柴拼成正方形。
示例 1:
输入: [1,1,2,2,2]
输出: true
解释: 能拼成一个边长为2的正方形，每边两根火柴。

示例 2:
输入: [3,3,3,3,4]
输出: false
解释: 不能用所有火柴拼成一个正方形。

注意:
给定的火柴长度和在 0 到 10^9之间。
火柴数组的长度不超过15。

思路：
1.若是nums中数量小于4个，则直接返回false，若nums所有数的和都除以4有余数，则直接返回false
2.对nums数组进行从大到小的排序，这样相当于剪枝，回溯法的时候可以省去一些尝试。
3.用数组存放四个值，相当于四个桶，然后把数值依次存入桶中。然后用深度优先搜素依次遍历，数组中的所有数，依次放入四个桶中，若桶中数值之和加上遍历到数之后和大于sum/4，则放入一个桶尝试，放入完成该数后，再尝试下一个数，若所有数都放置完成，并且每个桶中的和等于sum/4，则返回true，否则，返回false。
代码：

class Solution {
public:
    bool dfs(int id, int sum, vector<int>& nums, int buf[]){
        if(id == nums.size()){
            return buf[0] == sum && buf[1] == sum && buf[2] == sum && buf[3] == sum;
        }
        for(int i = 0; i < 4; i++){
            if(nums[id] + buf[i] > sum){
                continue;
            }
            buf[i] += nums[id];
            if(dfs(id + 1, sum, nums, buf)){
                return true;
            }
            buf[i] -= nums[id];
        }
        return false;
    }
    bool makesquare(vector<int>& nums) {
        if(nums.size() < 4){
            return false;
        }
        sort(nums.rbegin(), nums.rend());
        int sum = 0;
        for(int i = 0; i < nums.size(); i++){
            sum += nums[i];
        }
        if(sum % 4){
            return false;
        }
        int buf[4] = {0};
        return dfs(0, sum /4, nums, buf);
    }
};