[LeetCode] 473. Matchsticks to Square

题：https://leetcode.com/problems/matchsticks-to-square/

题目大意

数组nums 中只有 正整数，能否将 nums 中的元素 放入 4 个集合中，使得 每个集合中的元素和相同。

思路

和 416. Partition Equal Subset Sum非常相似，可以说是 前者的进阶版。先看上一道再做下一道会好很多。

使用 dfs，同样 对nums 进行了排序优化。

在最后判断中，若能将nums中 所有元素都分出去了，其实答案一定有解，所以判断 target ==0 ，没有必要。

未排序优化
Runtime: 3905 ms

class Solution {
    public boolean dfs(int[] nums,int index ,int[] target){
        if(index == nums.length)
            // return target[0] == 0 && target[1] == 0 && target[2] == 0 && target[3] == 0;
            return true;
        for(int i = 0 ; i <4 ;i++){
            if(target[i]>=nums[index]){
                target[i] -= nums[index];
                if(dfs(nums,index+1,target))
                    return true;
                target[i] += nums[index];       
            }
        }
        return false;
    }
    
    public boolean makesquare(int[] nums) {
        int _sum = 0 ;
        for(int num:nums){
            _sum += num;
        }
        if(_sum == 0 || (_sum &4) == 1)
            return false;
        _sum = _sum >>2;
        int[] target ={_sum,_sum,_sum,_sum};
        return dfs(nums,0,target);
    }
}

排序优化

class Solution {
    public boolean dfs(int[] nums,int index ,int[] target){
        if(index < 0)
            return target[0] == 0 && target[1] == 0 && target[2] == 0 && target[3] == 0;
        for(int i = 0 ; i <4 ;i++){
            if(target[i] >= nums[index]){
                target[i] -= nums[index];
                if(dfs(nums,index-1,target))
                    return true;
                target[i] += nums[index];
            }
        }
        return false;
    }
    
    public boolean makesquare(int[] nums) {
        int _sum = 0 ;
        for(int num:nums){
            _sum += num;
        }
        if(_sum == 0 || (_sum &4) == 1)
            return false;
        _sum = _sum >>2;
        int[] target ={_sum,_sum,_sum,_sum};
        Arrays.sort(nums);
        return dfs(nums,nums.length-1,target);
    }
}