LeetCode 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

执行用时为 12 ms 的范例：

static const auto _____ = []()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    return nullptr;
}();

class Solution {
public:
    vector<vector<int>> transpose(vector<vector<int>>& A) {
        int r = A.size();
        int l = A[0].size();
        if(r != l)
        {
            vector<vector<int>> B(l, vector<int>(r));
            for(int i=0; i<r; i++)
            {
                for(int j=0; j<l; j++)
                {
                    B[j][i] = A[i][j];
                }
            }
            return B;
        }
        else
        {
            for(int i=0; i<r; i++)
            {
                for(int j=i+1; j<r; j++)
                {
                        swap(A[i][j], A[j][i]);
                }
            }
            return A;
            
        }
        
    }
};

我的解答：

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

#STL是真的好用啊！