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 移除并返回队列头部的元素