深度优先搜索|473.火柴拼正方形

深度优先搜索|473.火柴拼正方形

火柴拼正方形

这个题我本来是想把 i 定为四条边,四条都收到了就结束,但是确实没有办法控制怎么去重复遍历火柴盒,现在的做法是把 i 定为len(matchsticks),然后一次一次的摆到正方形的list里去,一旦发现要超过了就可以摆到下一条边上去,如果摆的下就接着摆。
这里的剪枝是把火柴盒倒序排列了,因为是摆不下了就去下一条边,所以就从大到小往里放比较快一些。

class Solution:
    def makesquare(self, matchsticks: List[int]) -> bool:
        if sum(matchsticks) % 4 != 0:
            return False
        used = [False]*len(matchsticks)
        matchsticks.sort(reverse = True) #剪枝
        bord = [0]*4
        edge = int(sum(matchsticks)/4)
        def backtracking(i):
            if i == len(matchsticks):
                if bord == [edge]*4:
                    return True 
                else: 
                    return False

            for j in range(4):
                if bord[j] + matchsticks[i] > edge: continue 
                bord[j] += matchsticks[i]
                if backtracking(i+1): return True 
                bord[j] -= matchsticks[i]
            return False
        return backtracking(0)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
要实现火柴棍摆成正方形的问题,可以使用回溯法来解决。下面是一个用Python实现的示例代码: ```python def makesquare(nums): if not nums or len(nums) < 4: return False total_length = sum(nums) if total_length % 4 != 0: return False side_length = total_length // 4 nums.sort(reverse=True) sides = [0] * 4 def backtrack(index): if index == len(nums): return sides[0] == sides[1] == sides[2] == side_length for i in range(4): if sides[i] + nums[index] <= side_length: sides[i] += nums[index] if backtrack(index + 1): return True sides[i] -= nums[index] return False return backtrack(0) # 示例输入 nums = [1, 1, 2, 2, 2] result = makesquare(nums) print(result) ``` 这段代码中,我们首先判断了输入列表是否为空或长度小于4,如果是的话,直接返回False。然后计算出总,并判断是否能够被4整除,如果不能,则返回False接下来,我们将列表按降序排序,以便从大到小进行回溯。然后定义一个辅助函数`backtrack`来进行回溯操作。在回溯过程中,我们遍历四个边,如果当前边加上当前火柴棍的长度不超过边长,则将该火柴棍放到当前边上,并递归调用`backtrack`函数继续处理下一个火柴棍。如果最终能够将所有火柴棍放置完毕,并且四个边的长度都等于边长,则返回True,否则返回False。 示例输入中,火柴棍的长度分别为1, 1, 2, 2, 2,可以摆成一个边长为3的正方形,所以输出为True。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值