欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

一场风波过后,我就接着好好学习吧打卡(4)

程序员文章站 2022-07-14 23:10:23
...

简单到爆栈的矩阵转置:

class Solution {
public:
    vector<vector<int>> transpose(vector<vector<int>>& matrix) {
        int n=matrix[0].size();
        int m=matrix.size();
        vector<vector<int>> res(n,vector<int>(m));
        int k=m*n;
        for(int i=0;i<k;i++){
            res[i%n][i/n]=matrix[i/n][i%n];
        }
        return res;
    }
};

学到的无非就是一个

res[i%n][i/n]=matrix[i/n][i%n];

所以牢记就好,这样的话,n代表的就是矩阵的行数。转置之后的行数为matrix[0].size()。

第二题:https://leetcode-cn.com/problems/flipping-an-image/
其实也是简单的一批,但是可能我找到了最优解,只需要遍历一遍即可,感兴趣的话,小聪明还是有的

class Solution {
public:
    vector<vector<int>> flipAndInvertImage(vector<vector<int>>& image) {
        //相等则进行取反,否则不操作
        int n=image.size();
        int m=image[0].size();
        int mm=m/2;
        if(m%2==1) mm++;
        for(int i=0;i<n;i++){
            for(int j=0;j<mm;j++){
                if(image[i][j]==image[i][m-1-j]){
                    image[i][j]=image[i][m-1-j]=image[i][j]^1;
                }
            }
        }
        return image;
    }
};

牙疼,有点要命,智齿又开始骚动了。