无聊做了的水题 题目描述 考虑如下的3参数递归函数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 = 15, b = 15, c = 15), 程序仍然需要几个小时才能完成大量的递归运算。 程序的输入为若干行整数,每行整数包含3个数值,分别对应a、b和c的值。程序输入以-1 -1 -1结束。设计一个程序,快速计算w(a, b, c)并给出结果。
分析 首先暴力直接按模型递归肯定下场就是GG 所以呢,我们可以仔细观察发现,这个递归固定的值只会有一个答案(唯一) 所以便有了一个想法:用三维数组储存,递归前判断是否为空即可
#include <iostream> #include <cstdio> #include <memory.h> using namespace std; int wa[21][21][21]; int w(int a, int b, int c) { if(a <= 0 || b <= 0 || c <= 0) { wa[0][0][0] = 1; return wa[0][0][0]; } if(a > 20 || b > 20 || c > 20) { if(wa[20][20][20] == -1) wa[20][20][20] = w(20, 20, 20); return wa[20][20][20]; } if(a < b && b < c) { if(wa[a][b][c] == -1) wa[a][b][c] = w(a, b, c - 1) + w(a, b - 1, c - 1) - w(a, b - 1, c); return wa[a][b][c]; } if(wa[a][b][c] == -1) wa[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 wa[a][b][c]; } int main() { int x, y, z; memset(wa, -1, sizeof(wa)); while(1) { scanf("%d%d%d", &x, &y, &z); if(x == -1 && y == -1 && z == -1) break; printf("w(%d, %d, %d) = %d\n", x, y, z, w(x, y, z)); } }