在MATLAB中，有一個非常有用的函式 reshape，它可以將一個矩陣重塑為另一個大小不同的新矩陣，但保留其原始資料。

給出一個由二維陣列表示的矩陣，以及兩個正整數r和c，分別表示想要的重構的矩陣的行數和列數。

重構後的矩陣需要將原始矩陣的所有元素以相同的行遍歷順序填充。

如果具有給定引數的reshape操作是可行且合理的，則輸出新的重塑矩陣；否則，輸出原始矩陣。

示例 1:

輸入: 
nums = 
[[1,2],
 [3,4]]
r = 1, c = 4
輸出: 
[[1,2,3,4]]
解釋:
行遍歷nums的結果是 [1,2,3,4]。新的矩陣是 1 * 4 矩陣, 用之前的元素值一行一行填充新矩陣。

示例 2:

輸入: 
nums = 
[[1,2],
 [3,4]]
r = 2, c = 4
輸出: 
[[1,2],
 [3,4]]
解釋:
沒有辦法將 2 * 2 矩陣轉化為 2 * 4 矩陣。 所以輸出原矩陣。

import java.util.Scanner;

public class Main {

    public static void main(String[] args)
    {

        Scanner sc=new Scanner(System.in);

        int n=sc.nextInt();
        int m=sc.nextInt();
        int [][]arr=new int[n][m];
        for (int i=0;i<n;i++)
            for (int j=0;j<n;j++)
                arr[i][j]=sc.nextInt();
            int r=sc.nextInt();
            int c=sc.nextInt();
       Solution.matrixReshape(arr,r,c);


    }

}


class Solution {
    public static int[][] matrixReshape(int[][] nums, int r, int c) {

        int [][]reu=new int[r][c];//新矩陣
        int x=0;    //新矩陣索引x
        int y=0;     //新矩陣索引y
        if(r*c!=nums.length*nums[0].length)//如果不符合，返回原矩陣
            return nums;
        else
        {
            for (int i=0;i<nums.length;i++)
            {
                for (int j = 0; j < nums[0].length; j++)
                {
                    reu[x][y] = nums[i][j];

                    if(y==reu[0].length-1)          //新如果矩陣的一行填滿，則換行賦值
                    {
                        x++;
                        y=0;
                    }
                    else
                        y++;
                }

            }
        }

//        for (int i=0;i<reu.length;i++) {
//            System.out.println();
//
//            for (int j = 0; j < reu[0].length; j++)
//                System.out.print(reu[i][j] + " ");
//        }
        return reu;
    }
}
 

執行時間較長，待優化