leetcode_566

package leetcode.array;

import java.util.Arrays;

/**
 * Created by w84108989 on 2019/2/19.
 */
//在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 矩阵。 所以输出原矩阵。
public class juzhen_566 {
    public static void main(String[] args) {
       int[][] nums = {{1,2},{3,4}};
       matrixReshape(nums,1,2);
       matrixReshape(nums,2,3);
        matrixReshape(nums,1,4);
        System.out.println(Arrays.deepToString(matrixReshape(nums,1,4)));

    }

    public static int[][] matrixReshape(int[][] nums, int r, int c) {
        if(r * c != nums.length * nums[0].length){
            return nums;
        }else {
//            int [][] res = new int[r][c];
//            int []tmp = new int [nums.length * nums[0].length];
//            int l = 0;
//            for (int i = 0; i < nums.length; i++) {
//                for (int j = 0; j < nums[0].length; j++) {
//                    tmp[l] = nums[i][j];
//                    l++;
//                }
//            }
//            int k = 0;
//            for (int i = 0; i < r; i++) {
//                for (int j = 0; j < c; j++) {
//                    res[i][j] = tmp[k];
//                    k++;
//                    //System.out.println("res[][] = " + res[i][j]);
//                }
//            }
//            return res;
            int m = 0;
            int n = 0;
            int [][]res = new int[r][c];
            for (int i = 0; i < nums.length; i++) {
                for (int j = 0; j < nums[0].length; j++) {
                    res[m][n] = nums[i][j];
                    n++;
                    if (n == c){
                        m++;
                        n = 0;
                    }
                    if (m == r){
                        break;
                    }
                }
            }
            return res;
        }
    }
}