[LeetCode]566. Reshape the Matrix

charon____

于 2017-05-01 20:18:07 发布

阅读量933

点赞数 1

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/Lcharon/article/details/71074829

leetcode 专栏收录该内容

184 篇文章

订阅专栏

本文介绍了一个将矩阵重塑为新尺寸的算法实现，并提供了详细的代码示例。通过计算一维索引，该算法能够有效地更新目标矩阵中对应位置的数值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

[LeetCode]566. Reshape the Matrix

题目描述

这里写图片描述

思路

计算下标，根据下标更新对应位置的数值

代码

#include <iostream>
#include <vector>

using namespace std;

class Solution {
public:
    vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
        int row = nums.size(), col = nums[0].size();
        if (row * col != r * c)
            return nums;
        vector<vector<int>> res(r, vector<int>(c));
        for (int i = 0; i < row; ++i) {
            for (int j = 0; j < col; ++j) {
                int to1dim = i * col + j,
                    newI = to1dim / c,
                    newJ = to1dim % c;
                res[newI][newJ] = nums[i][j];
            }
        }
        return res;
    }

    void printMatrix(vector<vector<int>> nums) {
        for (auto p : nums) {
            for (auto q : p)
                cout << q << " ";
            cout << endl;
        }
    }
};

int main() {
    vector<vector<int>> nums = { {1,2},{3,4} }, res;
    Solution s;
    res = s.matrixReshape(nums, 2, 1);
    s.printMatrix(nums);
    s.printMatrix(res);
    system("pause");
    return 0;
}