2019-12-27-LeetCode【867. 转置矩阵】

867. 转置矩阵

给定一个矩阵 A， 返回 A 的转置矩阵。

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

示例 1：

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

示例2：

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

提示：

1. 1 <= A.length <= 1000
2. 1 <= A[0].length <= 1000

思路

创建一个矩阵，遍历取值即可。

代码

class Solution {
public:
    //只针对行和列相等的情况
    // vector<vector<int>> transpose(vector<vector<int>>& A) {
        
    //     int R=A.size();
    //     int C=A[0].size();
    //     int size = R*C;
    //     if(R!=C) return A;

    //     for(int i =0;i<size;i++)
    //     {
    //         if(i/C<i%C){
    //         int temp =A[i/C][i%C];//把原来的给temp
    //         A[i/C][i%C]=A[i%C][i/C];//后来的给原来
    //         A[i%C][i/C]=temp;//temp给后来的
    //         }
    //     }
    //     return A;
    // }

     vector<vector<int>> transpose(vector<vector<int>>& A) {
        int R=A.size();
        int C=A[0].size();
        int size = R*C;
        vector<vector<int> > res(C,vector<int>(R));//创建一个返回vector，指定转置后的行和列数
        for(int i=0;i<C;i++)
        {
            for(int j=0;j<R;j++)
            {
                res[i][j]=A[j][i];
            }
        }
        return res;
     }

};