dp入门课（五）

今天，将会是dp入门课的最后一节，以后我会出dp基础课和进阶课。

今天讲的内容，dfs里面的dp.

请先阅读dp入门课（一）

dp入门课（一）-CSDN博客

不会dfs的，可以自己上网上搜一搜。

这个算法跟我们讲过的传统dp很不一样，这个算法的出现，主要是基于dfs的。

先看一道题：P1434 [SHOI2002] 滑雪 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

这道题的解法很多：可以直接dp,也可以记忆化搜索（今天要讲的），也可以用图论（spfa求最长路）。

今天，我们就来讲一下记忆化搜索。

首先，这道题的暴力dfs代码很好写，如下：

#include <bits/stdc++.h>
using namespace std;
const int N = 5010;
int n, m;
int g[N][N];
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
int res = 0;
void dfs(int x, int y,int s) {
	res = max(res,s);
	for (int i = 0; i < 4; i ++ ) {
		int a = x + dx[i], b = y + dy[i];
		if (a >= 1 && a <= n && b >= 1 && b <= m && g[x][y] > g[a][b]) {
			dfs(a,b,s + 1);
		}
	}
}
int main() {
	cin >> n >> m;
	for (int i = 1; i <= n; i ++ ) {
		for (int j = 1; j <= m; j ++ ) {
			cin >> g[i][j];
		}
	}
	int t = 0;
	for (int i = 1; i <= n; i ++ ) {
		for (int j = 1; j <= m; j ++ ) {
			dfs(i,j,1);
			t = max(t,res);
			res = 0;
		}
	}
	cout << t;
	return 0;
}

这份代码获得了90分，其中一个点TLE了。

我们用一个f数组来记录一下，每次搜索到以前搜索过的点，咱们就直接返回f数组里的值，否则我们就继续搜索，并且把该点的值记录下来。

代码如下：

#include <bits/stdc++.h>
using namespace std;
const int N = 5010;
int n, m;
int g[N][N];
int f[N][N];
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};

int dp(int x, int y) {
	int &v = f[x][y];
	if (v != -1) {
		return v;
	}
	v = 1;
	for (int i = 0; i < 4; i ++ ) {
		int a = x + dx[i], b = y + dy[i];
		if (a >= 1 && a <= n && b >= 1 && b <= m && g[x][y] > g[a][b]) {
			v = max(v, dp(a, b) + 1);
		}
	}
	return v;
}
int main() {
	cin >> n >> m;
	for (int i = 1; i <= n; i ++ ) {
		for (int j = 1; j <= m; j ++ ) {
			cin >> g[i][j];
		}
	}
	memset(f, -1, sizeof f);
	int res = 0;
	for (int i = 1; i <= n; i ++ ) {
		for (int j = 1; j <= m; j ++ ) {
			res = max(res, dp(i, j));
		}
	}
	cout << res;
	return 0;
}

注意，f数组应该有一个初始值，好在每次判断该点是否被搜索过（当然，你在开一个标记数组也没问题）。