程序设计与算法（二）算法基础-郭炜 3.4.1 算24

例题:算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 <iostream>
#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)
{
	//用数组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;	//还剩下m个数，m = n - 2
			for(int k = 0;k < n;++k)
			{
				if(k != i && k != j)
				{
					b[m++] = a[k];	//把其余数放入b 
				}
			}
			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 main()
{
	for(int i=0;i<4;i++)
	{
		cin>>a[i];
	}
	if(count24(a,4))
		cout<<"YES"<<endl;
	else 
		cout<<"NO"<<endl;
	return 0;
}