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

最新推荐文章于 2024-08-27 00:14:06 发布

IMBA鑫鑫

最新推荐文章于 2024-08-27 00:14:06 发布

阅读量573

点赞数 1

分类专栏： LeetCode算法文章标签： Leetcode 算法

本文链接：https://blog.csdn.net/wangjianxin97/article/details/89255388

版权

LeetCode算法专栏收录该内容

9 篇文章 0 订阅

订阅专栏

给定一个整数数组 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 \div k$ 记为target。然后我们通过深度优先搜索，通过一个cursum来记录累加的和是否与我们的target相等，如果相等则继续递归，同时我们所需要的子集合就为k-1了，同时我们要清空cursum和start起始点。如果小于的话就从start开始遍历数组，通过一个标记visted数组来判断是否访问过，没访问过的话就调用递归函数，让我们的cursum累加当前数组访问到的元素值，如果返回true就说明结果是true 否则就将访问重置为false

class Solution {
public:
    bool canPartitionKSubsets(vector<int>& nums, int k) {
        int sum=accumulate(nums.begin(),nums.end(),0);//求和
        if(sum%k!=0)return false;
        vector<bool> visited(nums.size(),false);
        return dfs(nums,k,sum/k,0,0,visited);
    }
    bool dfs(vector<int> nums,int k,int target,int cursum,int start,vector<bool> 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<nums.size();++i){
                if(visited[i]==false){
                    visited[i]=true;//进行标记
                    if(dfs(nums,k,target,cursum+nums[i],i+1,visited))return true;
                    visited[i]=false;//否则重置设为未访问
                }
            }
            return false;
        }
        
    }
};