No.83 - LeetCode566

ShellDawn

于 2019-08-29 06:14:02 发布

阅读量94

点赞数

CC 4.0 BY-SA版权

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

本文链接：https://blog.youkuaiyun.com/ShellDawn/article/details/100129691

leetcode 专栏收录该内容

251 篇文章

订阅专栏

本文介绍了一个用于矩阵重塑的算法，该算法能够将一个给定的二维矩阵转换为另一个指定维度的矩阵，前提是两个矩阵的元素总数相等。通过遍历原始矩阵并将元素重新分配到新矩阵中，实现了矩阵形状的改变。

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

class Solution {
public:
    int N,M;
    int move(vector<vector<int>>& nums,int& x,int& y){
        int t = nums[x][y];
        y++;
        if(y == M){
            y = 0;
            x++;
        }
        return t;
    }
    vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
        N = nums.size();
        M = nums[0].size();
        if(N*M == r*c){
            vector<vector<int>> ans(r,vector<int>(c,0));
            int x,y;
            x = y = 0;
            for(int i=0;i<r;i++){
                for(int j=0;j<c;j++){
                    ans[i][j] = move(nums,x,y);
                }
            }
            return ans;
        }
        return nums;
    }
};