E3643.垂直翻转子矩阵
two pointers, matrix, https://leetcode.cn/problems/flip-square-submatrix-vertically/
给你一个 m x n 的整数矩阵 grid,以及三个整数 x、y 和 k。
整数 x 和 y 表示一个 正方形子矩阵 的左上角下标,整数 k 表示该正方形子矩阵的边长。
你的任务是垂直翻转子矩阵的行顺序。
返回更新后的矩阵。
示例 1:

输入: grid = [[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]], x = 1, y = 0, k = 3
输出: [[1,2,3,4],[13,14,15,8],[9,10,11,12],[5,6,7,16]]
解释:
上图展示了矩阵在变换前后的样子。
示例 2:

输入: grid = [[3,4,2,3],[2,3,4,2]], x = 0, y = 2, k = 2
输出: [[3,4,4,2],[2,3,2,3]]
解释:
上图展示了矩阵在变换前后的样子。
提示:
m == grid.lengthn == grid[i].length1 <= m, n <= 501 <= grid[i][j] <= 1000 <= x < m0 <= y < n1 <= k <= min(m - x, n - y)
这个问题要求我们将矩阵中一个指定的
算法思路
- 确定范围:
- 子矩阵的左上角坐标为
,边长为 。 - 行范围是从
到 。 - 列范围是从
到 。
- 子矩阵的左上角坐标为
- 垂直翻转的含义:
- 垂直翻转意味着将子矩阵的第一行与最后一行交换,第二行与倒数第二行交换,依此类推。
- 注意:只有子矩阵内部(即列下标在
之间)的元素需要交换,该行中位于子矩阵之外的元素保持不变。
- 实现步骤:
使用双指针法,令
top = x,bottom = x + k - 1。当
top < bottom时:- 遍历列下标
从 到 。 - 交换
grid[top][j]和grid[bottom][j]。 top向下移动一位(top += 1),bottom向上移动一位(bottom -= 1)。
Python 代码实现
- 遍历列下标
python
from typing import List
class Solution:
def reverseSubmatrix(self, grid: List[List[int]], x: int, y: int, k: int) -> List[List[int]]:
# 初始化双指针,分别指向子矩阵的最顶行和最底行
top = x
bottom = x + k - 1
# 进行垂直翻转,直到指针相遇或交错
while top < bottom:
# 仅交换子矩阵覆盖的列范围 [y, y + k - 1]
for j in range(y, y + k):
grid[top][j], grid[bottom][j] = grid[bottom][j], grid[top][j]
# 移动指针
top += 1
bottom -= 1
return grid复杂度分析
- 时间复杂度:
。我们需要遍历并交换子矩阵中大约一半的元素。由于 且 ,最大计算次数约为 次,效率非常高。 - 空间复杂度:
。我们是在原矩阵上进行原地修改,除了循环变量外没有使用额外的空间。
示例解析(示例 1)
输入:grid 为 4x4 矩阵,x=1, y=0, k=3
- 子矩阵行范围:1 到 3(包含 1, 2, 3 行)。
- 子矩阵列范围:0 到 2(包含 0, 1, 2 列)。
- 第 1 行的
[5, 6, 7]与第 3 行的[13, 14, 15]交换。 - 第 2 行是中间行,保持不变。
- 结果:第一行不变,二四行部分交换,三行不变。符合预期输出。
class Solution:
def reverseSubmatrix(self, grid: List[List[int]], x: int, y: int, k: int) -> List[List[int]]:
m, n = len(grid), len(grid[0])
for i in range(k//2):
for j in range(y, y + k):
grid[x + i][j], grid[x+k-1 - i][j] = grid[x+k-1 - i][j], grid[x + i][j]
return grid