力扣 473. 火柴拼正方形

题目

你将得到一个整数数组 matchsticks ，其中 matchsticks[i] 是第 i 个火柴棒的长度。你要用 所有的火柴棍 拼成一个正方形。你 不能折断 任何一根火柴棒，但你可以把它们连在一起，而且每根火柴棒必须 使用一次 。

如果你能使这个正方形，则返回 true ，否则返回 false 。

示例

输入: matchsticks = [1,1,2,2,2]
输出: true
解释: 能拼成一个边长为2的正方形，每边两根火柴。

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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/matchsticks-to-square
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：回溯

Java实现

class Solution {
    public boolean makesquare(int[] matchsticks) {
        int sum = 0, n = matchsticks.length;
        Arrays.sort(matchsticks);
        for (int num : matchsticks) sum += num;

        if (sum % 4 != 0) return false;
    
        int l = 0, r = n - 1;
        while (l < r) {
            int tmp = matchsticks[l];
            matchsticks[l] = matchsticks[r];
            matchsticks[r] = tmp; 
            l++;
            r--;
        }

        int[] edges = new int[4];
        int len = sum / 4;
        return track_back(0, edges, len, matchsticks);
    }

    public boolean track_back(int idx, int[] edges, int len, int[] matchsticks) {
        if (idx == matchsticks.length) return true;

        for (int i = 0; i < edges.length; i++) {
            edges[i] += matchsticks[idx];
            if (edges[i] <= len && track_back(idx + 1, edges, len, matchsticks)) return true;
            edges[i] -= matchsticks[idx];
        }

        return false;
    }
}