leetcode 火柴拼正方形 深搜

Posted 2022-07-28 clear-love

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

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

示例?1:

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

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/matchsticks-to-square

思路：1.人为规定每个边长由长火柴开始填(避免更多的情况
2.大顺序每个边长的枚举
3.剪枝：当一个长度的火柴失败之后，与之长度相同的火柴都不行，跳过这些情况；如果组成当前边长火柴是第一个或最后一个，却失败之后，那麽这个大情况就不行，必须全部剪掉

class Solution 
public:
    vector<bool> st;
    bool makesquare(vector<int>& nums) 
        int sum=0;
        st=vector<bool>(nums.size());
        sort(nums.begin(),nums.end());
        reverse(nums.begin(),nums.end());
        for(auto t:nums)sum+=t;
        if(sum%4||!nums.size())return false;
        return dfs(0,0,sum/4,nums,0);
    
    bool dfs(int cur,int len,int max_len,vector<int>& nums,int start)
        if(cur==4)return true;
        if(len==max_len)return dfs(cur+1,0,max_len,nums,0);
        for(int i=start;i<nums.size();++i)
            if(!st[i]&&len+nums[i]<=max_len)
                st[i]=true;
                if(dfs(cur,len+nums[i],max_len,nums,i+1))return true;
                st[i]=false;
                if(!len)return false;
                if(len+nums[i]==max_len)return false;
                while(i+1<nums.size()&&nums[i+1]==nums[i])i++;
            
        
        return false;
    
;