划分为K个相等的子集(元素不可重复用)

题目链接:https://leetcode-cn.com/problems/partition-to-k-equal-sum-subsets
题目描述:
给定一个整数数组  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

题解:


class Solution {
public:
    bool trackingBack(vector<int>& nums, int k, int pathSum, int startIndex, int target, vector<bool> used)
    {
        if(k == 0)                  //获得k组数据,返回true
            return true;
        if(pathSum == target)       //获得一组数据,重置pathSum,遍历索引为0
            return trackingBack(nums, k - 1, 0, 0, target, used);
        for(int i = startIndex; i < nums.size() && pathSum + nums[i] <= target; i++)
        {
            if(used[i])             //使用过的元素不再用
                continue;
            if(i > 0 && nums[i - 1] == nums[i] && used[i - 1] == false)       //同一层存在重复元素,去重
                continue;
            pathSum += nums[i];
            used[i] = true;
            if(trackingBack(nums, k, pathSum, i + 1, target, used))
                return true;
            pathSum -= nums[i];
            used[i] = false;
        }
        return false;

    }

    bool canPartitionKSubsets(vector<int>& nums, int k) {
        vector<bool> used(nums.size(), false);             //标记元素是否被使用
        int sum = 0;
        for(auto &item : nums)
        {
            sum += item;
        }
        if(sum % k != 0)
            return false;
        int target = sum / k;
        sort(nums.begin(), nums.end());
        return trackingBack(nums, k, 0, 0, target, used);
        
    }
    
};

原文地址:https://www.cnblogs.com/ZigHello/p/15094716.html