原题链接:旋转图像
个人解法
思路:
这个就是模拟坐标的变换,可以将坐标写下来,发现每一个坐标进行变换后,都会导致被变化的坐标也需要变换过去,才能确保不会发生值被覆盖。那么这就是一个递归的做法,当然也可以利用循环来做。
注意,这里进行坐标变换的时候要记录那些坐标已经进行了变换,这样做是为了再进行连锁变换的时候能及时停止。
时间复杂度: O ( n 2 ) O(n^2) O(n2)
代码:
class Solution {
public:
void trans(vector<vector<int>>& matrix, set<pair<int, int>> &st, int i, int j) {
int n = matrix.size();
int t_i = j, t_j = n - 1 - i;
int num = matrix[i][j], tmp = 0;
while(!st.count({t_i, t_j})) {
tmp = num;
num = matrix[t_i][t_j];
matrix[t_i][t_j] = tmp;
st.insert({t_i, t_j});
i = t_i, j = t_j;
t_i = j, t_j = n - 1 - i;
}
matrix[t_i][t_j] = num;
}
void rotate(vector<vector<int>>& matrix) {
int n = matrix.size();
set<pair<int, int>> st;
for(int i = 0;i < n;i ++) {
for(int j = 0;j < n;j ++) {
if(!st.count({i, j})) {
trans(matrix, st, i, j);
}
}
}
}
};
更好的解法
思路:
由模拟坐标的变化可以发现,坐标变换的规律是matrix[row][col] -> matrix[col][n - row - 1]
那么我们可以
先进行上下的水平变换matrix[row][col] -> matrix[n - row - 1][col]
再进行对称变换 matrix[n - row - 1][col] -> matrix[col][n - row - 1]
时间复杂度: O ( n 2 ) O(n^2) O(n2)
代码:
class Solution {
public:
void rotate(vector<vector<int>>& matrix) {
int n = matrix.size();
for(int i = 0;i < n / 2;i ++) {
for(int j = 0;j < n;j ++)
swap(matrix[i][j], matrix[n - i - 1][j]);
}
for(int i = 0;i < n;i ++) {
for(int j = 0;j < i;j ++)
swap(matrix[i][j], matrix[j][i]);
}
}
};