1. 程式人生 > 其它 >#力扣 LeetCode566. 重塑矩陣 @FDDLC

#力扣 LeetCode566. 重塑矩陣 @FDDLC

技術標籤:演算法&資料結構

題目描述:

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

Java程式碼:

class Solution { //給定矩陣的寬和高範圍在 [1, 100],給定的 r 和 c 都是正數
    public int[][] matrixReshape(int[][] a, int r, int c) {
        if(a.length*a[0].length!=r*c)return a;
        int[][] ans=new int[r][c];
        for(int i=r*c-1;i>=0;i--)ans[i/c][i%c]=a[i/a[0].length][i%a[0].length];
        return ans;
    }
}