算24 递归 C++

算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
//因为计算时使用的是浮点数，所以需要用下面这个函数来判断两个浮点数是否相等，当两个数的差值的绝对值小于10`-6时，可以看作相等
bool isZero(double x) {
	return fabs(x) <= EPS;
}


//用数组a里的n个数，计算24
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; //还剩下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;
			//计算前需要先判断除数是否为0
			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() {
	double t;
	while (cin>>t)
	{
		for (int i = 1; i < 4; i++)
			cin >> a[i];
		if (t == 0 && a[1] == 0 && a[2] == 0&&a[3]==0)break;
		if (count24(a, 4))
			cout << "YES" << endl;
		else cout << "NO" << endl;
	}
	return 0;
}