重塑矩陣[leetcode]
阿新 • • 發佈:2018-12-09
題目描述
在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 矩陣。 所以輸出原矩陣。
注意:
- 給定矩陣的寬和高範圍在 [1, 100]。
- 給定的 r 和 c 都是正數。
分析
對於這種二維陣列大小重新非配的問題的關鍵就是對應位置的座標轉換,最直接的辦法就是先把原陣列拉直,變成一條直線,然後再組成新的陣列。所以這道題我們先判斷給定陣列是否能重塑成給定的大小,就是看兩者的元素總數是否相同,直接行數乘以列數即可,然後我們新建一個目標大小的陣列,並開始遍歷,對於每個位置,我們先轉為拉直後的一維座標,然後在算出在原陣列中的對應位置賦值過來即可。
貼出程式碼
class Solution {
public int[][] matrixReshape(int[][] nums, int r, int c) {
int m = nums.length, n = nums[0].length;
if (r * c != m * n) {
return nums;
}
int[][] res;
res = new int[r][c];
for (int i = 0; i < r; ++i){
for ( int j = 0; j < c; ++j){
int k = i * c + j;
res[i][j] = nums[k / n][k % n];
}
}
return res;
}
}