Leetcode 698. 划分为k个相等的子集

给定一个整数数组 nums 和一个正整数 k,找出是否有可能把这个数组分成 k 个非空子集,其总和都相等。

示例 1:

输入: nums = [4, 3, 2, 3, 5, 2, 1], k = 4
输出: True
说明: 有可能将其分成 4 个子集(5),(1,4),(2,3),(2,3)等于总和。
注意:

1 <= k <= len(nums) <= 16
0 < nums[i] < 10000

思路:由于题目规定只能分成k个和相同的子集合,所以我们是能求出子集合的和 s u m ÷ k sum÷k sum÷k记为target。然后我们通过深度优先搜索,通过一个cursum来记录累加的和是否与我们的target相等,如果相等则继续递归,同时我们所需要的子集合就为k-1了,同时我们要清空cursum和start起始点。如果小于的话就从start开始遍历数组,通过一个标记visted数组来判断是否访问过,没访问过的话就调用递归函数,让我们的cursum累加当前数组访问到的元素值,如果返回true就说明结果是true 否则就将访问重置为false

class Solution {
public:
    bool canPartitionKSubsets(vector& nums, int k) {
        int sum=accumulate(nums.begin(),nums.end(),0);//求和
        if(sum%k!=0)return false;
        vector visited(nums.size(),false);
        return dfs(nums,k,sum/k,0,0,visited);
    }
    bool dfs(vector nums,int k,int target,int cursum,int start,vector visited){
        if(k==1)return true;//此时只需要一个子集合 当前的就是 满足
        else if(cursum>target)return false;
        else if(cursum==target){
            return dfs(nums,k-1,target,0,0,visited);//我们找到了当前符合的子集合,要继续查找下一个,同时要把cursum和start清零
        }
        else {
            for(int i=start;i

你可能感兴趣的:(LeetCode算法)