力扣473——火柴拼正方形（DFS回溯+剪枝）

题目（中等）

还记得童话《卖火柴的小女孩》吗？现在，你知道小女孩有多少根火柴，请找出一种能使用所有火柴拼成一个正方形的方法。不能折断火柴，可以把火柴连接起来，并且每根火柴都要用到。

输入为小女孩拥有火柴的数目，每根火柴用其长度表示。输出即为是否能用所有的火柴拼成正方形。

示例 1:

输入: [1,1,2,2,2]
输出: true

解释: 能拼成一个边长为2的正方形，每边两根火柴。
示例 2:

输入: [3,3,3,3,4]
输出: false

解释: 不能用所有火柴拼成一个正方形。
注意:

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

解题思路

首先排除总和不能构成的情况，并计算出正方形边长；
之后对于每个火柴，可以在四条边中任选，进行dfs深搜；
选一条边后回溯的选另一条边，直到处理完所有火柴；
中间配合剪枝进行优化，对火柴降序排序，先处理大火柴，能更快的剪枝。

代码

class Solution {
public:
    vector<int> m;
    bool dfs(int pos, int a1, int a2, int a3, int a4) {
        if(pos == m.size()) {
            return (a1 == 0 && a2 == 0 && a3 == 0 && a4 == 0);
        }
        int x = m[pos];
        if(a1 < 0 || a2 < 0 || a3 < 0 || a4 < 0) return false;        
        return dfs(pos+1,a1-x,a2,a3,a4) || dfs(pos+1,a1,a2-x,a3,a4) || dfs(pos+1,a1,a2,a3-x,a4) || dfs(pos+1,a1,a2,a3,a4-x);
    }
    bool makesquare(vector<int>& matchsticks) {
        long long sum = 0;
        m = matchsticks;
        sort(m.begin(), m.end(), greater<int>());
        int n = m.size();
        for(auto x : m) sum += x;
        if(sum % 4 != 0) return false;
        long long l = sum / 4;
        for(auto x : m) {
            if(x > l) return false;
        }
        return dfs(0, l, l, l, l);
    }
};