566. 重塑矩阵【ans[x / c][x % c] = nums[x / n][x % n]】

LIZHUOLONG1

已于 2023-03-22 22:43:32 修改

阅读量106

点赞数

分类专栏： LeetCode刷题文章标签：矩阵 c语言算法

于 2023-01-07 20:30:51 首次发布

本文链接：https://blog.csdn.net/LIZHUOLONG1/article/details/128594843

版权

LeetCode刷题专栏收录该内容

305 篇文章 4 订阅

订阅专栏

566. 重塑矩阵

在这里插入图片描述

C代码1:

在这里插入图片描述

int** matrixReshape(int** nums, int numsSize, int* numsColSize, int r, int c, int* returnSize, int** returnColumnSizes) {
    int m = numsSize;                    // 将数组作为指针传递过来，这个是复制还是引用？--引用
    int n = numsColSize[0];
    if (m * n != r * c) {
        *returnSize = numsSize;  // 行
        *returnColumnSizes = numsColSize;
        return nums;
    }
    *returnSize = r;  // 行
    *returnColumnSizes = malloc(sizeof(int) * r); // 创建数组行（数组存值）
    int** ans = malloc(sizeof(int*) * r);         // 返回体，创建数组行（存指针） 二者第一步都是创建数组，区别存放值/地址

    for (int i = 0; i < r; i++) {
        (*returnColumnSizes)[i] = c;  // 数组存列值
        ans[i] = malloc(sizeof(int) * c); // 数组存数组（指针）
    }
    for (int x = 0; x < m * n; ++x) {
        ans[x / c][x % c] = nums[x / n][x % n];
    }
    return ans;
}

Java代码1：

public int[][] matrixReshape(int[][] nums, int r, int c) {
    int m = nums.length;
    int n = nums[0].length;
    if (m * n != r * c) {
        return nums;
    }
    int[][] ans = new int[r][c];
    for (int x = 0; x < m * n; ++x) {  // 没法做两个for循环
        ans[x / c][x % c] = nums[x / n][x % n];  // 对列整除、求余
    }
    return ans;
}