题目:
你有 4 张写有 1 到 9 数字的牌。你需要判断是否能通过 *,/,+,-,(,) 的运算得到 24。
示例 1:
输入: [4, 1, 8, 7]
输出: True
解释: (8-4) * (7-1) = 24
示例 2:
输入: [1, 2, 1, 2]
输出: False
注意:
除法运算符 / 表示实数除法,而不是整数除法。例如 4 / (1 - 2/3) = 12 。
每个运算符对两个数进行运算。特别是我们不能用 - 作为一元运算符。例如,[1, 1, 1, 1] 作为输入时,表达式 -1 - 1 - 1 - 1 是不允许的。
你不能将数字连接在一起。例如,输入为 [1, 2, 1, 2] 时,不能写成 12 + 12 。
代码:
class Solution {
public:
static constexpr int TARGET = 24;
static constexpr double EPSILON = 1e-6;
static constexpr int ADD = 0, MULTIPLY = 1, SUBTRACT = 2, DIVIDE = 3;
bool judgePoint24(vector<int> &nums) {
vector<double> l;
for (const int &num : nums) {
l.emplace_back(static_cast<double>(num));
}
return solve(l);
}
bool solve(vector<double> &l) {
if (l.size() == 0) {
return false;
}
if (l.size() == 1) {
return fabs(l[0] - TARGET) < EPSILON;
}
int size = l.size();
for (int i = 0; i < size; i++) {
for (int j = 0; j < size; j++) {
if (i != j) {
vector<double> list2 = vector<double>();
for (int k = 0; k < size; k++) {
if (k != i && k != j) {
list2.emplace_back(l[k]);
}
}
for (int k = 0; k < 4; k++) {
if (k < 2 && i > j) {
continue;
}
if (k == ADD) {
list2.emplace_back(l[i] + l[j]);
} else if (k == MULTIPLY) {
list2.emplace_back(l[i] * l[j]);
} else if (k == SUBTRACT) {
list2.emplace_back(l[i] - l[j]);
} else if (k == DIVIDE) {
if (fabs(l[j]) < EPSILON) {
continue;
}
list2.emplace_back(l[i] / l[j]);
}
if (solve(list2)) {
return true;
}
list2.pop_back();
}
}
}
}
return false;
}
};
思路:
一共有4个数和3个运算操作,因此可能性非常有限。一共有多少种可能性呢?
首先从4个数字中有序地选出2个数字,共有4x3=12种选法,并选择加、减、乘、除4种运算操作之一,用得到的结果取代选出的2个数字,剩下3个数字;
然后在剩下的3个数字中有序地选出2个数字,共有3x2=6种选法,并选择4种运算操作之一,用得到的结果取代选出的2个数字,剩下2个数字;
最后剩下2个数字,有2种不同的顺序,并选择4种运算操作之一;
因此,总共有12*4*6*4*2*4=9216种不同的可能性;
可以通过回溯的方法遍历所有不同的可能性。具体做法是,使用一个列表存储目前的全部数字,每次从列表中选出2个数字,再选择一种运算操作,用计算得到的结果,取代选出的2个数字,这样列表中的数字就减少了1个。重复上述步骤,直到列表中只剩下1个数字,这个数字就是一种可能的结果,如果结果等于24,则说明可以通过运算得到24。如果所有的可能性的结果都不等于24,则说明无法通过运算得到24。
实现时,有一些细节需要注意。
1)除法运算为实数除法,因此结果为浮点数,列表中存储的数字也都是浮点数。在判断结果是否等于24时应该考虑精度误差,这道题中,误差小于10-6可以认为是相等;
2)进行除法运算时,除数不能为0,如果遇到除数为0的情况,则这种可能性可以直接排除。由于列表中存储的数字是浮点数,因此判断除数是否为0的时候,应该考虑精度误差,这道题中,当一个数字的绝对值小雨10-6的时候,可以认为该数字等于0;
还有一个可以优化的点
1)加法和乘法都满足交换率,因此如果选择的运算操作是加法或者乘法,则对于选出的2个数字不需要考虑不同的顺序,在遇到第二种顺序的时候可以不进行运算,直接跳过;