你有 4 张写有 1 到 9 数字的牌。你需要判断是否能通过 *
,/
,+
,-
,(
,)
的运算得到 24。
示例 1:
输入: [4, 1, 8, 7] 输出: True 解释: (8-4) * (7-1) = 24
参考:
先4个选2个数即12种做4次运算,再三个、两个。
共12×4×6×4×2×4=9216 种不同的可能性。
回溯法解题
注意:
由于除法是实数运算所以需要用浮点数,结果小于10^-6即算相等
同理,除数为0的时候也不可以,所以除数小于10^-6即算相等
优化:加法乘法满足结合率,所以可以不用重复计算
注意:每次遍历一个算数符号你就应该回溯一次
class Solution {
static final int TARGET = 24;
static final double EPSILON = 1e-6;
static final int ADD = 0, MULTIPLY = 1, SUBTRACT = 2, DIVIDE = 3;
public boolean judgePoint24(int[] nums) {
// 把nums放入list
List<Double> list = new ArrayList<Double>();
for (int num : nums) {
list.add((double) num);
}
return solve(list);
}
public boolean solve(List<Double> list) {
// 如果list的大小为0 说明数据错误
if (list.size() == 0) {
return false;
}
// list的大小为1 说明list的数据处理完毕,返回结果
if (list.size() == 1) {
// 此处注意结果差距小于10^-6即可
return Math.abs(list.get(0) - TARGET) < EPSILON;
}
int size = list.size();
// 选择两个数,list数据里面选择两个数
for (int i = 0; i < size; i++) {
for (int j = 0; j < size; j++) {
// 如果选择的两个数不是同一个继续
if (i != j) {
List<Double> list2 = new ArrayList<Double>();
// 把除了选择的两个数的之外的数放入list
for (int k = 0; k < size; k++) {
if (k != i && k != j) {
list2.add(list.get(k));
}
}
// 两数加减乘除操作
// 加入操作完的数
for (int k = 0; k < 4; k++) {
// 加和乘两个操作则两数只进行一次
if (k < 2 && i > j) {
continue;
}
// 加减乘除对应操作
if (k == ADD) {
list2.add(list.get(i) + list.get(j));
} else if (k == MULTIPLY) {
list2.add(list.get(i) * list.get(j));
} else if (k == SUBTRACT) {
list2.add(list.get(i) - list.get(j));
} else if (k == DIVIDE) {
// 除法
// 除数接近0则不进行该操作
if (Math.abs(list.get(j)) < EPSILON) {
continue;
} else {
list2.add(list.get(i) / list.get(j));
}
}
// 把处理完的另一list继续递归,递归找到则说明有
if (solve(list2)) {
return true;
}
// 回溯
list2.remove(list2.size() - 1);
}
}
}
}
// 递归完还没找到说明不对
return false;
}
}