Partition to K Equal Sum Subsets

题目描述

解题思路

题目大意如下，给你一个一维int数组，且每个数大于0，再给你一个数字k，问是否可以将该数组分成k个子集，这k个子集的并集刚好为该int数组，交集为空，最关键的是，这k个子集每个子集的数字之和都相等。
仔细思考不难发现，要想满足题意，首先应该判断这个int数组的总和sum是否可以整除k，若能整除则就到了本题目的难点，怎么才能刚好划分成k个子集每个子集的数字和等于sum/k呢？
我想到的是先对数组进行排序，之后从大到小开始递推考虑，假设函数gofind(int target, int index)代表从下标index开始，能否从数组中找到某几个数，他们的和刚好等于target呢？（并且我优先考虑大数,相加的这些数令其为0以表示被用过下次不能再被使用，即从大数组中删掉。）那么gofind(int target, int index)便取决于gofind(target-nums[i], i-1)啦。
话不多说，上代码~

C++代码实现

class Solution {
public:
    bool gofind(vector<int>& nums, int target, int index, int& countt) {
    //cout << "要在前" << index+1 << "个数字找有没有和为" << target << "的" << endl;
    if (target < 0) {
        return false;
    }
    // 能否以当前下标index开始，寻找某几个数字，加起来和为target
    //（优先考虑大数,相加的这些数令其为0以表示被用过） 
    for (int i = index; i >= 0; --i) {
        if (nums[i] == 0 || nums[i] > target) {
            continue;
        } else if (nums[i] == target) {
            countt++;
            //cout << nums[i] << " " << i << endl;
            nums[i] = 0;
            return true;
        } else {
            //cout << target << "-(" << i << ")" << nums[i] << endl;
            if (gofind(nums, target-nums[i], i-1, countt)) {
                nums[i] = 0;
                return true;    
            }
        }

    }
    return false;
}
    bool canPartitionKSubsets(vector<int>& nums, int k) {
     if (k == 1) {
        return true;
    }
    int sum = 0;
    int length = nums.size();
    for (int i = 0; i < length; ++i) {
        sum += nums[i];
    }
    if (sum % k != 0) {
        return false;
    }
    int target = sum / k;
        int countt = 0;
    //cout << sum << " " << target << endl;
    sort(nums.begin(), nums.end());
    for (int i = length - 1; i >= 0; --i) {
        if (nums[i] > target) {
            return false;
        } else if (nums[i] == target) {
            countt++;
            //cout << countt << " " <<  nums[i] << " " << i << endl;
            nums[i] = 0;  // 已被标记 
        } else if (nums[i] == 0) {
            continue;
        } else {
            if (!gofind(nums, target, i, countt)) {
                return false;
            }
        }
    }

    for (int i = 0; i < length; ++i) {
        if (nums[i] != 0) {
            // cout << nums[i] << " " << i << endl;
            return false;
        }
    }
    //cout << countt << endl;
    return countt == k;
    }
};

题目样例运行截图

反思

感觉这道题的思路还算ok，不过复杂度的话确实蛮高的。而且在实现过程中，没能很好的考虑函数传参的问题，导致debug了一晚。。。总的来说，收获蛮多的~