ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

566. Reshape the Matrix

2021-01-02 21:34:06  阅读:253  来源: 互联网

标签:Matrix nums Reshape 566 matrix reshape new col row


package LeetCode_566

/**
 * 566. Reshape the Matrix
 * https://leetcode.com/problems/reshape-the-matrix/
 * In MATLAB, there is a very useful function called 'reshape',
 * which can reshape a matrix into a new one with different size but keep its original data.
You're given a matrix represented by a two-dimensional array,
and two positive integers r and c representing the row number and column number of the wanted reshaped matrix, respectively.
The reshaped matrix need to be filled with all the elements of the original matrix in the same row-traversing order as they were.
If the 'reshape' operation with given parameters is possible and legal, output the new reshaped matrix; Otherwise, output the original matrix.

Example 1:
Input:
nums =
[[1,2],
[3,4]]
r = 1, c = 4
Output:
[[1,2,3,4]]
Explanation:
The row-traversing of nums is [1,2,3,4]. The new reshaped matrix is a 1 * 4 matrix, fill it row by row by using the previous list.

Example 2:
Input:
nums =
[[1,2],
[3,4]]
r = 2, c = 4
Output:
[[1,2],
[3,4]]
Explanation:
There is no way to reshape a 2 * 2 matrix to a 2 * 4 matrix. So output the original matrix.

Note:
1. The height and width of the given matrix is in range [1, 100].
2. The given r and c are all positive.
 * */
class Solution {
    /*
    * solution: create new array by old array, Time:O(m*n), Space:O(r*c)
    * */
    fun matrixReshape(nums: Array<IntArray>, r: Int, c: Int): Array<IntArray> {
        val m = nums.size
        val n = nums[0].size
        if (r >= m && c >= n) {
            return nums
        }
        val result = Array(r) { IntArray(c) }
        var row = 0
        var col = 0
        //reshape new array by nums, so use m,n
        for (i in 0 until m) {
            for (j in 0 until n) {
                result[row][col] = nums[i][j]
                col++
                //check if need to create next row
                if (col == c) {
                    //create next row and reset the col
                    row++
                    col = 0
                }
            }
        }
        return result
    }
}

 

标签:Matrix,nums,Reshape,566,matrix,reshape,new,col,row
来源: https://www.cnblogs.com/johnnyzhao/p/14224007.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有