24点问题(增加数据)·回溯

题目信息

一副牌，除了大小王之外还有52张，从1到13每个数目各有四张。要求设计一个程序，对于任意给出52张牌中的四张，运用＋－×÷四种运算来判断能否每个数只能用一次，但不能不用，算出24来。

测试样例

1 1 1 1
1 2 3 4

no
yes

解答

#include <cstdio>
#include <cmath>
#include <iostream>

using namespace std;

#define EPS 0.0000001

bool flag[5];   //标记该位是否被计算过
double num[5];   //每一位的值

bool Calc24(int step)
{
    if (step == 4)
    {
        for (int i = 0; i < 4; i++)
        {//结果可能在1、2、3、4位中（取决于运算的顺序）
            if (!flag[i] && fabs(num[i] - 24) <= EPS)
            {//剩余位上（结果）的值几乎为24
                return true;
            }
        }
        return false;
    }

    for (int i = 1; i <= 4; i++)
    {//总共循环4次
        if (!flag[i])
        {//如果此位置没有访问过
            for (int j = i + 1; j <= 4; j++)
            {//找第二个数
                if (!flag[j])
                {//同时保证这个数没有被访问过
                    flag[j] = true;//由于计算结果储存在第i位可以继续参与运算，而第j位不再继续
                    double temp1 = num[i], temp2 = num[j];//分别取出待运算的两位

                    //分别搜索：两种数的 6 种运算
                    num[i] = temp1 + temp2;
                    if (Calc24(step + 1))
                    {
                        return true;
                    }

                    //两种减法
                    num[i] = temp1 - temp2;
                    if (Calc24(step + 1))
                    {
                        return true;
                    }

                    num[i] = temp2 - temp1;
                    if (Calc24(step + 1))
                    {
                        return true;
                    }

                    num[i] = temp1 * temp2;
                    if (Calc24(step + 1))
                    {
                        return true;
                    }

                    if (fabs(temp1 - 0) <= EPS)
                    {   //排除分母为零的情况
                        num[i] = temp2 / temp1;
                        if (Calc24(step + 1))
                        {
                            return true;
                        }
                    }

                    if (fabs(temp2 - 0) <= EPS)
                    {   //排除分母为零的情况
                        num[i] = temp1 / temp2;
                        if (Calc24(step + 1))
                        {
                            return true;
                        }
                    }

                    //还原，便于回溯
                    num[i] = temp1;
                    flag[j] = false;
                }
            }
        }
    }
    return false;//一直没有结果
}

int main()
{
    freopen("E://test.txt", "r", stdin);
    while (EOF != scanf("%lf %lf %lf %lf", &num[1], &num[2], &num[3], &num[4]))
    {
        for (int i = 1; i < 5; i++)
        {//一定注意初始化
            flag[i] = false;
        }

        if (Calc24(1))
        {
            cout << "yes" << endl;
        }
        else
        {
            printf("no\n");
        }
    }
    return 0;
}

想法

首先大致列举 4 位数字组合运算的所有可能情况，大致可以将导致结果不同的因素有：

• 四位数字的排列顺序：如 2、3 、4、1 与 2、4、3、1
• 四位数字的运算顺序（括号如何加）：如（2 + 3) - (4 + 1) 与 (2 + 3) - 4 + 1
• 四位数字之间的运算（3个运算符）：如 （2 + 3) - (4 + 1) 与 （2 + 3) - (4 - 1)

先看一个小细节，通过观察发现，其实第一个因素（不同的排列顺序）不同（2、3因素相同时）或许不会有影响，如：（2 + 3) - (4 + 1) 与 （3 + 2) - (1 + 4)。其实就是：加、乘与前后顺序无关，减、除与前后顺序有关。
分别考虑三种因素会使得回溯变得很复杂，而且上面也说明了1、2因素单独列举时会产生重复部分，故这里考虑将1、2因素合并考虑，第3个因素单独考虑。那么，算法的大概思路如下：

• 从四个数中选取两个数进行运算，运算结果为x，此时还剩下三个数待运算（x与未参与运算的两个数）
• 从三个数中选取两个数进行运算，运算结果为x，此时还剩下两个数待运算（x与未参与运算的一个数）
• 选取剩下的两个数进行运算，得到最终的结果y
• 若y=24，即这样子选取的结果是可以的，输出yes。否则重新选取（若全部都选取完都没有找到结果为24，就要输出no）