洛谷 P1464 Function JAVA详解

题目描述

对于一个递归函数 w(a,b,c)
如果 a≤0 或 b≤0 或 c≤0 就返回值1。
如果 a>20 或 b>20 或 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 时，调用的次数将非常的多。你要想个办法才行。

注意：例如 w(30,−1,0) 又满足条件 1 又满足条件 2，请按照最上面的条件来算，答案为 1。

输入格式

会有若干行。并以 −1,−1,−1 结束。

输出格式

输出若干行，每一行格式：
w(a, b, c) = ans
注意空格。

输入输出样例

输入

1 1 1
2 2 2
-1 -1 -1

输出

w(1, 1, 1) = 2
w(2, 2, 2) = 4

说明/提示

数据规模与约定
保证输入的数在
[−9223372036854775808,9223372036854775807] 之间，并且是整数。
保证不包括 −1,−1,−1 的输入行数 T 满足 1≤T≤10⁵。

思路解答

1. 数据类型
   题目输入数已经很明确跟我们指出范围，直接开long
2. 解题思路

• 记忆化搜索！！
  记忆化搜索：按着自顶向下的顺序，但是每求解一个状态，就将它的解保存下来，以后再次遇到这个状态的时候，就不必重新求解了。
  由于题目递归会导致搜索太深而时间超而且很多是复用的，因此我们可以用数组边递归边储存前21个数的答案，在后面递归时便可直接用到储存的答案，简化它。
• 为什么是前21个数？
  因为当输入的a/b/c＞20时便直接递归w（20，20，20），因此无论多大的数都跟此条件答案一样，所以在输入时我们可判断到输入>20便直接赋值21，让他在函数中能进入a/b/c＞20这个条件而数据又不会太大即可。假设是设置前20个数，那么变量赋值20则无法进入该条件判断。

总结

当我们在题目中看到条件递归太深，递归到后面复用性比较高时便可以考虑 记忆化搜索来储存前面数据（即复用性高）的答案，来实现优化。

代码

import java.io.*;

public class Main{
    static long ans[][][]=new long[22][22][22];
    static StringBuffer answer=new StringBuffer();
    public static void main(String[] args) throws IOException {
        StreamTokenizer streamTokenizer = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));
        PrintWriter printWriter = new PrintWriter(new OutputStreamWriter(System.out));
        while (true){
            int ta=0,tb=0,tc=0;
            streamTokenizer.nextToken();
            long a=(long)streamTokenizer.nval;
            streamTokenizer.nextToken();
            long b=(long)streamTokenizer.nval;
            streamTokenizer.nextToken();
            long c=(long)streamTokenizer.nval;if (a==-1&&b==-1&&c==-1) break;
            if (a>20) ta = 21;
            else ta=(int)a;
            if (b>20) tb=21;
            else tb=(int)b;
            if (c>20) tc=21;
            else tc=(int)c;
            answer.append("w("+String.valueOf(a)+", "+String.valueOf(b)+", "+String.valueOf(c)+") = "+String.valueOf(w(ta,tb,tc))+"\n");

        }
        printWriter.println(answer);
        printWriter.flush();
    }
    public static long w(int a,int b,int c){
        if (a<=0||b<=0||c<=0) return 1;
        else if (ans[a][b][c]!=0) return ans[a][b][c];
        else if (a>20||b>20||c>20) ans[a][b][c]=w(20,20,20);
        else if (a<b&&b<c) ans[a][b][c]=w(a,b,c-1)+w(a,b-1,c-1)-w(a,b-1,c);
        else ans[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 ans[a][b][c];
    }
}