划分为k个相等的子集

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

思路:

  •  nums集合的所有元素之和为sum,将nums分成k个非空子集,即每个子集的所有元素和为target = sum / k
  • 将nums从小到大排序,假设有k个盒子
  • 先取最大的一个数,逐个盒子比较,放到当前盒子内数总和加上该数小于target的盒子中,将这个数从原集合中去掉
  • 递归进行前一步,直到原集合没有元素了为止
#include <vector>
#include <algorithm>
using namespace std;
class Solution {
public:
	bool canPartitionKSubsets(vector<int>& nums, int k) {
		int sum = 0;
		for (size_t i = 0; i < nums.size(); ++i)
			sum += nums[i];
		if (sum	% k != 0)
			return false;

		const int target = sum / k;

		sort(nums.begin(), nums.end());
		while (!nums.empty() && nums[nums.size() - 1] == target)
		{
			nums.pop_back();
			--k;
		}
		vector<int> group(k, 0);
		return partition(group, nums, target);
	}

	bool partition(vector<int> groups, vector<int> nums, int target)
	{
		if (nums.empty())
			return true;

		int v = nums[nums.size() - 1];
		nums.pop_back();

		for (size_t i = 0; i < groups.size(); ++i)
		{
			if (v + groups[i] <= target)
			{
				groups[i] += v;
				if (partition(groups, nums, target))
					return true;

				groups[i] -= v;
			}

			if (groups[i] == 0)
				break;
		}

		return false;
	}
};

 

C++中,递归是一种非常强大的工具,可以用来处理各种问题,包括求解数组的子集。以下是一个简单的例子,展示了如何使用递归来求解一个数组的所有非空子集,并将这些子集存储在二维数组中。 首先,我们可以定义一个递归函数,它接受数组、当前位置和用于存储子集的二维数组作为参数。然后,递归函数将根据当前位置选择包含当前元素或不包含当前元素两种情况,并继续对下一个位置进行处理,直到处理完所有位置。 这里提供一个简单的实现框架: ```cpp #include <iostream> #include <vector> void printSubsets(int index, int n, const std::vector<int>& arr, std::vector<int>& subset, std::vector<std::vector<int>>& result) { if (index == n) { // 当前位置已经处理完毕,打印或存储子集 result.push_back(subset); return; } // 不包含当前位置的元素 printSubsets(index + 1, n, arr, subset, result); // 包含当前位置的元素 subset.push_back(arr[index]); printSubsets(index + 1, n, arr, subset, result); subset.pop_back(); // 回溯 } int main() { std::vector<int> arr = {1, 2, 3}; // 示例数组 std::vector<int> subset; // 当前子集 std::vector<std::vector<int>> result; // 存储所有子集的二维数组 printSubsets(0, arr.size(), arr, subset, result); // 打印所有子集 for (const auto& s : result) { for (int num : s) { std::cout << num << " "; } std::cout << std::endl; } return 0; } ``` 在这个代码中,`printSubsets` 函数是递归的核心。它检查当前是否已经处理完数组中的所有元素,如果是,则将当前的子集添加到结果中。如果不是,则它首先尝试不添加当前元素的子集,然后尝试添加当前元素的子集,之后通过`pop_back`回溯,保证每次递归调用结束后子集中的元素与开始时相同。 这个程序会输出数组的全部子集,包括空集和数组本身。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值