华东师范考研机试：慢速的递归

因为要准备复试机试就每天刷两道题

D - 慢速的递归 POJ - 1579

大家应该对递归不陌生，有很多问题使用递归解决会又方便又简洁。

我们现在定义这样一个递归函数w(a, b, c):

如果 a <= 0 或 b <= 0 或 c <= 0, 那么 w(a, b, c) 的值为1

如果 a > 20 或 b > 20 或 c > 20, 则 w(a, b, c) 的值为:
w(20, 20, 20)

如果 a < b 并且 b < c, 此时 w(a, b, c) 的值等于:
w(a, b, c-1) + w(a, b-1, c-1) - w(a, b-1, c)

其它情况下w(a, b, c)的值为:
w(a-1, b, c) + w(a-1, b-1, c) + w(a-1, b, c-1) - w(a-1, b-1, c-1)

形式上这个函数不难实现，但是如果我们直接这样定义递归函数，对于并不那么大的a,b,c(比如a=b=c=15)来说，程序都有可能要运行数个小时，因为这个递归收敛太慢，运算所需要的计算量太大了！现在希望你来做点什么，快速计算出指定变量a,b,c的函数值.

Input

输入有多行，每行是由空格分隔的三个整数a,b,c，最后一行是 -1,-1,-1表示输入的结束，这一行不需要处理。
Output
除最后一行之外，对输入数据的每一行的a,b,c分别计算并按样例中的格式输出w(a,b,c)的值
Sample Input

1 1 1
2 2 2
10 4 6
50 50 50
-1 7 18
-1 -1 -1

Sample Output

w(1, 1, 1) = 2
w(2, 2, 2) = 4
w(10, 4, 6) = 523
w(50, 50, 50) = 1048576
w(-1, 7, 18) = 1

我的代码

#include<stdio.h>
int main()
{
    int w[21][21][21],a,b,c;
    for(int i=0;i<21;i++)
    {
        for(int j=0;j<21;j++)
        {
            for(int k=0;k<21;k++)
            {
                if(i==0||j==0||k==0)
                    w[i][j][k]=1;
                else if(i<j&&j<k)
                    w[i][j][k]=w[i][j][k-1]+w[i][j-1][k-1]-w[i][j-1][k];
                else
                    w[i][j][k]=w[i-1][j][k]+w[i-1][j-1][k]+w[i-1][j][k-1]-w[i-1][j-1][k-1];
            }
        }
    }
    while(scanf("%d%d%d",&a,&b,&c)&&(a!=-1||b!=-1||c!=-1))
    {
        printf("w(%d, %d, %d) = ",a,b,c);
        if(a<=0||b<=0||c<=0)
            printf("1\n");
        else if(a>20|b>20||c>20)
            printf("%d\n",w[20][20][20]);
        else
            printf("%d\n",w[a][b][c]);
    }
    return 0;
}

参考答案

#include <cstdio>
#include <cstring>

#define MAXN 21
int bw[MAXN][MAXN][MAXN];

int w(int a, int b, int c)
{
    
    if(a<=0 || b<=0 || c<=0)
    {
        return 1;
    }
    if(a>20 || b>20 || c>20)
    {
        return w(20,20,20);
    }
    
    if(bw[a][b][c]!=0)
    {
        return bw[a][b][c];
    }
    else
    {
        if(a<b && b<c)
        {
            bw[a][b][c]=w(a, b, c-1) + w(a, b-1, c-1) - w(a, b-1, c);
        }
        else
        {
            bw[a][b][c]=w(a-1, b, c) + w(a-1, b-1, c) + w(a-1, b, c-1) - w(a-1, b-1, c-1);
        }
        return bw[a][b][c];
    }
}

int main(void)
{
    memset(bw,0,sizeof(int)*MAXN*MAXN*MAXN);
    int a,b,c;
    while (1)
    {
        scanf("%d%d%d",&a,&b,&c);
        if(a==-1 && b==-1 && c==-1)
        {
            break;
        }
        printf("w(%d, %d, %d) = %d\n",a,b,c,w(a,b,c));
    }
    return 0;
}

反思

类似于动态规划，我们需要存储已经计算过的w的值，按照题目给出的顺序计算就没有问题。