Matchsticks to Square

原創

Czy_whlg

2020-02-21 15:16

1.解析

題目大意，數組中的所有元素是否可以組成正方形？

2. 解析

這道題本質上和Partition Equal Subset Sum是一樣的，這裏是求解數組元素組成4個相等的數，首先，如果數組所有元素的和不能整除4，那肯定不滿足條件。這裏要注意的一點是，讓數組從大到小進行排列，主要是爲了避免重複掃描小的元素。不然無法OJ。

class Solution {
public:
    bool makesquare(vector<int>& nums){
        if (nums.empty() || nums.size() < 4) return false;
        int sum = accumulate(nums.begin(), nums.end(), 0);
        if (sum % 4 != 0) return false; //和不滿足4的倍數，則不可能分解成4個相等的數
        vector<int> sides(4, 0);
        sort(nums.rbegin(), nums.rend()); //從 大->小 排列，防止遞歸重複掃描
        return DFS(nums, 0, sides, sum / 4);        
    }
    
    bool DFS(vector<int>& nums, int pos, vector<int> sides, int target){
        if (pos >= nums.size()){
            return sides[0] == target && sides[1] == target && sides[2] == target;
        }
        for (int i = 0; i < 4; ++i){
            if (sides[i] + nums[pos] > target) continue; //如若當前數的值大於目標值(總和的1/4)
            sides[i] += nums[pos];
            if (DFS(nums, pos + 1, sides, target)) return true;
            sides[i] -= nums[pos];
        }
        return false;
    }
};

類似的題目 ：

Partition Equal Subset Sum