Leetcode-转置矩阵(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]]

解题思路
该题比较简单
其实就是个将R,C 变成 C,R
转置矩阵就是把 R 行 C 列的矩阵，转成 C 行 R 列的矩阵，原来矩阵中 matrix[i][j] 的位置，会交换到新矩阵的ans[j][i] 位置。

代码
Python

class Solution:
    def transpose(self, matrix: List[List[int]]) -> List[List[int]]:
        R,C = len(matrix),len(matrix[0])
        ans = [[0]*R for i in range(C)]
        for i in range(R):
            for j in range(C):
                ans[j][i] = matrix[i][j]
        return ans

C++

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

Java

class Solution {
    public int[][] transpose(int[][] matrix) {
        int R = matrix.length;
        int C = matrix[0].length;
        int[][] ans = new int[C][R];
        for(int i = 0; i < R;i++){
            for(int j = 0; j < C; j++){
                ans[j][i] = matrix[i][j];
            }
        }
        return ans;
    }
}