题目描述
题目链接:旋转图像
给定一个 n × n 的二维矩阵 matrix 表示一个图像。请你将图像顺时针旋转 90 度。
你必须在 原地 旋转图像,这意味着你需要直接修改输入的二维矩阵。请不要 使用另一个矩阵来旋转图像。
举例
示例 1:
输入:matrix = [[1,2,3],[4,5,6],[7,8,9]],输出:[[7,4,1],[8,5,2],[9,6,3]]
【注:图片来源:leetcode截图】
示例 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截图】
解题方法: 先水平翻转swap(matrix[i][j],matrix[n-1-i][j]),后主对角线翻转swap(matrix[i][j], matrix[j][i]。
代码
#include <iostream>
#include <vector>
#include <string>
void rotate(std::vector<std::vector<int>>& matrix) {
int rnum = matrix.size();
if (rnum <= 0) {
return;
}
int cnum = matrix[0].size();
if (cnum <= 0) {
return;
}
if (rnum == 1 && cnum == 1) { //1行1列,直接返回
return;
}
// 先水平翻转(注意原点是左上角,向下是x轴,向右是y轴,跟几何意义不一样)
int top = 0;
int bottom = cnum - 1;
for (int i = 0; i < rnum / 2; i++) {
for (int j = 0; j < cnum; j++) {
std::swap(matrix[i][j], matrix[rnum - 1 - i][j]);
}
}
// 再斜对称翻转(只翻转主对角线一边的元素)
for (int i = 0; i < rnum; i++) {
for (int j = 0; j < i; j++) {
std::swap(matrix[i][j], matrix[j][i]);
}
}
}
void print_matrix(std::vector<std::vector<int>>& matrix) {
for(auto& record : matrix) {
for (auto value : record) {
std::cout << value << " ";
}
std::cout << std::endl;
}
}
int main()
{
// case1: matrix = [[1,2,3],[4,5,6],[7,8,9]]==>[[7,4,1],[8,5,2],[9,6,3]]
std::cout << "========" << std::endl;
std::cout << "input:" << std::endl;
std::vector<std::vector<int>> matrix = {
{1, 2, 3},
{4, 5, 6},
{7, 8, 9}
};
print_matrix(matrix);
std::cout << "output:" << std::endl;
rotate(matrix);
print_matrix(matrix);
// case2: 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]]
std::cout << "========" << std::endl;
std::cout << "input:" << std::endl;
matrix = {
{5, 1, 9, 11},
{2, 4, 8, 10},
{13, 3, 6, 7},
{15, 14, 12, 16}
};
print_matrix(matrix);
std::cout << "output:" << std::endl;
rotate(matrix);
print_matrix(matrix);
return 0;
}
代码运行结果如下:
========
input:
1 2 3
4 5 6
7 8 9
output:
7 4 1
8 5 2
9 6 3
========
input:
5 1 9 11
2 4 8 10
13 3 6 7
15 14 12 16
output:
15 13 2 5
14 3 4 1
12 6 8 9
16 7 10 11