LeetCode初级算法：旋转图像

以下是本人的C++算法学习笔记，记录在博客上以供自己随时查阅

题目描述：给定一个 n × n 的二维矩阵表示一个图像，将图像顺时针旋转 90 度。
说明：你必须在原地旋转图像，这意味着你需要直接修改输入的二维矩阵。请不要使用另一个矩阵来旋转图像。

解题思路：先将原数组转置，再将每行数组元素进行旋转

旋转函数：

void rotate(vector<vector<int>>& matrix) {

	int n = matrix.size();
	for (int i = 0; i < n; i++) {
		for (int j = i + 1; j < n; j++) {
			swap(matrix[i][j], matrix[j][i]);
		}
		reverse(matrix[i].begin(), matrix[i].end());
	}
}

完整代码：

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;

void rotate(vector<vector<int>>& matrix) {

	int n = matrix.size();
	for (int i = 0; i < n; i++) {
		for (int j = i + 1; j < n; j++) {
			swap(matrix[i][j], matrix[j][i]);
		}
		reverse(matrix[i].begin(), matrix[i].end());
	}
}

int main() {

	int n = 10;

	// 使用vector建立一个二维数组
	// 1.建立一个n*n大小的数组
	vector<vector<int>> matrix(n);
	for (int i = 0; i < n; i++)
		matrix[i].resize(n);

	// 2.初始化数组
	// matrix.size() 二维数组的行数
	// matrix[0].size() 二维数组的列数
	for (int i = 0; i < matrix.size(); i++) {
		for (int j = 0; j < matrix[0].size(); j++) {
			matrix[i][j] = i;
		}
	}
	
	rotate(matrix);

	// 遍历数组
	for (int i = 0; i < 10; i++) {
		for (int j = 0; j < 10; j++) {
			cout << matrix[i][j] << " ";
		}
		cout << endl;
	}
	cout << endl;

	system("pause");
	return 0;
}

原始数组：

 顺时针旋转90°之后的数组：