LeetCode698. 划分为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

解题思路：

主要是深度优先遍历DFS思想，将数组nums划分为k个子集，每个子集和为nums/k。 设置一个vector<int> flag来记录nums数组内元素的使用情况。 此题与leetcode 40.组合总和 2的递归方法类似。

代码如下：

class Solution {
public:
    bool canPartitionKSubsets(vector<int>& nums, int k) {
        int sum = 0;
        for(auto i : nums)
            sum += i;
        if(sum % k != 0) return false;
        sort(nums.begin(),nums.end());
        target = sum / k;
        vector<int> flag(nums.size(),0);
        return dfs(nums, flag, 0, 0, k);
        
    }
    bool dfs(vector<int>& nums, vector<int> flag, int start, int index, int k){
        if(k == 0) return true;
        for(int i = start; i < nums.size() && nums[i] + index <= target; ++i){
            if(flag[i] == 0){
                flag[i] = 1;
                if((nums[i] + index) == target)
                    return dfs(nums, flag, 0, 0, k -1);
                else if(dfs(nums, flag, i + 1, nums[i] + index, k))
                    return true;
                flag[i] = 0;
            }
        }
        return false;
    }
    int target = 0;
};