牛客算法基础noob50 矩阵转置

描述

牛牛有一个矩阵 𝑎𝑛,𝑚an,m​，他想编写一个程序，计算转置后的矩阵 𝑎𝑛,𝑚⊤an,m⊤​。

【名词解释】
【转置矩阵】转置矩阵是将矩阵的行列互换得到的新矩阵。

输入描述：

在一行中输入两个整数 𝑛,𝑚n,m (1≦𝑛,𝑚≦10)(1≦n,m≦10)，分别表示矩阵的行数和列数。
接下来 𝑛n 行，每行输入 𝑚m 个整数 𝑎𝑖,1,𝑎𝑖,2,…,𝑎𝑖,𝑚ai,1​,ai,2​,…,ai,m​ (−109≦𝑎𝑖,𝑗≦109)(−109≦ai,j​≦109)，用空格分隔。

输出描述：

输出 𝑚m 行 𝑛n 列，为矩阵的转置结果。每个元素之后使用一个空格。

import java.util.Scanner;

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();//表示矩阵的行数
        int m = in.nextInt();//表示矩阵的列数
        int[][] arr1 = new int[n][m];
        int[][] arr2 = new int[m][n];
        for(int i = 0;i<n;i++){//遍历每一行
            for(int j = 0;j<m;j++){//遍历每一列
                arr1[i][j] = in.nextInt();
                arr2[j][i] = arr1[i][j];//转置
            }
        }
        //输出转置矩阵
        for(int i  = 0;i<m;i++){//转置矩阵的行：0到m-1
            for(int j = 0;j<n;j++){//转置矩阵的列：0到n-1
                System.out.print(arr2[i][j] + " ");//每个元素之后使用一个空格
            }
            System.out.println();//每行结束换行
        }
        
    }
}