【每日一题】LeetCode 3127.构造相同颜色的正方形（数组、枚举、矩阵）

【每日一题】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;
    }
}