leetcode回溯698

/**
<p>给定一个整数数组&nbsp;&nbsp;<code>nums</code> 和一个正整数 <code>k</code>，找出是否有可能把这个数组分成 <code>k</code> 个非空子集，其总和都相等。</p>

<p>&nbsp;</p>

<p><strong>示例 1：</strong></p>

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

<p><strong>示例 2:</strong></p>

<pre>
<strong>输入:</strong> nums = [1,2,3,4], k = 3
<strong>输出:</strong> false</pre>

<p>&nbsp;</p>

<p><strong>提示：</strong></p>

<ul>
	<li><code>1 &lt;= k &lt;= len(nums) &lt;= 16</code></li>
	<li><code>0 &lt; nums[i] &lt; 10000</code></li>
	<li>每个元素的频率在 <code>[1,4]</code> 范围内</li>
</ul>
<div><div>Related Topics</div><div><li>位运算</li><li>记忆化搜索</li><li>数组</li><li>动态规划</li><li>回溯</li><li>状态压缩</li></div></div><br><div><li> 599</li><li> 0</li></div>
*/

//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
	public static boolean canPartitionKSubsets(int[] nums, int k) {
		//平分的桶子大于数组个数
		if (k > nums.length) {
			return false;
		}

		//所有数字和无法平分
		int sum = 0;
		for (int i = 0; i < nums.length; i++) {
			sum += nums[i];
		}
		if (sum % k != 0) {
			return false;
		}

		int[] bucket = new int[k];
		int target = sum / k;
		return backtrack(nums, 0, bucket, target);
	}

	//以数字的角度
	static boolean backtrack(int[] nums, int index, int[] bucket, int target) {
		if (index == nums.length) {
			for (int i = 0; i < bucket.length; i++) {
				if (bucket[i] != target) {
					return false;
				}
				return true;
			}
		}
		for (int i = 0; i < bucket.length; i++) {
			//剪枝 桶子满了
			if (bucket[i] + nums[index] > target) {
				continue;
			}

			bucket[i] += nums[index];
			if (backtrack(nums, index + 1, bucket, target)) {
				return true;
			}
			bucket[i] -= nums[index];
		}
		return false;
	}
}
//leetcode submit region end(Prohibit modification and deletion)