3127. 构造相同颜色的正方形

题目描述：

给你一个二维 3 x 3 的矩阵 grid ，每个格子都是一个字符，要么是 'B' ，要么是 'W' 。字符 'W' 表示白色，字符 'B' 表示黑色。

你的任务是改变 至多一个 格子的颜色，使得矩阵中存在一个 2 x 2 颜色完全相同的正方形。

如果可以得到一个相同颜色的 2 x 2 正方形，那么返回 true ，否则返回 false 。

示例 1：

输入： grid = [["B","W","B"],["B","W","W"],["B","W","B"]]
输出： true
解释：
修改 `grid[0][2]` 的颜色，可以满足要求。

示例 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` 已经包含一个 `2 x 2` 颜色相同的正方形了。

提示：

• grid.length == 3
• grid[i].length == 3
• grid[i][j] 要么是 'W' ，要么是 'B' 。

---

解题分析及思路：

方法：枚举

思路：

枚举所有（四种）情况，以单元格(i,j)为例，只需判断四个方向是否满足条件即可。

即判断除自己以外能组成2 x 2的正方单元格颜色是否一致即可（因为可以改变一个格子颜色），无需关注为哪一种颜色。

• 左上角：(i-1,j-1)到(i,j)是否满足条件，即判断(i-1,j-1)、(i-1,j)、(i,j-1)是否相同
• 右上角：(i-1,j+1)到(i,j)是否满足条件，即判断(i-1,j+1)、(i-1,j)、(i,j+1)是否相同
• 左下角：(i+1,j-1)到(i,j)是否满足条件，即判断(i+1,j-1)、(i+1,j)、(i,j-1)是否相同
• 右下角：(i+1,j+1)到(i,j)是否满足条件，即判断(i+1,j+1)、(i+1,j)、(i,j+1)是否相同

func canMakeSquare(grid [][]byte) bool {
	for i := range grid {
		for j := range grid[i] {
			if i-1 >= 0 && j-1 >= 0 && grid[i-1][j-1] == grid[i-1][j] && grid[i][j-1] == grid[i-1][j] {
				return true
			}
			if i-1 >= 0 && j+1 < len(grid[i]) && grid[i-1][j] == grid[i-1][j+1] && grid[i][j+1] == grid[i-1][j+1] {
				return true
			}
			if i+1 < len(grid) && j-1 >= 0 && grid[i][j-1] == grid[i+1][j-1] && grid[i+1][j-1] == grid[i+1][j] {
				return true
			}
			if i+1 < len(grid) && j+1 < len(grid[i]) && grid[i+1][j] == grid[i][j+1] && grid[i+1][j] == grid[i+1][j+1] {
				return true
			}
		}
	}
	return false
}

复杂度：

• 时间复杂度：O(N₂)，N为3
• 空间复杂度：O(1)

执行结果：

• 执行耗时:0 ms,击败了100.00% 的Go用户
• 内存消耗:2.1 MB,击败了100.00% 的Go用户