题解 | #划分等和序列#

划分等和序列

http://www.nowcoder.com/practice/89ceab10d36140ce8f0d38c56e417f02

回溯 首先对 nums 求和, 如果 s % k != 0 说明没有 k 个组合,判断 nums 中最大的值是否大于 s // k ,如果大于则也不符合条件; 进行递归,直到 k = 0 或 k = 1, 每次记录已经访问过的节点并回溯 代码如下:

#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
# 
# @param nums int整型一维数组 
# @param k int整型 
# @return bool布尔型
#
class Solution:
    def candivide(self , nums: List[int], k: int) -> bool:
        # write code here
        nums.sort(reverse=True)
        s = sum(nums)
        if s % k != 0 or nums[0] > s // k:
            return False
        visited = [False for _ in range(len(nums))]
        def back(k, target,visited):
            if k == 0:
                return True
            if target == 0:
                return back(k - 1, target, visited)
            for i in range(len(nums)):
                if visited[i]:
                    continue
                if target < nums[i]:
                    continue
                visited[i] = True
                if back(k, target - nums[i], visited):
                    return True
                visited[i] = False
            return False
        return back(k, s // k, visited)
全部评论

相关推荐

程序员花海:实习和校招简历正确格式应该是教育背景+实习+项目经历+个人评价 其中项目经历注意要体现业务 实习经历里面的业务更是要自圆其说 简历模板尽可能保持干净整洁 不要太花哨的
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务