Leetcode 473. 火柴拼正方形 C++

Leetcode 473. 火柴拼正方形

题目

还记得童话《卖火柴的小女孩》吗？现在，你知道小女孩有多少根火柴，请找出一种能使用所有火柴拼成一个正方形的方法。不能折断火柴，可以把火柴连接起来，并且每根火柴都要用到。

输入为小女孩拥有火柴的数目，每根火柴用其长度表示。输出即为是否能用所有的火柴拼成正方形。

示例 1:

输入: [1,1,2,2,2]
输出: true

解释: 能拼成一个边长为2的正方形，每边两根火柴。

示例 2:

输入: [3,3,3,3,4]
输出: false

解释: 不能用所有火柴拼成一个正方形。

注意:

1. 给定的火柴长度和在 0 到 10^9之间。
2. 火柴数组的长度不超过15。

题解

dfs深搜回溯
对于一个正方形，它有四个边长，我们要求这四个边长要相等，我们用int partition[4]数组表示
我们可以先将nums数组进行逆序排序，同时我们求出来可能拼出来的正方形的边长，我们选择当前火柴属于哪个边长时，需要限定加上它后要小于等于目标边长。详细过程见代码

代码

int partition[4];
    bool dfs(int idx,int target,vector<int>& nums){
        if(idx == nums.size()){
            return (partition[0] == partition[1] && partition[1] == partition[2] && partition[2] == partition[3]);
        }
        for(int i=0; i<4; i++){
            if(partition[i] + nums[idx] > target)
                continue;
            partition[i] += nums[idx];
            if(dfs(idx+1,target,nums))  return true;
            partition[i] -= nums[idx];
        }
        return false;
    }
public:
    bool makesquare(vector<int>& nums) {
        int sum = 0;
        if(nums.size() < 4)
            return false;
        for(int i=0; i<nums.size(); i++)
            sum += nums[i];
        if(sum % 4 != 0)
            return false;
        int target = sum / 4;
        sort(nums.begin(), nums.end(),greater<int>());
        return dfs(0,target,nums);
    }

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