算法学习之递归--4个数组成24的问题

题目：给出4个小于10个正整数，你可以使用加减乘除4种运算以及括号把这4个数连接起来得到一个表达式。

现在的问题是，是否存在一种方式使得得到的表达式的结果等于24。 这里加减乘除以及括号的运算结果和运算的优先级跟我们平常的定义一致（这里的除法定义是实数除法）。 比如，对于5，5，5，1，我们知道5 * (5 – 1 / 5) = 24，因此可以得到24。又比如，对于1，1，4，2，我们怎么都不能得到24。

输入：
输入数据包括多行，每行给出一组测试数据，包括4个小于10个正整数。最后一组测试数据中包括4个0，表示输入的结束，这组数据不用处理。
输出：
对于每一组测试数据，输出一行，如果可以得到24，输出“YES”；否则，输出“NO”。
样例输入
5 5 5 1
1 1 4 2
0 0 0 0
样例输出
YES
NO

题目分析：

n个数算24，必有两个数要先算。这两个数算的结果，和剩余n-2个数，就构成了n-1个数求24的问题
枚举先算的两个数，以及这两个数的运算方式。
注意：浮点数比较是否相等，不能用 ==

详细代码如下，对照注释仔细体会：

#include<iostream>
#include<cmath>
using namespace std;
double a[5];
#define EPS 1e-6                    //10^(-6)，当浮点数小于这个数可认为是0(可以更小，如-8次方)
bool isZero(double x){              //判断浮点数是否为0
    return fabs(x) <= EPS;
}
bool count24(double a[], int n){    //数组a中n个数能否算出24
    if(n == 1){                     //数组中仅有一个元素
        if(isZero(a[0]-24))
            return true;
        else
            return false;
    }
    double b[5];
    for(int i=0;i<n-1;i++){         //枚举两个数的组合
        for(int j=i+1;j<n;j++){
            int m=0;
            for(int k=0;k<n;k++){   //将剩下的n-2个数存放到数组b中
                if(k!=i && k!=j)
                    b[m++]=a[k];
            }
            b[m]=a[i]+a[j];         //元素b[m]是a[i]和a[j]相加
            if(count24(b,m+1))
                return true;
            b[m]=a[i]-a[j];         //元素b[m]是a[i]和a[j]相减
            if(count24(b,m+1))
                return true;
            b[m]=a[j]-a[i];
            if(count24(b,m+1))
                return true;
            b[m]=a[i]*a[j];         //元素b[m]是a[i]和a[j]相乘
            if(count24(b,m+1))
                return true;
            if(!isZero(a[j])){      //元素b[m]是a[i]和a[j]相除
                b[m]=a[i]/a[j];
                if(count24(b,m+1))
                return true;
            }
            if(!isZero(a[i])){
                b[m]=a[j]/a[i];
                if(count24(b,m+1))
                return true;
            }
        }
    }
    return false;
}
int main(){
    while(true){
        for(int i=0;i<4;i++)        //输入4个元素
            cin>>a[i];
        if(isZero(a[0])&&isZero(a[1])&&isZero(a[2])&&isZero(a[3]))
            break;
        if(count24(a,4))            //可以算出24
            cout<<"YES"<<endl;
        else
            cout<<"NO"<<endl;
    }
    return 0;
}