leetcode 473. Matchsticks to Square（火柴棍组正方形）

蓝羽飞鸟

已于 2022-07-13 10:57:05 修改

阅读量106

点赞数

分类专栏： leetcode 文章标签： leetcode 算法

于 2022-07-13 10:56:18 首次发布

本文链接：https://blog.csdn.net/level_code/article/details/125759093

版权

leetcode 专栏收录该内容

490 篇文章 4 订阅

订阅专栏

这篇博客讨论了一个利用DFS算法解决的火柴棍问题，目标是用所有火柴棍摆成一个正方形。首先检查火柴棍总数是否能被4整除，然后对火柴棍排序并从大到小进行DFS搜索，确保每条边的长度不超过目标值。在搜索过程中，避免边长超过目标或与已使用的边长相等的情况。该算法通过递归尝试不同的组合，寻找可行的正方形解决方案。

摘要由CSDN通过智能技术生成

在这里插入图片描述

摆火柴棍的问题，要用到所有的火柴棍，摆成一个正方形，也就是4条边长要相等。
可用多个火柴棍拼成一条边。

思路：

DFS
因为要用到所有的火柴棍，而且正方形的4条边长是相等的，边长总长度为所有火柴棍之和，
那么如果边长和不是4的倍数，直接返回false，

下面要按照每条边长为（边长和/4）来DFS

先把火柴棍长度排序，然后从大到小搜索，

边长初始长度为0，加上火柴棍长度 > target(每条边长）时，跳过，
相同的长度也跳过。

具体例子贴个图，链接
在这里插入图片描述

public boolean makesquare(int[] matchsticks) {
    int n = matchsticks.length;
    if(n < 4) return false;
    
    int sum = 0;
    int[] edges = new int[4];
    
    for(int len : matchsticks) {
        sum += len;
    }
    
    if(sum % 4 != 0) return false;
    
    Arrays.sort(matchsticks);
    
    return dfs(matchsticks, edges, n-1, sum/4);
}

boolean dfs(int[] sticks, int[] edges, int index, int target) {
    if(index == -1) return true;
    
    for(int i = 0; i < 4; i ++) {
        if(edges[i] + sticks[index] > target || i > 0 && edges[i] == edges[i-1]) continue;
        
        edges[i] += sticks[index];
        if(dfs(sticks, edges, index-1, target)) return true;
        edges[i] -= sticks[index];
    }
    return false;
}