思路一:暴力dfs,时间复杂度为O(n^n),时间复杂度有些高。
思路二:子集划分+子集个数有限制->状压dp 时间复杂度O(n * 3^n)是固定的
状压dp相关知识:
// 遍历u的所有非空子集
for (int s = u; s; s = (s - 1) & u) {
//s表示u的一个子集 u^s表示去掉s后的子集
}
具体思路:
dp[i][j]表示消耗的集合个数为i,选择的饼干为j所获得饼干最大总数的最小值。
i的取值为1->k j的取值为1->(1<<n)
dp[i][j]=min(max(dp[i-1][j^s], sum[s]));
class Solution {
public:
int distributeCookies(vector<int>& cookies, int k) {
int n = cookies.size();
vector<int> sum(1 << n);
for (int i = 1; i < (1 << n); ++i) {
for (int j = 0; j < n; ++j) {
if (i & (1 << j)) sum[i] += cookies[j];
}
}
vector<vector<int>> dp(k, vector<int>(1 << n));
//dp[0]表示消耗1组,所获得的值
dp[0] = sum;
for (int i = 1; i < k; ++i) {
for (int j = 1; j < (1 << n); ++j) {
dp[i][j] = INT_MAX;
for (int s = j; s; s = (s - 1) & j) {
dp[i][j] = min(dp[i][j], max(dp[i - 1][j ^ s], sum[s]));
}
}
}
return dp[k - 1][(1 << n) - 1];
}
};