【洛谷新手村】【记忆化搜索dfs】P1464 Function

题目链接

题目描述

对于一个递归函数w(a,b,c)

• 如果a≤0 or b≤0 or c≤0就返回值11.
• 如果a>20or b>20 or c>20就返回w(20,20,20)
• 如果a<b并且b<c就返回w(a,b,c-1)+w(a,b-1,c-1)-w(a,b-1,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均为15时，调用的次数将非常的多。你要想个办法才行.

/* absi2011 : 比如 w(30,−1,0)既满足条件1又满足条件2

这种时候我们就按最上面的条件来算

所以答案为1

*/

---

【题解】：

直接记忆化搜索即可，但是因为要考虑到有些是否有搜索过，所以要自己多搜一遍。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll dp[50][50][50];
ll w(ll a,ll b,ll c){
    if(a<=0||b<=0||c<=0){return 1;}
    if(a>20||b>20||c>20){return w(20,20,20);}
    if(a<b&&b<c){
        if(dp[a][b][c-1]==0) dp[a][b][c-1]=w(a,b,c-1);
        if(dp[a][b-1][c-1]==0) dp[a][b-1][c-1]=w(a,b-1,c-1);
        if(dp[a][b-1][c]==0) dp[a][b-1][c]=w(a,b-1,c);
        return dp[a][b][c]=dp[a][b][c-1]+dp[a][b-1][c-1]-dp[a][b-1][c];
    }else{
        if(dp[a-1][b][c]==0)dp[a-1][b][c]=w(a-1,b,c);
        if(dp[a-1][b-1][c]==0)dp[a-1][b-1][c]=w(a-1,b-1,c);
        if(dp[a-1][b][c-1]==0)dp[a-1][b][c-1]=w(a-1,b,c-1);
        if(dp[a-1][b-1][c-1]==0)dp[a-1][b-1][c-1]=w(a-1,b-1,c-1);
        return dp[a][b][c]
        =dp[a-1][b][c]
        +dp[a-1][b-1][c]
        +dp[a-1][b][c-1]
        -dp[a-1][b-1][c-1];
    }
}
int main()
{
    ll a,b,c;
    while(~scanf("%lld%lld%lld",&a,&b,&c)){
        if(a==-1&&b==-1&&c==-1){
            break;
        }
        ll ans=w(a,b,c);
        printf("w(%lld, %lld, %lld) = %lld\n",a,b,c,ans);
    }
    return 0;
}