No.83 - LeetCode566

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

摘要生成于 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;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值