还记得童话《卖火柴的小女孩》吗?现在,你知道小女孩有多少根火柴,请找出一种能使用所有火柴拼成一个正方形的方法。不能折断火柴,可以把火柴连接起来,并且每根火柴都要用到。
输入为小女孩拥有火柴的数目,每根火柴用其长度表示。输出即为是否能用所有的火柴拼成正方形。
示例 1:
输入: [1,1,2,2,2]
输出: true
解释: 能拼成一个边长为2的正方形,每边两根火柴。
示例 2:
输入: [3,3,3,3,4]
输出: false
解释: 不能用所有火柴拼成一个正方形。
注意:
- 给定的火柴长度和在 0 到 10^9之间。
- 火柴数组的长度不超过15。
public boolean makesquare(int[] nums) {
if (nums == null || nums.length == 0)
return false;
if (nums.length < 4)
return false;
int sumLength = 0;
for (int i = 0; i < nums.length; i++) {
sumLength += nums[i];
}
if (sumLength % 4 != 0)
return false;
Arrays.sort(nums);
reverseNums(nums);
return backTrace(nums, 0, new int[4], sumLength / 4);
}
private void reverseNums(int[] nums) {
int i = 0, j = nums.length - 1;
while (i < j) {
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
i++;
j--;
}
}
private boolean backTrace(int[] nums, int index, int[] sums, int target) {
if (index == nums.length) {
if (sums[0] == target && sums[1] == target && sums[2] == target && sums[2] == target)
return true;
else
return false;
} else {
for (int i = 0; i < 4; i++) {
if (sums[i] + nums[index] > target) {
continue;
}
sums[i] += nums[index];
if (backTrace(nums, index + 1, sums, target)) {
return true;
}
sums[i] -= nums[index];
}
}
return false;
}