【日常训练】473. 火柴拼正方形

题目

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

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

示例 1:

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

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

提示:
1 <= matchsticks.length <= 15
1 <= matchsticks[i] <= 10⁸

代码

package dayLeetCode;

import java.util.Arrays;

public class dayleetcode473 {
    // dfs
    public boolean makesquare(int[] matchsticks) {
        int sum = 0;
        for (int i = 0; i < matchsticks.length; i++){
            sum += matchsticks[i];
        }
        if (sum % 4 != 0){
            return false;
        }
        int len = sum / 4;
        // 存储四个边的长度
        int[] edges = new int[4];

        // 从大到小排序
        Arrays.sort(matchsticks);
        for (int i = 0, j = matchsticks.length - 1; i < j; i++,j--){
            int tmp = matchsticks[i];
            matchsticks[i] = matchsticks[j];
            matchsticks[j] = tmp;
        }

        // index为matchsticks的下标 用来标记当前处理到哪一根火柴
        return dfs(0, matchsticks, edges, len);
    }

    private boolean dfs(int index, int[] matchsticks, int[] edges, int len) {
        // 全部处理完
        if (index == matchsticks.length){
            return true;
        }
        for (int i = 0; i < edges.length; i++){
            edges[i] += matchsticks[index];
            if (edges[i] <= len && dfs(index + 1,matchsticks,edges,len)){
                return true;
            }
            edges[i] -= matchsticks[index];
        }
        return false;
    }

    public static void main(String[] args) {
        dayleetcode473 obj = new dayleetcode473();
        System.out.println(obj.makesquare(new int[]{1, 1, 2, 2, 2}));
    }
}