递归——算24

题目：给出4个小于10的正整数，你可以使用加减乘除4种运算以及括号把这4个数连接起来得到一个表达式。现在的问题是，是否存在一种方式使得得到的表达式的结果等于24.

输入：输入数据包括多行，每行给出一组测试数据，包括4个小于10的正整数。最后一组测试数据中包括4个0，表示输入的结束，这组测试不用处理。

输出：对于每一组测试数据，输出一行，如果可以得到24，输出“YES”；否则，输出“NO”。

样例输入

5  5  5  1

1  1  4  2

0  0  0  0

样例输出

YES

NO

 

代码：

#include "stdafx.h"
#include <iostream>
#include <algorithm>
#include <cmath>

using namespace std;

double a[5];
#define EPS 1e-6;

bool isZero(double x)
{
    return fabs(x)<=EPS;
}

bool count24(double a[],int n)
{
    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)
            {
                if(k!=i&&k!=j)
                    b[m++] = a[k];
            }

            b[m] =a[i]+a[j];
            if(count24(b,m+1))
                return true;

            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];
            if(count24(b,m+1))
                return true;

            if(!isZero(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 _tmain(int argc, _TCHAR* argv[])
{
    double a[4] = {5,5,5,1};
    bool result = count24(a,sizeof(a)/sizeof(a[0]));
    cout<<(result?"YES":"NO")<<endl;

    return 0;
}