【每日一题】LeetCode 3127.构造相同颜色的正方形(数组、枚举、矩阵)
1. 题目描述
给定一个二维 3 x 3 的矩阵 grid
,每个格子都是一个字符,要么是 ‘B’ 表示黑色,要么是 ‘W’ 表示白色。
任务是改变至多一个格子的颜色,使得矩阵中存在一个 2 x 2 颜色完全相同的正方形。
如果可以得到一个相同颜色的 2 x 2 正方形,那么返回 true
,否则返回 false
。
2. 思路分析
要解决这个问题,我们需要检查矩阵中是否存在一个 2 x 2 的子矩阵,使得这个子矩阵中的所有四个格子颜色相同。如果存在,直接返回 true
。如果不存在,我们尝试改变一个格子的颜色,再次检查是否存在这样的 2 x 2 子矩阵。
遍历子矩阵
遍历矩阵的每个可能的 2 x 2 子矩阵的左上角位置。
检查颜色
对于每个这样的位置,检查四个相邻格子的颜色是否相同。
返回结果
如果找到符合条件的子矩阵,返回 true
。
无解返回
如果遍历完所有可能的子矩阵都没有找到,返回 false
。
3. 输入示例
示例 1
grid = [["B","W","B"],["B","W","W"],["B","W","B"]]
输出:true
示例 2
grid = [["B","W","B"],["W","B","W"],["B","W","B"]]
输出:false
示例 3
grid = [["B","W","B"],["B","W","W"],["B","W","W"]]
输出:true
提示:
grid.length == 3
grid[i].length == 3
grid[i][j] 要么是 ‘W’ ,要么是 ‘B’ 。
4. 代码实现
class Solution {
/**
* 检查是否可以将矩阵中的一个格子颜色改变,使得存在一个2x2的同色正方形。
* @param grid 给定的二维字符矩阵,包含 'B' 和 'W'。
* @return 如果可以找到这样的2x2正方形,返回 true,否则返回 false。
*/
public boolean canMakeSquare(char[][] grid) {
int row = grid.length; // 矩阵的行数
int col = grid[0].length; // 矩阵的列数
// 遍历矩阵,寻找可能的2x2同色正方形的起始点
for (int i = 0; i < col - 1; i++) {
for (int j = 0; j < col - 1; j++) {
// 检查四个相邻的格子是否形成2x2同色正方形
// 情况1: 左上角和右上角以及左上角和左下角同色
if (grid[i][j] == grid[i][j + 1] && grid[i][j] == grid[i + 1][j]) {
return true;
}
// 情况2: 右上角和右下角以及左上角和右下角同色
else if (grid[i][j + 1] == grid[i][j] && grid[i][j + 1] == grid[i + 1][j + 1]) {
return true;
}
// 情况3: 右下角和左下角以及左下角和左上角同色
else if (grid[i][j + 1] == grid[i + 1][j + 1] && grid[i + 1][j] == grid[i + 1][j + 1]) {
return true;
}
// 情况4: 左下角和左上角以及左下角和右上角同色
else if (grid[i + 1][j] == grid[i][j] && grid[i + 1][j] == grid[i + 1][j + 1]) {
return true;
}
}
}
// 如果没有找到任何符合条件的2x2同色正方形,返回 false
return false;
}
}