信息学奥赛一本通 1416：【17NOIP普及组】棋盘

题目

1416：【17NOIP普及组】棋盘

代码

注意，朴素深搜过不了，要记忆化搜索。

而且需要剪枝：若到达一个点时，花费已经超过当前最优解，就不用继续搜索

解法1：记忆化搜索

#include<iostream>
#include<cstring>
using namespace std;

const int N = 114;
int a[N][N];  // 棋盘，0表示无色，1表示红色，2表示黄色
int f[N][N];  // 记忆化数组
int n, m;
int dir[4][2] = { -1,0,0,1,1,0,0,-1 };  // 方向数组
int ans;  // 答案

// 当前走到(x,y)，已经花money元，magic表示能否用魔法
void dfs(int x, int y, int money, bool magic) {
	// 剪枝
	// 当前花费已经比之前到达该点时的花费还大，不可能是最优
	if (f[x][y] <= money) return;

	// 记录一下
	f[x][y] = money;

	// 到达终点
	if (x == m && y == m) ans = min(ans, money);
	else {
		// 取出当前点颜色
		int color = a[x][y];

		// 扩展
		for (int i = 0; i < 4; i++) {
			int nx = x + dir[i][0], ny = y + dir[i][1];
			// 没有越界
			if (nx > 0 && nx <= m && ny > 0 && ny <= m) {
				int color2 = a[nx][ny];  // 取出这个点的颜色

				if (color2 == 0) {  // 没有颜色
					if (magic) {  // 且可以施展魔法
						a[nx][ny] = color;  // 施展魔法
						dfs(nx, ny, money + 2, false);  // 花2元，下一步不能再用魔法了
						a[nx][ny] = 0;      // 魔法失效
					}
				}
				else if (color2 == color) dfs(nx, ny, money, true);  // 颜色一样
				// 颜色不一样
				else if (color != color2) dfs(nx, ny, money + 1, true);  // 花1元
			}
			
		}
	}
}


int main() {
	// 读写优化
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);

	// 初始化
	memset(f, 0x3f, sizeof f);  // 初始化极大值
	int init = f[0][0];   // 取出初始值，便于判断无解
	ans = init;

	// 读入数据
	cin >> m >> n;
	while (n--) {
		int x, y, c;
		cin >> x >> y >> c;
		a[x][y] = c + 1;  // 0表无色，所以加1
	}

	// 深搜
	dfs(1, 1, 0, true);

	cout << (ans == init ? -1 : ans) << endl;
	return 0;
}

评测结果