LeetCode C++27-旋转图像

题目描述

       题目链接：旋转图像

      给定一个 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