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.对nums进行逆序排序，若sum(nums) % k不是整数，即不能把nums分为k个和为target的子集，返回false。
2.若nums[0] > target,则直接返回false。PS:不加这句会导致超时。
3.设置标记数组，对使用过的元素进行标记，减少重复计算，
4.在进行递归时,如果sum已经超过target，则应该回溯到上一个状态，直接对nums的下一个元素进行递归计算。
5.在不能找到符合条件的子集时，返回false。

class Solution {
public:
    bool canPartitionKSubsets(vector<int>& nums, int k)
    {
        int sum = 0;
        for(int i = 0; i < nums.size(); ++i)
        {
            sum += nums[i];
        }
        if(sum % k != 0) return false;//不能被整除,返回
        int target = sum / k;
        sort(nums.begin(), nums.end(), greater<int>());//从大到小排序,在计算dfs中的sum时,从最大的元素开始累加可以减少递归的次数
        if(target < nums[0]) return  false;//不加这一句会超时
        vector<int> mark(nums.size(),0);//标记数组,标记该元素已经被使用过,减少重复次数
        return dfs(nums, mark, 0, k, target);
    }

    bool dfs(vector<int>& nums, vector<int>& mark, int sum, int k, int target)
    {
        if(k == 0) return true;//找到了k个子集

        if(sum == target) return dfs(nums, mark, 0, k-1, target);//找到了一个和为target的子集,sum置0,k减去一

        for(int i = 0; i < nums.size(); ++i)//依次循环计算sum
        {
            if(mark[i]) continue;//如果该元素已经被其他子集占有,则直接跳过
            if(sum > target) return false;//sum比target大,则直接返回false

            //将该元素标记为使用过
            mark[i] = 1;
            if(dfs(nums, mark, sum + nums[i], k, target)) return true;//一直递归知道找到k个子集时,返会true

            mark[i] = 0;//回溯后,该元素不符合要求,将该元素的使用标志置为0
        }
        return false;//遍历完整个数组都不能找到符合要求的子集
    }
};