回溯 首先对 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)