LeetCode566——Reshape the Matrix

题意
将一个矩阵转换成指定行列数的新矩阵，若原矩阵元素数目与新矩阵数目不等，则返回新矩阵。

代码
方法一：将原矩阵中的元素依次取出同时放入新矩阵。

public class Solution {    public int[][] matrixReshape(int[][] nums, int r, int c) {        int r_original=nums.length;        int c_original=nums[0].length;        if(r_original*c_original==r*c){            int[][] newArray=new int[r][c];            for(int i=0;i<r_original*c_original;i++){                newArray[i/c][i%c]=nums[i/c_original][i%c_original];            }            return newArray;        }        else return nums;    }}

此处有两点需要注意：1、通过nums.length确定原矩阵的函数，通过nums[0].length确定原矩阵的列数；
2、通过nums[i/c_original][i%c_original]将原矩阵中的元素依次按行取出。

方法二：先将原矩阵中的元素按行取出放入队列，然后再从队列中存入新矩阵

import java.util.LinkedList;import java.util.Queue;public class Solution2 {    public int[][] matrixReshape(int[][] nums, int r, int c){        int r_original=nums.length;        int c_original=nums[0].length;        if(r_original*c_original==r*c) {            Queue <Integer> queue=new LinkedList <> ();            for(int i=0;i<nums.length;i++) {                for(int j=0;j<nums[0].length;j++) {                    queue.add(nums[i][j]);                }            }        int[][] result=new int[r][c];        for(int i=0;i<r;i++) {            for(int j=0;j<c;j++) {                result[i][j]=queue.remove();            }        }        return result;        }        else        return nums;    }}

此方法应该注意的是队列的使用以及
add 在队列尾部增加一个元素
remove 移除并返回队列头部的元素