转置矩阵

给定一个矩阵 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 <= A.length <= 1000
1 <= A[0].length <= 1000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/transpose-matrix
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

class Solution {
public:
    vector<vector<int>> transpose(vector<vector<int>>& A) 
    {
        int hang = A.size();
        if(hang == 0)
            return A;
        int lie  = A[0].size();
        vector<vector<int>> ret;
        if(hang == lie)
        {
            for(int i = 0; i < hang; i++)
            {
                for(int j = i + 1; j < lie; j++)
                {
                    int tmp= A[i][j];
                    A[i][j] = A[j][i];
                    A[j][i] = tmp;
                }
            }
            return A;
        }
        else
        {
            vector<int> tmp(hang);
            for(int i = 0;  i < lie; i++)
            {
                ret.push_back(tmp);
            }
            for(int i = 0; i < hang; i++)
            {
                //ret[i][i] = A[i][i];
                for(int j = 0; j < lie; j++)
                {
                    // int tmp= A[i][j];
                    // A[i][j] = A[j][i];
                    // A[j][i] = tmp;
                    //ret[i][j] = A[j][i];
                    ret[j][i] = A[i][j];
                }
            }
        }
        return ret;
    }
};