题目 | 开心消消乐 |
难度 | 易 |
题目说明 | 给定一个 N 行 M 列的二维矩阵,矩阵中每个位置的数字取值为 0 或 1,矩阵示例如: 1 1 0 0 0 0 0 1 0 0 1 1 1 1 1 1 现需要将矩阵中所有的 1 进行反转为 0,规则如下: 1) 当点击一个 1 时,该 1 被反转为 0,同时相邻的上、下、左、右,以及左上、左下、右上、右下 8 个方向的 1 (如果存在 1)均会自动反转为 0; 2) 进一步地,一个位置上的 1 被反转为 0 时,与其相邻的 8 个方向的 1 (如果存在 1)均会自动反转为 0; 按照上述规则示例中的矩阵只最少需要点击 2 次后,所有均值 0 。请问,给定一个矩阵,最少需要点击几次后,所有数字均为 0? |
输入描述 | 第一行输入两个整数,分别表示矩阵的行数 N 和列数 M,取值范围均为 [1,100] 接下来 N 行表示矩阵的初始值,每行均为 M 个数,取值范围 [0,1]。 |
输出描述 | 输出一个整数,表示最少需要点击的次数。 |
补充说明 | 无 |
------------------------------------------------------ | |
示例 | |
示例1 | |
输入 | 3 3 1 0 1 0 1 0 1 0 1 |
输出 | 1 |
说明 | 上述样例中,四个角上的 1 均在中间的 1 的相邻 8 个方向上,因此只需要点击一次即可。 |
示例2 | |
输入 | 4 4 |
输出 | 2 |
说明 | 在上述 4 * 4 的矩阵中,只需要点击 2 次(左上角 和 右下角),即可将所有的 1 消除。 |
题目解读:
点击一个 1,它相邻八个方向的 1 变成 0,与此同时,这八个方向如果存在 1 变成 0,那么与它相邻的八个方向的 1 也会变成 0。
有点像扫雷游戏。
分析与思路:
此题可以采用广度优先搜索或深度优先搜索,遍历所有点击的情况。然后根据所有点击情况,计算出最小的点击次数。
时间复杂度为 O(),空间复杂度为 O()。
与《华为OD机考算法题:机器人活动区域》有些类似。
Java代码
- import java.util.ArrayList;
- import java.util.HashSet;
- import java.util.List;
- import java.util.Scanner;
- import java.util.Set;
-
- /**
- * 开心消消乐
- *
- * @since 2023.10.13
- * @version 0.2
- * @author Frank
- *
- */
- public class HappyCollapse {
- public static void main(String[] args) {
- Scanner sc = new Scanner(System.in);
- while (sc.hasNext()) {
- String line = sc.nextLine();
- String[] rc = line.split(" ");
- int row = Integer.parseInt(rc[0]);
- int column = Integer.parseInt(rc[1]);
-
- int[][] matrix = new int[row][column];
- for (int i = 0; i < row; i++) {
- line = sc.nextLine();
- String[] strColumnValue = line.split(" ");
- int[] number = new int[column];
- for (int j = 0; j < column; j++) {
- number[j] = Integer.parseInt(strColumnValue[j]);
- }
- matrix[i] = number;
- }
-
- processHappyCollapse(matrix, row, column);
- }
-
- }
-
- static class Node {
- int i;
- int j;
-
- public Node(int i, int j) {
- this.i = i;
- this.j = j;
- }
-
- @Override
- public int hashCode() {
- return (i + " " + j).hashCode();
- }
-
- @Override
- public boolean equals(Object obj) {
- if (!(obj instanceof Node))
- return false;
- Node node = (Node) obj;
- return node.i == i && node.j == j;
- }
- }
-
- private static void processHappyCollapse(int[][] matrix, int row, int column) {
- Set
nodeSet = new HashSet(); - List
nodeList = new ArrayList(); - for (int i = 0; i < row; i++) {
- for (int j = 0; j < column; j++) {
- if (matrix[i][j] == 1) {
- Node node = new Node(i, j);
- nodeList.add(node);
-
- nodeSet.add(node);
- }
- }
- }
-
- int[] rowColumn = new int[2];
- rowColumn[0] = row;
- rowColumn[1] = column;
-
- int minCnt = Integer.MAX_VALUE;
- for (int i = 0; i < nodeList.size(); i++) {
- List
copyOfList = new ArrayList(); - copyOfList.addAll(nodeList);
-
- Set
copyOfSet = new HashSet(); - copyOfSet.addAll(nodeSet);
-
- Node node = nodeList.get(i);
-
- copyOfList.remove(i);
- copyOfSet.remove(node);
-
- int cnt = startClickNode(node, copyOfList, copyOfSet, rowColumn);
- if (cnt < minCnt) {
- minCnt = cnt;
- }
- }
- System.out.println(minCnt);
-
- }
-
- private static int startClickNode(Node node, List
nodeList, Set nodeSet, int[] rowColumn) { - int clickCnt = 1;
- clickCollapseNode(node, nodeList, nodeSet, rowColumn);
-
- if (nodeList.size() == 0) {
- return 1;
- }
-
- int minCnt = Integer.MAX_VALUE;
- for (int i = 0; i < nodeList.size(); i++) {
- List
copyOfList = new ArrayList(); - copyOfList.addAll(nodeList);
-
- Set
copyOfSet = new HashSet(); - copyOfSet.addAll(nodeSet);
-
- Node curNode = nodeList.get(i);
-
- copyOfList.remove(i);
- copyOfSet.remove(curNode);
-
- int cnt = startClickNode(curNode, copyOfList, nodeSet, rowColumn);
- if (cnt < minCnt) {
- minCnt = cnt;
- }
-
- }
- return clickCnt + minCnt;
- }
-
- private static void clickCollapseNode(Node node, List
nodeList, Set nodeSet, int[] rowColumn) { - int row = rowColumn[0];
- int column = rowColumn[1];
-
- if (node.i >= 1) {
- colapseNode(new Node(node.i - 1, node.j), nodeList, nodeSet, rowColumn);
- }
- if (node.i < row - 1) {
- colapseNode(new Node(node.i + 1, node.j), nodeList, nodeSet, rowColumn);
- }
- if (node.j >= 1) {
- colapseNode(new Node(node.i, node.j - 1), nodeList, nodeSet, rowColumn);
- }
- if (node.j < column - 1) {
- colapseNode(new Node(node.i, node.j + 1), nodeList, nodeSet, rowColumn);
- }
- if (node.i >= 1 && node.j >= 1) {
- colapseNode(new Node(node.i - 1, node.j - 1), nodeList, nodeSet, rowColumn);
- }
- if (node.i >= 1 && node.j < column - 1) {
- colapseNode(new Node(node.i - 1, node.j + 1), nodeList, nodeSet, rowColumn);
- }
- if (node.i < row - 1 && node.j >= 1) {
- colapseNode(new Node(node.i + 1, node.j - 1), nodeList, nodeSet, rowColumn);
- }
- if (node.i < row - 1 && node.j < column - 1) {
- colapseNode(new Node(node.i - 1, node.j + 1), nodeList, nodeSet, rowColumn);
- }
- }
-
- private static void colapseNode(Node node, List
nodeList, Set nodeSet, int[] rowColumn) { - if (nodeSet.contains(node)) {
- nodeList.remove(node);
- nodeSet.remove(node);
- clickCollapseNode(node, nodeList, nodeSet, rowColumn);
- }
- }
-
- }
JavaScript代码
- const rl = require("readline").createInterface({ input: process.stdin });
- var iter = rl[Symbol.asyncIterator]();
- const readline = async () => (await iter.next()).value;
- void async function() {
- while (line = await readline()) {
- var rowColumn = line.split(" ");
- var row = parseInt(rowColumn[0]);
- var column = parseInt(rowColumn[1]);
-
- var matrix = new Array();
- for (var i = 0; i < row; i++) {
- line = await readline();
- var strNumbers = line.split(" ");
- var numbers = new Array(column);
- for (var j = 0; j < column; j++) {
- numbers[j] = parseInt(strNumbers[j]);
- }
- matrix[i] = numbers;
- }
-
- processHappyCollapse(matrix, row, column);
- }
- }();
-
- function processHappyCollapse(matrix, row, column) {
- var nodeSet = new Set();
- var nodeList = new Array();
- for (var i = 0; i < row; i++) {
- for (var j = 0; j < column; j++) {
- if (matrix[i][j] == 1) {
- var node = i + "," + j;
- nodeList.push(node);
- nodeSet.add(node);
- }
- }
- }
-
- var rowColumn = new Array();
- rowColumn[0] = row;
- rowColumn[1] = column;
-
- var minCnt = Number.MAX_VALUE;
- for (var i = 0; i < nodeList.length; i++) {
- var copyOfList = Array.from(nodeList);
- var copyOfSet = new Set(nodeSet);
-
- var node = nodeList[i];
-
- copyOfList.splice(i, 1);
- copyOfSet.delete(node);
-
- var cnt = startClickNode(node, copyOfList, copyOfSet, rowColumn);
- if (cnt < minCnt) {
- minCnt = cnt;
- }
- }
- console.log(minCnt);
- }
-
- function startClickNode(node, nodeList, nodeSet, rowColumn) {
- var clickCnt = 1;
- clickCollapseNode(node, nodeList, nodeSet, rowColumn);
-
- if (nodeList.length == 0) {
- return 1;
- }
-
- var minCnt = Number.MAX_VALUE;
- for (var i = 0; i < nodeList.length; i++) {
- var copyOfList = Array.from(nodeList);
- var copyOfSet = new Set(nodeSet);
-
- var curNode = nodeList[i];
-
- copyOfList.splice(i, 1);
- copyOfSet.delete(curNode);
-
- var cnt = startClickNode(curNode, copyOfList, nodeSet, rowColumn);
- if (cnt < minCnt) {
- minCnt = cnt;
- }
- }
- return clickCnt + minCnt;
- }
-
- function clickCollapseNode(node, nodeList, nodeSet, rowColumn) {
- var row = rowColumn[0];
- var column = rowColumn[1];
-
- var nodeStrValue = node.split(",");
- var nodeValue = new Array();
- nodeValue[0] = parseInt(nodeStrValue[0]);
- nodeValue[1] = parseInt(nodeStrValue[1]);
-
- for (var i = -1; i <= 1; i++) {
- if (nodeValue[0] + i < 0 || nodeValue[0] + i > row - 1) {
- continue;
- }
- for (var j = -1; j <= 1; j++) {
- if (i == 0 && j == 0) {
- continue;
- }
- if (nodeValue[1] + j < 0 || nodeValue[1] + j > column - 1) {
- continue;
- }
- colapseNode((nodeValue[0] + i) + "," + (nodeValue[1] + j), nodeList, nodeSet, rowColumn);
- }
- }
-
- }
-
- function colapseNode(node, nodeList, nodeSet, rowColumn) {
- if (nodeSet.has(node)) {
- var idx = nodeList.indexOf(node);
- nodeList.splice(idx, 1);
- nodeSet.delete(node);
- clickCollapseNode(node, nodeList, nodeSet, rowColumn);
- }
- }
虽然此题的难度标记为“易”,实际编码量比较大。难度并不小,在短时间内写出代码并不是很容易。
(完)