1. 题目描述
给定一个二维 3 x 3 的矩阵 grid,每个格子都是一个字符,要么是 ‘B’ 表示黑色,要么是 ‘W’ 表示白色。任务是改变至多一个格子的颜色,使得矩阵中存在一个 2 x 2 颜色完全相同的正方形。如果可以得到一个相同颜色的 2 x 2 正方形,那么返回 true,否则返回 false。示例:
grid = [["B","W","B"],["B","W","W"],["B","W","B"]]
输出:true
grid = [["B","W","B"],["W","B","W"],["B","W","B"]]
输出:false
grid = [["B","W","B"],["B","W","W"],["B","W","W"]]
输出:true
提示:
grid.length == 3
grid[i].length == 3
grid[i][j] 要么是 ‘W’ ,要么是 ‘B’ 。
2.分析
要解决这个问题,我们需要检查矩阵中是否存在一个 2 x 2 的子矩阵,使得这个子矩阵中的所有四个格子颜色相同。如果存在,直接返回 true
。如果不存在,我们尝试改变一个格子的颜色,再次检查是否存在这样的 2 x 2 子矩阵。对于每个这样的位置,检查四个相邻格子的颜色是否相同。如果找到符合条件的子矩阵,返回 true
。如果遍历完所有可能的子矩阵都没有找到,返回 false
。
3.编程实现
(1)Java实现
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< row-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;
}
public static void main(String[] args)
{
Solution solution = new Solution();
// 示例矩阵
// char[][] grid = {
// {'B', 'W', 'B'},
// {'B', 'W', 'W'},
// {'B', 'W', 'B'}
// };
// char[][] grid = {
// {'B', 'W', 'B'},
// {'W', 'B', 'W'},
// {'B', 'W', 'B'}
// };
char[][] grid = {
{'B', 'W', 'B'},
{'B', 'W', 'W'},
{'B', 'W', 'W'}
};
System.out.println(solution.canMakeSquare(grid)); // 输出: false
}
}
(2)C++实现
#include <iostream>
#include <vector>
class Solution
{
public:
/**
* 检查是否可以将矩阵中的一个格子颜色改变,使得存在一个2x2的同色正方形。
* @param grid 给定的二维字符矩阵,包含 'B' 和 'W'。
* @return 如果可以找到这样的2x2正方形,返回 true,否则返回 false。
*/
bool canMakeSquare(std::vector<std::vector<char>>& grid)
{
int row = grid.size(); // 矩阵的行数
int col = grid[0].size(); // 矩阵的列数
// 遍历矩阵,寻找可能的2x2同色正方形的起始点
for (int i = 0; i < row - 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;
}
};
int main()
{
Solution solution;
// 示例矩阵
std::vector<std::vector<char>> grid = {
{'B', 'W', 'B'},
{'B', 'W', 'W'},
{'B', 'W', 'W'}
};
std::cout << solution.canMakeSquare(grid) << std::endl; // 输出: false
return 0;
}