递归形式的问题——算24

问题描述

• 算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的问题。
枚举先算的两个数，以及这两个数的运算方式。
边界条件：一个数算24

• 注意：浮点数比较是否相等，不能用 ==

• C语言代码示例：

#include<stdio.h>
#include<math.h>

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

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

int count24(double a[], int n){
  //用数组a里的n个数，计算24
  double b[5];
  int i, j, m, k;
  if(n == 1){
    if(isZero(a[0] - 24))
      return 1;
    else
      return 0;
  }
  for(i=0; i < n-1; ++i)
    for(j=i+1; j < n; ++j){
      //枚举两个数的组合
      m = 0;//还剩下m个数，m = n - 2
      for(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 1;
      b[m] = a[i] - a[j];
      if( count24(b, m+1) )
        return 1;
      b[m] = a[j] - a[i];
      if( count24(b, m+1) )
        return 1;
      b[m] = a[i] * a[j];
      if( count24(b, m+1) )
        return 1;
      if( !isZero(a[j]) ){
        b[m] = a[i]/a[j];
        if(count24(b, m+1))
          return 1;
      }
      if( !isZero(a[i]) ){
        b[m] = a[j]/a[i];
        if( count24(b, m+1) )
          return 1;
       }
   }
   return 0;
}

int main(){
  int i;
  while(1) {
    for(i=0; i < 4; ++i)
       scanf("%lf",&a[i]);
    if( isZero(a[0]) )
       break;
    if( count24(a, 4) )
       printf("YES\n");
    else
       printf("NO\n");
  }
  return 0;
}

• 运行示例
  
  注：文中问题及代码参考 MOOC——《程序设计与算法》（北京大学 郭炜）

• 算法系列目录：

  • 一、递归形式的问题
    • 汉诺塔问题（Hanoi)
    • N皇后问题
    • 逆波兰表达式
    • 四则运算表达式求值
    • 爬楼梯
    • 放苹果
    • 算24