473. 火柴拼正方形

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

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

示例1:

image.png

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

示例2:

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

提示:

  • 1 <= matchsticks.length <= 15
  • 1 <= matchsticks[i] <= 108

方法:回溯算法

class Solution {
        public static boolean makesquare(int[] matchsticks) {
        int sum = 0, len = matchsticks.length;
        for (int matchstick : matchsticks) {
            sum += matchstick;
        }
        if (sum % 4 != 0) {
            return false;
        }
        // 排序
        Arrays.sort(matchsticks);
        return backtrace(matchsticks, sum >> 2, len - 1, new int[4]);
    }

    static boolean backtrace(int[] matchsticks, int target, int index, int[] size) {
        if (index == -1) {
            // 火柴用完了
            return size[0] == size[1] && size[1] == size[2] && size[2] == size[3];
        }
        for (int i = 0; i < 4; i++) {
            // size[i] == size[i - 1]即上一个分支的值和当前分支的一样,上一个分支没有成功,
            if (size[i] + matchsticks[index] > target || (i > 0 && size[i] == size[i - 1])) {
                continue;
            }
            //如果当前火柴放到size[i]这个边上,长度不大于target,我们就放上面
            size[i] += matchsticks[index];
            if (backtrace(matchsticks, target, index - 1, size)) {
                return true;
            }
            size[i] -= matchsticks[index];
        }
        return false;
    }
}

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

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容