【LeetCode】数组-5(566)-按照要求输出矩阵

题目要求：

思路一：借助队列，先顺序读入input矩阵，然后按照output要求向output矩阵输入

【正确代码】

 class Solution {
     public int[][] matrixReshape(int[][] nums, int r, int c) {
         int[][] res = new int[r][c];
         if (nums == null || nums.length * nums[0].length != r * c) {
             return nums;
         }
         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]);
             }
         }
         for (int i = 0; i < r; i++) {
             for (int j = 0; j < c; j++) {
                 res[i][j] = queue.remove();
             }
         }
         return res;
     }
 }

时间复杂度：O(m*n)

空间复杂度：O(m*n)

思路二：不借助辅助空间，直接根据output的要求向output矩阵输入

正常向output矩阵中读入数据，如果 cols > c 的话，rows++

【正确代码】（蓝色标记处，第一次写时出错）

 class Solution {
     public int[][] matrixReshape(int[][] nums, int r, int c) {
         int[][] res = new int[r][c];
         if (nums == null || nums.length * nums[0].length != r * c) {
             return nums;
         }
         int rows = 0, cols = 0;
         for (int i = 0; i < nums.length; i++) {
             for (int j = 0; j < nums[0].length; j++) {
                 res[rows][cols] = nums[i][j];
                 cols++;
                 if (cols >= c) {
                     rows++;
                     cols = 0;
                 }
             }
         }
         return res;
     }
 }

时间复杂度：O(m*n)

空间复杂度；O(m*n)

思路三：使用矩阵转换的公式 nums[n∗i+j]

满足基本的 nums.length * nums[0].length == r * c 条件情况下，使用矩阵目前的 （长度／c ）决定行数，（长度 % c)决定列数

【正确代码】（今天什么状态，刚刚又把count++给丢了「囧」）

 class Solution {
     public int[][] matrixReshape(int[][] nums, int r, int c) {
         int[][] res = new int[r][c];
         if (nums == null || nums.length * nums[0].length != r * c) {
             return nums;
         }
         int count = 0;
         for (int i = 0; i < nums.length; i++) {
             for (int j = 0; j < nums[0].length; j++) {
                 res[count / c][count % c] = nums[i][j];
                 count++;
             }
         }
         return res;
     }
 }

时间复杂度：O(m*n)

空间复杂度：O(m*n)