Leetcode 473. 火柴拼正方形： 回溯剪枝搜素

 

显然，如果数字之和不是4的倍数，那么一定不能拼成功，直接返回。最后拼接的结果就是4个sum/4的边长，那么如何来搜索呢？

搜索的方法是每次看0到3号桶，然后层级是index，需要把所有nums，都放入到0-3号桶中，并且把桶放满

level1       1 2 3 4

level2       1 2 3 4

level3       1 2 3 4

.... 

leveln       1 2 3 4

如果不能凑成，火柴一定不会用完

class Solution {
public:

    bool makesquare(vector<int>& nums) {
        int sum = 0;
        if(nums.size()<4) return false;
        for(auto num:nums) sum+=num;
        if(sum%4) return false;
        sort(nums.begin(),nums.end(), greater<int>());
        vector<int> edge(4);
        return helper(nums,0,edge,sum/4);
    }

    bool helper(vector<int> &nums, int index, vector<int> & edge, int k){
        if(index==nums.size()) return true;
        for(int i=0;i<4;i++){
            if(edge[i]+nums[index]>k) continue;
            edge[i]+=nums[index];
            if(helper(nums,index+1,edge,k)) return true;
            edge[i]-=nums[index];
        }
        return false;
    }

};