ACM PKU 1579 Function Run Fun 记忆化搜索

ACM PKU 1579 Function Run Fun 记忆化搜索

Source

Problem: 1579  User: lnmm
Memory: 108K  Time: 0MS
Language: C++  Result: Accepted

#include " stdio.h "
int  s[ 21 ][ 21 ][ 21 ];

int  w( int  a, int  b, int  c)
{
    if(a<=0||b<=0||c<=0) return 1;
    else if(a>20||b>20||c>20) return w(20,20,20);
    else if(s[a][b][c]>0) return s[a][b][c];
    else if(a<b&&b<c) s[a][b][c]=w(a,b,c-1)+w(a,b-1,c-1)-w(a,b-1,c);
    else 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);
    
    return s[a][b][c];
}

void  main()
{
    int a,b,c;
    int k;
    int o,p,q;
    while(scanf("%d%d%d",&a,&b,&c)&&(a!=-1||b!=-1||c!=-1))
    {
        for(o=0;o<21;o++)
            for(p=0;p<21;p++)
                for(q=0;q<21;q++)
                    s[o][p][q]=-1;


        k=w(a,b,c);
        printf("w(%d, %d, %d) = %d\n",a,b,c,k);
    }
return;
}

这道题直接递归显然是TLE的.
记忆化搜索就是好啊,呵呵,有固定模式了,解题就是快.
不过要切忌粗心.