【Java】P1464 Function—方法：记忆化搜索—（OJ：洛谷）

链接：https://www.luogu.com.cn/problem/P1464
来源：洛谷

时间限制：1.00s
内存限制：125.00MB

题目描述

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

如果a≤0 or b≤0 or c≤0就返回值1.
如果a>20 or 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
*/

输入格式

会有若干行。

并以−1 −1 −1结束。

保证输入的数在[−9223372036854775808,9223372036854775807]之间，并且是整数。

输出格式

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

输入

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

输出

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

说明/提示

记忆化搜索

方法：记忆化搜索

记忆化搜索=搜索的形式+动态规划的思想

百科

一般说来，动态规划总要遍历所有的状态，而搜索可以排除一些无效状态。
更重要的是搜索还可以剪枝，可能剪去大量不必要的状态，因此在空间开销上往往比动态规划要低很多。
记忆化算法在求解的时候还是按着自顶向下的顺序，但是每求解一个状态，就将它的解保存下来，以后再次遇到这个状态的时候，就不必重新求解了。
这种方法综合了搜索和动态规划两方面的优点，因而还是很有实用价值的。

总结

建立一个数组，将每次求出的值保存下来，当下次使用的时候直接调用，减少重复的计算。

Java源代码

package 新手村;

import java.util.Scanner;

/**
 *@Title P1464Function.java
 *@description TODO
 *@time 2019年12月1日 上午11:32:18
 *@author Baisu
 *@version 1.0
 */
public class P1464Function {

	public static long [][][] dp = new long[50][50][50]; 
	
	public static long w (long a, long b, long 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(dp[(int) a][(int) b][(int) c]!=0) {	//检测dp数组，查找如果有这个值，就直接返回答案
			return dp[(int) a][(int) b][(int) c];
		}
		else if(a<b&&b<c) {
			dp[(int) a][(int) b][(int) c] = w(a,b,c-1)+w(a,b-1,c-1)-w(a,b-1,c);
			//计算dp[a][b][c]
		}
		else {
			dp[(int) a][(int) b][(int) 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);
			//计算dp[a][b][c]
		}
		return dp[(int) a][(int) b][(int) c];
		//再加一个return，返回dp[a][b][c]
	}
	
	public static void main(String[] args) {
		Scanner sc = new Scanner (System.in);
		while(true) {
			long a = sc.nextLong();
			long b = sc.nextLong();
			long c = sc.nextLong();
			if(a==-1&&b==-1&&c==-1) {
				break;
			}
			else {
				System.out.println("w("+a+", "+b+", "+c+") = "+w(a, b, c));			
			}
		}
	}

}