洛谷p1464

宏定义函数习得！

其实宏定义函数就相当于简单的替换。

#define W_MEN(a,b,c)(w_men[a][b][c] ? w_men[a][b][c]:w_men[a][b][c]=w(a,b,c))

其实当return W_MEN(a,b,c)时就相当于return w_men[a][b][c] ? w_men[a][b][c]:w_men[a][b][c]=w(a,b,c)；

大大地减少了代码重用！

 

#include <iostream>

using namespace std;

#define W_MEN(a,b,c)(w_men[a][b][c] ? w_men[a][b][c]:w_men[a][b][c]=w(a,b,c))

int a, b, c;
int w_men[25][25][25] = { 0 };


int w(int a, int b, int c) {
	int ans;
	if (a <= 0 || b <= 0 || c <= 0) {
		return 1;
	}
	else if (a > 20 || b > 20 || c > 20) {
		
		return W_MEN(20, 20, 20);
	}
	else if (a < b && b < c) {
		return W_MEN(a, b, c - 1) + W_MEN(a, b - 1, c - 1) - W_MEN(a, b - 1, c);
	}
	else {
		return W_MEN(a - 1, b, c) + W_MEN(a - 1, b - 1, c) + W_MEN(a - 1, b, c - 1) - W_MEN(a - 1, b - 1, c - 1);
	}
}

int main() {
	
	while (true) {
		cin >> a >> b >> c;
		if (a == -1 && b == -1 && c == -1) {
			return 0;
		}
		cout <<"w("<<a<<", "<<b<<", "<<c<<") = "<< w(a, b, c)<<endl;

	}
	return 0;
}

完整代码如上，注意输出的格式即可。