题目
给你一幅由 N × N 矩阵表示的图像,其中每个像素的大小为 4 字节。请你设计一种算法,将图像旋转 90 度。
不占用额外内存空间能否做到?
示例 1:
给定 matrix =
[
[1,2,3],
[4,5,6],
[7,8,9]
],
原地旋转输入矩阵,使其变为:
[
[7,4,1],
[8,5,2],
[9,6,3]
]
示例 2:
给定 matrix =
[
[ 5, 1, 9,11],
[ 2, 4, 8,10],
[13, 3, 6, 7],
[15,14,12,16]
],
原地旋转输入矩阵,使其变为:
[
[15,13, 2, 5],
[14, 3, 4, 1],
[12, 6, 8, 9],
[16, 7,10,11]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/rotate-matrix-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
show me the code - golang
其实上来看到说不使用额外的内存空间,我第一反应是位运算,哈哈,后来发现想多了,因为每个数字占据4个字节,int类型也就4个字节,已经占满了。
所以根据我们的知识储备,swap这种交换操作,认为是O(1)的空间,所以属于没有占用额外的空间。我们根据旋转矩阵的基础知识。采用先沿着左上到右下的对角线翻转,再左右翻转就能得到顺时针旋转90度的矩阵啦。
时间复杂度:O(n^2)
空间复杂度:O(1)
func rotate(matrix [][]int) {
n := len(matrix)
//对角翻转,对角线为左上到右下
for i:=0;i<n-1;i++ {
for j:=i+1;j<n;j++ {
tmp := matrix[i][j]
matrix[i][j] = matrix[j][i]
matrix[j][i] = tmp
}
}
//左右翻转
mid := n>>1
for i:=0;i<n;i++ {
for j:=0;j<mid;j++ {
tmp := matrix[i][j]
matrix[i][j] = matrix[i][n-1-j]
matrix[i][n-1-j] = tmp
}
}
}
总结
本题用到了学习矩阵的一些基础知识,适时回顾一下。