2021-08-26

记忆化搜索——记忆宏

题目描述 来自洛谷P1464

输入格式
会有若干行。

并以-1,-1,-1−1,−1,−1结束。

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

输出格式
输出若干行，每一行格式：

w(a, b, c) = ans

注意空格。

输入输出样例
输入 #1复制

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

输出 #1复制

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

说明/提示
记忆化搜索

代码如下：

#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#include<cmath>
#include<iomanip>
#include<cfloat>
#include<climits>
using namespace std;
#define dp(x,y,z) (dp[x][y][z] ? dp[x][y][z] : dp[x][y][z] = w(x, y, z))
int a=1,b=1,c=1;
int dp[25][25][25];
int w(int a, int b, int c){
    if(a<=0 || b<=0 || c<=0) return 1;
    if(a > 20 || b > 20 || c > 20) return dp(20,20,20);
    if(a < b && b < c) return dp(a,b,c-1)+dp(a,b-1,c-1) - dp(a,b-1,c);
    return 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(){
	while(true){
		cin>>a>>b>>c;
		if(a==-1 && b==-1 && c==-1) break;
	    cout<<"w("<<a<<", "<<b<<", "<<c<<") = "<<w(a,b,c)<<endl;
	}
}

源代码作者博客：

https://www.luogu.com.cn/blog/vel/solution-p1464