679. 24 点游戏 c++解法 回溯

题目描述

链接：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 。

解法

class Solution
{
public:
    bool judgePoint24(vector<int> &cards)
    {
        vector<double> curNums;
        for (int i = 0; i < cards.size(); ++i)
        {
            curNums.push_back((double)cards[i]);
        }
        return judgePoint24Core(curNums);
    }

private:
    const double ex = 1e-6;
    bool judgePoint24Core(vector<double> &cards)
    {
        int n = cards.size();
        if (n == 0)
        {
            return false;
        }
        if (n == 1)
        {
            return abs(cards[0] - 24) < ex;
        }
        for (int i = 0; i < n; ++i)
        {
            for (int j = 0; j < n; ++j)
            {
                if (i != j)
                {
                    vector<double> curNums;
                    for (int k = 0; k < n; ++k)
                    {
                        if (k == i || k == j)
                        {
                            continue;
                        }
                        curNums.push_back(cards[k]);
                    }
                    //a+b
                    curNums.push_back(cards[i] + cards[j]);
                    if (judgePoint24Core(curNums))
                    {
                        return true;
                    }
                    curNums.pop_back();
                    //a-b
                    curNums.push_back(cards[i] - cards[j]);
                    if (judgePoint24Core(curNums))
                    {
                        return true;
                    }
                    curNums.pop_back();
                    //a*b
                    curNums.push_back(cards[i] * cards[j]);
                    if (judgePoint24Core(curNums))
                    {
                        return true;
                    }
                    curNums.pop_back();
                    //a/b
                    if (cards[j] != 0)
                    {
                        curNums.push_back(cards[i] / cards[j]);
                        if (judgePoint24Core(curNums))
                        {
                            return true;
                        }
                        curNums.pop_back();
                    }
                }
            }
        }

        return false;
    }
};