划分为k个相等的子集

问题描述 :
给定一个整数数组 nums 和一个正整数 k，找出是否有可能把这个数组分成 k 个非空子集，其总和都相等。

示例 1：
输入： nums = [4, 3, 2, 3, 5, 2, 1], k = 4
输出： True
说明： 有可能将其分成 4 个子集（5），（1,4），（2,3），（2,3）等于总和。

输入说明 :
首先输入nums数组的长度n，
然后输入n个整数，以空格分隔。
最后输入k。
1 <= k <= n <= 16
0 < nums[i] < 10000

输出说明 :
输出true或false

输入范例 :
7
4 3 2 3 5 2 1
4

输出范例 :
true

#include<iostream>
#include<vector>
#include<numeric>
#include<algorithm>
using namespace std;
bool backtrace(const vector<int> &nums,vector<int> &sums,int i,int k,int average){
	if(i==nums.size())  return true;
	for(int j=0;j<k;j++){
		if(sums[j]<average&&sums[j]+nums[i]<=average){
			sums[j]=sums[j]+nums[i];
			if(backtrace(nums,sums,i+1,k,average)){
				return true;
			}
			sums[j]=sums[j]-nums[i];//回溯 
		}
	}
	return false;
}
bool canPartitionKSubsets(vector<int>& nums, int k){
	int s=accumulate(nums.begin(),nums.end(),0);
	if(s%k!=0)  return false;
	sort(nums.begin(),nums.end(),greater<int> ());
	vector<int> sums(k,0);
	return backtrace(nums,sums,0,k,s/k);
}


int main(){
	vector<int> nums;
	int n,tmp,k;
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>tmp;
		nums.push_back(tmp);
	}
	cin>>k;
	bool res=canPartitionKSubsets(nums,k);
	cout<<(res?"true":"false")<<endl;
	return 0; 
}