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

简单到爆栈的矩阵转置:

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;
    }
};

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

猜你喜欢

转载自blog.csdn.net/weixin_47741017/article/details/114109209