0回溯/桶 LeetCode473. 火柴拼正方形

最新推荐文章于 2024-10-04 20:45:38 发布

18阿鲁

最新推荐文章于 2024-10-04 20:45:38 发布

阅读量86

点赞数

文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/weixin_43260719/article/details/125708967

版权

本文探讨如何避免将大数盲目填充导致正方形构建问题，通过回溯和剪枝优化算法。介绍使用排序和回溯方法，以及针对数组[5,5,5,5,4,4,4,4,3,3,3,3]的案例分析。关键步骤包括给数组排序、剪枝规则和dfs搜索策略。

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

分析

回溯
不能单纯把较大的数挨个添入桶中就完了，这样会使有的原本可以组正方形的不能构成正方形，比如：[5,5,5,5,4,4,4,4,3,3,3,3]，若只按照大小填充，会在第1个桶里面填成5，5，导致最终四个桶最终不相等，实际应该每个桶添成5，4，3才对。

给数组排序，Arrays.sort()compare方法的参数类型只能使对象，所以实现不了由大到小的排序，只能从右向左遍历了。

回溯把火柴分别尝试填入不同的桶中
剪枝：
1.当前桶与前面的桶容量相同则不遍历当前桶。
2.若已找到满足的方案就不再向后面遍历了。

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

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