LeetCode 473. 火柴拼正方形

题目

473. 火柴拼正方形

题目描述

还记得童话《卖火柴的小女孩》吗?现在,你知道小女孩有多少根火柴,请找出一种能使用所有火柴拼成一个正方形的方法。不能折断火柴,可以把火柴连接起来,并且每根火柴都要用到。
输入为小女孩拥有火柴的数目,每根火柴用其长度表示。输出即为是否能用所有的火柴拼成正方形。
示例 1:
输入: [1,1,2,2,2]
输出: true
解释: 能拼成一个边长为2的正方形,每边两根火柴。
示例 2:
输入: [3,3,3,3,4]
输出: false
解释: 不能用所有火柴拼成一个正方形。
注意:
给定的火柴长度和在 0 到 10^9之间。
火柴数组的长度不超过15。

题解

首先可以去除火柴数小于4,总长度不能除于4的并且最大长度大于平均长度的,然后就是把火柴一根根填到四条边看能不能形成正方形
ps 对火柴进行长短排序从大到小放入可以减少搜索时间
pss 好像杭电oj上有道一样的题

代码

 public boolean makesquare(int[] nums) {

        if (nums.length < 4)
            return false;
        int sum = 0, max = 0;
        for (int i = 0; i < nums.length; i++) {
            max = Math.max(max, nums[i]);
            sum += nums[i];
        }
        int avg = sum / 4;
        if (sum % 4 != 0 || max > avg)
            return false;
        Arrays.sort(nums);
        return setnum(nums, nums.length - 1, new int[5], avg);
    }

    private boolean setnum(int[] nums, int index, int[] sums, int avg) {
        if (index == -1)
            return true;

        for (int j = 0; j < 4; j++) {
            if (sums[j] + nums[index] <= avg) {
                sums[j] += nums[index];
                if (setnum(nums, index - 1, sums, avg))
                    return true;
                sums[j] -= nums[index];
            }
        }
        return false;
    }

结果

image.png
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容