566重塑矩阵
这题也不需要什么算法,简单地遍历就行了
class Solution {
public:
vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
int m = nums.size();
int n = nums[0].size();
if(m * n != r * c) return nums;
vector<vector<int>> res(r);
int count = 0;
int row = 0;
for(int i = 0; i < m; i ++){
for(int j = 0; j < n; j++){
res[row].push_back(nums[i][j]);
count++;
if(count == c){
count = 0;
row++;
}
}
}
return res;
}
};
看了题解之后,发现人家写的比我简单很多,抓住前后下标的关系,用一个等式就能写出来
初始化可以这么写
vector<vector<int>> ans(r, vector<int>(c))
for (int x = 0; x < m * n; ++x) {
ans[x / c][x % c] = nums[x / n][x % n];
}
时间复杂度O(mn),空间复杂度O(1)
48旋转图像
顺时针旋转,并且要求原地旋转,这和189的环状数组解法是一样的,区别在于判断需要进行的替换的有哪些数,判断外部的循环条件才是难点(如果另外开一个visit数组判断当然就非常简单,但是显然我们不会这么做)
我是画了一下图,找到了循环的条件
- 首先,替换是一圈一圈进行的,进行替换的当前和下一个的下标是(i, j)-> (j, n - i - 1),总共需要替换的圈数是n/2
- 在第i圈,有i 到n - i - 2需要分别进行环状替换
class Solution {
public:
void rotate(vector<vector<int>>& matrix) {
int n = matrix.size();
for(int i = 0; i < n/2; i++){ //几圈
for(int j = i; j < n - i - 1; j++){//这圈中要交换的数字
int pre = matrix[i][j];
int curi = i;
int curj = j;
do{
int nexti = curj;
int nextj = n - curi - 1;
swap(pre, matrix[nexti][nextj]);
curi = nexti;
curj = nextj;
}while(curi != i || curj != j);
}
}
}
};
又在简单问题上出现了错误,用do while循环,while里面的条件是当条件满足的时候进行循环,不满足的时候退出
时间复杂度O(N平方),空间O(1)
因为数组每个都要旋转,所以数组全部遍历一遍是无法避免的