洛谷 P1123 取数游戏（Java）

题目描述

一个N×M的由非负整数构成的数字矩阵，你需要在其中取出若干个数字，使得取出的任意两个数字不相邻（若一个数字在另外一个数字相邻8个格子中的一个即认为这两个数字相邻），求取出数字和最大是多少。

输入格式

第1行有一个正整数T，表示了有T组数据。

对于每一组数据，第一行有两个正整数N和M，表示了数字矩阵为N行M列。

接下来N行，每行M个非负整数，描述了这个数字矩阵。

输出格式

T行，每行一个非负整数，输出所求得的答案。

输入输出样例

输入 #1

3
4 4
67 75 63 10
29 29 92 14
21 68 71 56
8 67 91 25
2 3
87 70 85
10 3 17
3 3
1 1 1
1 99 1
1 1 1

输出 #1

271
172
99

说明/提示

对于第1组数据，取数方式如下：

[67] 75 63 10

29 29 [92] 14

[21] 68 71 56

8 67 [91] 25

对于20%的数据，N,M≤3；

对于40%的数据，N,M≤4；

对于60%的数据，N,M≤5；

对于100%的数据，N,M≤6,T≤20。

解题思路

---

• 首先解读一下题目大意，给定有 T 组数据，在接下来有 N 和 M ，每组数据有N * M 个数从中选出若干个数，选中的数它附近（上，下，左，右，左上，右上，左下，右下）这八个方位的数不能再选，当然选中的数也不能再选，最后选出来的加起来最大。
• 看一下数据量，对于全部数据量也不超过N,M≤6,T≤20，由此可以推断出这道题的解法可以暴搜，在搜索的时候要知道每一个数都有俩个状态选和不选，在搜索的时候是按照一行一行的进行搜索，也就是说当 Y ≥ M 时要进行换行操作，当X ≥ N 表示N行以内的数字可以选中的都已选中。
• 如果还是不是太明白的话，就参考一下面的代码，代码中都有详细的注解。

代码

import java.io.BufferedOutputStream;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.io.StreamTokenizer;

public class Main{
	static class Input { // 优化输入
		BufferedReader is = new BufferedReader(new InputStreamReader(System.in));
		StreamTokenizer in = new StreamTokenizer(is);

		public int nextInt() throws IOException {
			in.nextToken();
			return (int) in.nval;
		}
	}

	static int arr[][]; // 存放数据
	static int n, m, max; // n：行，m：列，max：最大值
	static boolean vis[][]; // 用来标记
	static int arrX[] = { 0, -1, 1, 0, 0, -1, -1, 1, 1 }; // 八个方位包括自身
	static int arrY[] = { 0, 0, 0, -1, 1, -1, 1, -1, 1 };

	public static void main(String[] args) throws IOException {
		Input in = new Input();
		PrintWriter pr = new PrintWriter(new BufferedOutputStream(System.out)); // 优化输出
		int T = in.nextInt(); // T组数据
		while (T != 0) {
			n = in.nextInt();
			m = in.nextInt();
			arr = new int[n][m];
			vis = new boolean[n][m];
			for (int i = 0; i < n; i++)
				for (int j = 0; j < m; j++)
					arr[i][j] = in.nextInt();
			DFS(0, 0, 0); // 初始化都为零
			pr.println(max);
			max = 0;
			T--;
		}
		pr.flush();
	}

	/**
	 * @param x   表示第几行
	 * @param y   表示第几列
	 * @param sum 表示总和
	 */
	public static void DFS(int x, int y, int sum) {
		if (y >= m) { // 如果y大于等于m表示已经到达了这一行的尾部，就需要把x+=1，y=0（下一行，第一个数）开始搜
			x += 1;
			y = 0;
		}
		if (x >= n) { // 当x大于等n时就表示搜完了
			max = max > sum ? max : sum; // 更新最大值
			return;
		}
		DFS(x, y + 1, sum); // 不选这个数
		if (judge(x, y)) { // 选取这个数，当然也要判断这个数可以不可以选
			vis[x][y] = true; // 标记这个已被选中
			DFS(x, y + 1, sum + arr[x][y]); // 递归到下一层，y+1是为了把整行的数都要搜索到
			vis[x][y] = false;// 回溯取消标记
		}

	}

	public static boolean judge(int x, int y) { // 判断这个数的八个方位，包括这个数本身
		for (int i = 0; i <= 8; i++) {
			int X = x + arrX[i];
			int Y = y + arrY[i];
			if (X >= 0 && Y >= 0 && X < n && Y < m)
				if (vis[X][Y]) // 如果它附近有被选中的，就返回false
					return false;
		}
		return true; // 否则返回true
	}

}

---

总结：最后有什么不懂的或者有什么疑惑，欢迎您在评论区留言~