博客信息

566. 重塑矩阵

发布时间:『 2018-12-09 01:03:57』  博客类别:  阅读(101) 评论(0)

在MATLAB中,有一个非常有用的函数 reshape,它可以将一个矩阵重塑为另一个大小不同的新矩阵,但保留其原始数据。

给出一个由二维数组表示的矩阵,以及两个正整数rc,分别表示想要的重构的矩阵的行数和列数。

重构后的矩阵需要将原始矩阵的所有元素以相同的行遍历顺序填充。

如果具有给定参数的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 矩阵。 所以输出原矩阵。

注意:

  1. 给定矩阵的宽和高范围在 [1, 100]。
  2. 给定的 r 和 c 都是正数。

 

@Test
public void matrixReshape() {
    int[][] nums=new int[][]{{1,2},{3,4}};
    Arrays.deepToString(matrixReshape(nums,4,1));

}

public int[][] matrixReshape(int[][] nums, int r, int c) {
    int sourceR=nums.length;
    int sourceC=nums[0].length;
    if (sourceR*sourceC!=r*c){
        return nums;
    }
    int tr=0;
    int tc=0;
    int[][] newNums=new int[r][c];
    for (int i = 0; i < sourceR; i++) {
        for (int j = 0; j < sourceC; j++) {
            newNums[tr][tc]=nums[i][j];
            tc++;
            if (tc==c){
                tc=0;
                tr++;
                if (tr==r){
                    tr=0;
                }
            }
        }

    }
    return newNums;
}

 

关键字:     算法       重塑矩阵  

上一篇:566. 重塑矩阵

下一篇:2019.02.14

评论信息
暂无评论