LeetCode 679. 24 点游戏 dfs

地址 https://leetcode-cn.com/problems/24-game/

你有 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 。

算法1
首先考虑的是DFS，抽取任意两个数字进行任意计算，
然后将结果放入到抽取池中，进行下一次抽取.
这样可以遍历到所有的可能计算方式，不必考虑括号。
另一点需要注意的就是分数方案，需要使用double。

C++ 代码

class Solution {
public:
   vector<char> ops = { '+','-','*','/' };


bool judge(const vector<double>& nums, double target)
{
    if (nums.size() == 1) {
        return abs(nums[0] - (double)target)< 1e-5;
    }


    for (int i = 0; i < nums.size(); i++) {
        for (int j = i + 1; j < nums.size(); j++) {
            vector<double> copy = nums;
            double a = nums[i]; double b = nums[j];
            copy.erase(copy.begin() + i);
            copy.erase(copy.begin() + j-1);

            copy.push_back(a + b);
            if (judge(copy, 24)) 
                return true;
            copy.pop_back();

            copy.push_back(a * b);
            if (judge(copy, 24)) 
                return true;
            copy.pop_back();

            copy.push_back(a - b);
            if (judge(copy, 24)) 
                return true;
            copy.pop_back();

            copy.push_back(b - a);
            if (judge(copy, 24)) 
                return true;
            copy.pop_back();

            if (a != 0) {
                copy.push_back(b / a);
                if (judge(copy, 24)) 
                    return true;
                copy.pop_back();
            }

            if (b != 0) {
                copy.push_back(a / b);
                if (judge(copy, 24)) 
                    return true;
                copy.pop_back();
            }
        }
    }

    return false;
}


bool judgePoint24(vector<int>& nums) {
    vector<double> v(nums.begin(),nums.end());
    return judge(v, 24);
}
};