【Leetcode】566. Reshape the Matrix

題目地址：

https://leetcode.com/problems/reshape-the-matrix/

給定一個矩陣，要求將其變換爲一個$r$行$c$列的矩陣，使得兩者按行遍歷結果一樣。若無法變換，則返回原矩陣。用一維轉二維的方法，將一個一維座標和二維座標做一一對應即可。代碼如下：

public class Solution {
    public int[][] matrixReshape(int[][] nums, int r, int c) {
        int n = nums.length, m = nums[0].length;
        // 如果方格個數不一樣則不可能轉換成功，返回原數組即可
        if (n * m != r * c) {
            return nums;
        }
        
        int[][] res = new int[r][c];
        int idx = 0;
        
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
            	// idx對應的二維座標是(idx / c, idx % c)
                res[idx / c][idx % c] = nums[i][j];
                idx++;
            }
        }
        
        return res;
    }
}

時間複雜度$O(nm)$，空間$O(1)$（不包含返回結果的空間）。