还记得童话《卖火柴的小女孩》吗?现在,你知道小女孩有多少根火柴,请找出一种能使用所有火柴拼成一个正方形的方法。不能折断火柴,可以把火柴连接起来,并且每根火柴都要用到。
输入为小女孩拥有火柴的数目,每根火柴用其长度表示。输出即为是否能用所有的火柴拼成正方形。
示例 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.length < 4)
return false;
int sum = 0;
for (int i = 0; i < nums.length; i++) {
sum += nums[i];
}
if (sum % 4 != 0)
return false;
int[] arr = new int[4];
return doIt(arr, nums, 0, sum / 4);
}
private boolean doIt(int[] arr, int[] nums, int pos, int len) {
if (pos >= nums.length) {
return arr[0] == len && arr[1] == len && arr[2] == len && arr[3] == len;
}
for (int i = 0; i < 4; i++) {
if (arr[i] + nums[pos] > len)
continue;
arr[i] += nums[pos];
if (doIt(arr, nums, pos + 1, len))
return true;
arr[i] -= nums[pos];
}
return false;
}