你將得到一個整數(shù)數(shù)組 matchsticks 煌珊,其中 matchsticks[i] 是第 i個火柴棒的長度芍躏。你要用 所有的火柴棍拼成一個正方形屯阀。你 不能折斷 任何一根火柴棒固灵,但你可以把它們連在一起捅伤,而且每根火柴棒必須 使用一次 。
如果你能使這個正方形巫玻,則返回 true 丛忆,否則返回 false 。
示例1:
輸入: matchsticks = [1,1,2,2,2]
輸出: true
解釋: 能拼成一個邊長為2的正方形仍秤,每邊兩根火柴熄诡。
示例2:
輸入: matchsticks = [3,3,3,3,4]
輸出: false
解釋: 不能用所有火柴拼成一個正方形。
提示:
- 1 <= matchsticks.length <= 15
- 1 <= matchsticks[i] <= 108
方法:回溯算法
class Solution {
public static boolean makesquare(int[] matchsticks) {
int sum = 0, len = matchsticks.length;
for (int matchstick : matchsticks) {
sum += matchstick;
}
if (sum % 4 != 0) {
return false;
}
// 排序
Arrays.sort(matchsticks);
return backtrace(matchsticks, sum >> 2, len - 1, new int[4]);
}
static boolean backtrace(int[] matchsticks, int target, int index, int[] size) {
if (index == -1) {
// 火柴用完了
return size[0] == size[1] && size[1] == size[2] && size[2] == size[3];
}
for (int i = 0; i < 4; i++) {
// size[i] == size[i - 1]即上一個分支的值和當前分支的一樣诗力,上一個分支沒有成功凰浮,
if (size[i] + matchsticks[index] > target || (i > 0 && size[i] == size[i - 1])) {
continue;
}
//如果當前火柴放到size[i]這個邊上,長度不大于target,我們就放上面
size[i] += matchsticks[index];
if (backtrace(matchsticks, target, index - 1, size)) {
return true;
}
size[i] -= matchsticks[index];
}
return false;
}
}
來源:力扣(LeetCode)
鏈接:https://leetcode.cn/problems/matchsticks-to-square
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有袜茧。商業(yè)轉(zhuǎn)載請聯(lián)系官方授權(quán)菜拓,非商業(yè)轉(zhuǎn)載請注明出處。