LeetCode 698. 划分为k个相等的子集（回溯）

文章目录

1. 题目
2. 解题

1. 题目

给定一个整数数组 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/partition-to-k-equal-sum-subsets
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

类似题目：LeetCode 416. 分割等和子集（动态规划）

普通回溯，超时，59 / 149 个通过测试用例

class Solution {int target;bool ans = false;int K;
public:bool canPartitionKSubsets(vector<int>& nums, int k) {int sum = accumulate(nums.begin(), nums.end(), 0);K = k;if(sum%k) return false;target = sum/k;vector<int> partsum(k, 0);dfs(nums, partsum, 0);return ans;}void dfs(vector<int>& nums, vector<int>& partsum, int i){if(ans == true) return;if(i == nums.size()){bool flag = true;for(int k = 0; k < K; k++){if(partsum[k] != target){flag = false;break;}}if(flag)ans = true;return;}for(int j = 0; j < K; ++j){if(partsum[j]+nums[i] <= target){partsum[j] += nums[i];dfs(nums, partsum, i+1);partsum[j] -= nums[i];}}}
};

优化，从大到小排序，可以尽快的让每个部分的和凑够

class Solution {int target;bool ans = false;int K;
public:bool canPartitionKSubsets(vector<int>& nums, int k) {int sum = accumulate(nums.begin(), nums.end(), 0);K = k;if(sum%k) return false;target = sum/k;sort(nums.rbegin(), nums.rend());//从大到小排序if(nums[0] > target)//最大的数超过了，不行return false;vector<int> partsum(k, 0);dfs(nums, partsum, 0);return ans;}void dfs(vector<int>& nums, vector<int>& partsum, int i){if(ans == true) return;if(i == nums.size()){bool flag = true;for(int k = 0; k < K; k++){if(partsum[k] != target){flag = false;break;}}if(flag)ans = true;return;}for(int j = 0; j < K; ++j){if(partsum[j]+nums[i] <= target){partsum[j] += nums[i];dfs(nums, partsum, i+1);partsum[j] -= nums[i];}}}
};

168 ms 9.2 MB

我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号（Michael阿明），一起加油、一起学习进步！

LeetCode 698. 划分为k个相等的子集（回溯）相关推荐

Leetcode 698. 划分为k个相等的子集解题思路及C++实现
解题思路: 这是一个典型的深度优先搜索问题.使用一个相同大小的数组visited来记录元素是否被访问过(即是被已被划分到相应的子集中). 每找到一个划分子集,相应对k减一,再继续执行深度优先搜索程序. ...
698. 划分为k个相等的子集：给定一个整数数组 nums 和一个正整数 k，找出是否有可能把这个数组分成 k 个非空子集，其总和都相等。
题目描述给定一个整数数组 nums 和一个正整数 k,找出是否有可能把这个数组分成 k 个非空子集,其总和都相等. 示例 1: 输入: nums = [4, 3, 2, 3, 5, 2, 1], k ...
leetcode 698. Partition to K Equal Sum Subsets | 698. 划分为k个相等的子集（回溯法）
题目 https://leetcode.com/problems/partition-to-k-equal-sum-subsets/ 题解一上来以为是 dp(想到了左神讲的,将一个数组分成两个尽可能 ...
力扣 698. 划分为k个相等的子集
题目给定一个整数数组 nums 和一个正整数 k,找出是否有可能把这个数组分成 k 个非空子集,其总和都相等. 示例输入: nums = [4, 3, 2, 3, 5, 2, 1], k = 4 ...
leetcode算法题--划分为k个相等的子集★
原题链接:https://leetcode-cn.com/problems/partition-to-k-equal-sum-subsets/ 相关问题:分隔等和子集递归的方法其实不难想,但是很难写 ...
Leetcode 1400：构造K个回文字符串（超详细的解法！！！）
给你一个字符串 s 和一个整数 k .请你用 s 字符串中所有字符构造 k 个非空回文串 . 如果你可以用 s 中所有字符构造 k 个回文字符串,那么请你返回 True ,否则返回 False ...
LeetCode 698.Partition_to_k_equal_subsets. Three different solutions: DFS, DP, DP topdown + DFS
Problem Partition a set A=(a1,a2,...,an)A=(a_1,a_2,...,a_n)A=(a1,a2,...,an) to kkk disjointed sub ...
划分计算机发展的四个阶段主要依据,计算机的发展划分为4个阶段，分别为什么、什么、什么和什么。...
计算机的发展划分为4个阶段,分别为什么.什么.什么和什么.以下文字资料是由(历史新知网www.lishixinzhi.com)小编为大家搜集整理后发布的内容,让我们赶快一起来看一下吧! 计算机的发展划 ...
把一个c类的网络地址192-168-1-0划分为6个子网，请计算出...
6<2^3=8 所以可划分为8个等长子网,子网号从主机号位借3位,每个子网可容纳2^5-2=30台主机,其中主机号全1全0不可用 11000000.10101000.00000001.00000 ...

LeetCode 698. 划分为k个相等的子集（回溯）

文章目录

1. 题目

2. 解题

LeetCode 698. 划分为k个相等的子集（回溯）相关推荐

最新文章

热门文章