POJ 1579 记忆递归

题意： if a <= 0 or b <= 0 or c <= 0, then w(a, b, c) returns:  1 
if a > 20 or b > 20 or c > 20, then w(a, b, c) returns:  w(20, 20, 20) 
if a < b and b < c, then w(a, b, c) returns:  w(a, b, c-1) + w(a, b-1, c-1) - w(a, b-1, c) 

otherwise it returns: 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都大于15后，运行时间会非常长，所以递归有所改变。

思路：刚开始我也没做过这种题目，看了discuss才明白要用记忆递归的方式，就是用一个三维的数组来记忆递归的结果，这样就不会使时间那么长了，而且很省时……

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int s[50][50][50];
int w(int a,int b,int c)
{
    if(a<=0||b<=0||c<=0) return 1;
    if(s[a][b][c]) return s[a][b][c]; //这句刚开始我想省了，但是运行时间还是很长，这是递归的过程中要提前结束递归以节约时间的，所以这句不能省！！！
    else if(a>20||b>20||c>20) return s[a][b][c]=w(20,20,20);
    else if(a<b&&b<c) return s[a][b][c]=w(a,b,c-1)+w(a,b-1,c-1)-w(a,b-1,c);
    else return s[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);
}
int main()
{
    int a,b,c;
    while(cin>>a>>b>>c)
    {
        if(a==-1&&b==-1&&c==-1) break;
        memset(s,0,sizeof(s));
        printf("w(%d, %d, %d) = %d\n",a,b,c,w(a,b,c));
    }
    return 0;
}