[LeetCode] Partition to K Equal Sum Subsets

题目

分析

这题虽然归在DP里面，但是最后其实用的是dfs进行求解的。因为这题看起来如果使用dfs会更加的直观，和之前的combination sum2之类的其实会比较相似。
这题的关键问题在于超时了，我也不明白为什么会超时，但是最后我把sort那部分给去掉就没问题了，但是这部分只是O(logN),可能是因为排序之后会导致找的这个过程变慢，反而是无序的更容易找到问题的解。

时间复杂度分析

时间复杂度，由于只是DFS，而且是一个无向图，所以时间复杂度应该是O($n^2$)

代码

其中count是用来避免k == 0 的特殊情况的

#include <iostream>
#include <vector>
using namespace std;

class Solution {
public:
    bool canPartitionKSubsets(vector<int>& nums, int k) {
        // special cases
        if (k == 1) return true;
        if (nums.size() < k) return false;
        int sum = 0;
        for (int i = 0; i < nums.size(); i++) sum += nums[i];
        if (sum % k != 0) return false;
        subsetSum = sum / k;
        visit = vector<bool>(nums.size(), false);
        this->size = nums.size();
        this->_nums = nums;
        return dfs(0, 0, k, 0);
    }

    bool dfs(int startIndex, int sum, int remainder, int count) {
        // if only remain 1 subset, it can always be divided
        if (remainder == 1) return true;
        // successfully divede a subset and begin a recursive divide
        if (sum == subsetSum && count > 0) return dfs(0, 0, remainder -1, 0);

        for (int i = startIndex; i < size; i++) {
            if (!visit[i] && sum + _nums[i] <= subsetSum) {
                visit[i] = true;
                if (dfs(i+1, sum + _nums[i], remainder, count+1)) return true;
                visit[i] = false;
            }
        }
        return false;
    }

    vector<bool> visit;
    vector<int> _nums;
    int subsetSum;
    int size;
};