二维vector的一些细节(用于实现LeetCode566.重塑矩阵)

﹏ヽ暗。殇╰゛Y 2022-12-17 13:58 160阅读 0赞

一、二维vector的一些细节  
①定义和初始化

vector<vector<int> > V(r, vector<int>(c));

> r表示V中有r个vector，每个vector有c个元素，故为r \* c的二维矩阵。  
> ②获取行数和列数

row = V.size();
    col = V[0].size();

二、LeetCode566.重塑矩阵  
1.题目  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl8zNzQ3NzAwOQ_size_16_color_FFFFFF_t_70_pic_center]  
2.AC代码  
关键 : 第i个数对应的(row\_id, col\_id)，和**磁盘管理的位示图法**相似。

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

![在这里插入图片描述][20201023215909733.png_pic_center]  
3.大佬代码(暂未找到)

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl8zNzQ3NzAwOQ_size_16_color_FFFFFF_t_70_pic_center]: /images/20221122/406771f2b310466893e9c446bc2bfaf8.png
[20201023215909733.png_pic_center]: /images/20221122/2cc5d54e53604cf08e12b37aa4f3a4e5.png