LeetCode473. 火柴拼正方形
题目描述:
你将得到一个整数数组 matchsticks ,其中 matchsticks[i] 是第 i 个火柴棒的长度。你要用 所有的火柴棍 拼成一个正方形。你 不能折断 任何一根火柴棒,但你可以把它们连在一起,而且每根火柴棒必须 使用一次 。
如果你能使这个正方形,则返回 true ,否则返回 false 。
题目分析:此题就是使用数组中的所有元素,且每一个元素值可以使用一次,判断所有的元素是否可以构成一个正方形。
解题思路:
以数组nums,[1,2,2,2,1]为例。
我们可以建立一个长度为4数组桶res用来存放nums中的元素,若元素可以去全部放入则返回true,反之返回false;
- 对数组进行排序,计算数组所有元素的和即正方形的周长sum,然后用周长对4取余若不为0,直接返回false;
- 根据正方形的周长sum计算出其边长b;
- 在遍历数组,将数组中的元素依次放入桶中,若遇到当前存放进去的元素小于边长,那么我们就递归判断在nums中的下一个元素是否可以放入桶中,若可以则返回true。
- 若没有返回则尝试回溯到之前的状态在去循环判断其他的元素,直到循环结束,返回false。
代码:
public static boolean makesquare(int[] matchsticks) {
int sum = 0;
for(int i = 0;i < matchsticks.length;i++){
sum += matchsticks[i];
}
if(sum % 4 != 0 || matchsticks.length < 4){
return false;
}
//正方形边长
int b = sum / 4;
//存放数据的桶
int[] arr = new int[4];
return backTrup(0,matchsticks,b,arr);
}
private static boolean backTrup(int r, int[] matchsticks, int b, int[] arr) {
//递归结束条件
if (r >= matchsticks.length){
return true;
}
//按从大到小排序
Arrays.sort(matchsticks);
for (int i = 0,j = matchsticks.length -1;i < j;i++,j--){
int temp = 0;
temp = matchsticks[i];
matchsticks[i] = matchsticks[j];
matchsticks[j] = temp;
}
//循环4个桶
for (int i = 0; i < 4; i++) {
//存入数据
arr[i] += matchsticks[r];
//当前的数据小于边长且下一个元素可以放入桶中
if (arr[i] <= b && backTrup(r + 1, matchsticks, b, arr)){
return true;
}
//回溯
arr[i] -= matchsticks[r];
}
return false;
}