还记得童话《卖火柴的小女孩》吗?现在,你知道小女孩有多少根火柴,请找出一种能使用所有火柴拼成一个正方形的方法。不能折断火柴,可以把火柴连接起来,并且每根火柴都要用到。
输入为小女孩拥有火柴的数目,每根火柴用其长度表示。输出即为是否能用所有的火柴拼成正方形。
示例 1:
输入: [1,1,2,2,2]
输出: true
解释: 能拼成一个边长为2的正方形,每边两根火柴。
示例 2:
输入: [3,3,3,3,4]
输出: false
解释: 不能用所有火柴拼成一个正方形。
class Solution {
public boolean makesquare(int[] nums) {
if (nums.length == 0) {
return false;
}
int sum = 0;
for (int num : nums) {
sum += num;
}
if (sum % 4 != 0) {
return false;
}
int target = sum / 4;
return ResultForDPS(nums, 0, target, target, target, target);
}
private boolean ResultForDPS(int[] nums, int index, int one, int two, int three, int four) {
if(index==nums.length){
return one==two&&two==three&&three==four;
}
if(one<0||two<0||three<0||four<0){
return false;
}
return ResultForDPS(nums,index+1,one-nums[index],two,three,four)||
ResultForDPS(nums,index+1,one,two-nums[index],three,four)||
ResultForDPS(nums,index+1,one,two,three-nums[index],four)||
ResultForDPS(nums,index+1,one,two,three,four-nums[index]);
}
}
网友评论