力扣刷题之867. 转置矩阵

题目简述

给你一个二维整数数组 matrix， 返回 matrix 的 转置矩阵 。

矩阵的 转置 是指将矩阵的主对角线翻转，交换矩阵的行索引与列索引。

示例 1：

输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出：[[1,4,7],[2,5,8],[3,6,9]]

示例 2：

输入：matrix = [[1,2,3],[4,5,6]]
输出：[[1,4],[2,5],[3,6]]

提示

• m == matrix.length
• n == matrix[i].length
• 1 <= m, n <= 1000
• 1 <= m * n <= 105
• -10^9 <= matrix[i][j] <= 10^9

题目分析

看到题目的第一眼就想到直接把ij互换即可，但是如果简单的for循环会把已经换好的再换回去，最后和初始一样，所以其实只需要取对角线右上方的元素即可。代码如下：

class Solution {
public:
    vector<vector<int>> transpose(vector<vector<int>>& matrix) {
        int n=matrix.size(),m=matrix[0].size(),tmp,a=1;
        for(int i=0;i<m-1;i++){
            for(int j=a;j<n;j++){
                if(i!=j){
                    tmp=matrix[i][j];
                    matrix[i][j]=matrix[j][i];
                    matrix[j][i]=tmp;
                }
            }
            a++;
        }
        return matrix;
    }
};

但是实际运行后发现无法通过，忽略了行列不相等的情况，于是换一种思路，代码如下：

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