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;
        }
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值