力扣-566题重塑矩阵（C++）- vector二维数组获取行数和列数

女爷i 2022-08-30 00:51 210阅读 0赞

题目链接：[https://leetcode-cn.com/problems/reshape-the-matrix/][https_leetcode-cn.com_problems_reshape-the-matrix]  
题目如下：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwNDY3Njcw_size_16_color_FFFFFF_t_70]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwNDY3Njcw_size_16_color_FFFFFF_t_70 1]

class Solution {
        
    public:
        vector<vector<int>> matrixReshape(vector<vector<int>>& mat, int r, int c) {
        
            //tip:
            //vector二维数组获取行数和列数
            //1、行数：mat.size()
            //2、列数：mat[0].size()
            vector<vector<int>> result;
            int row=mat.size(),col=mat[0].size();
            
            if(row*col!=r*c) return mat;
            
            int row1=0,col1=0;
            for(int i=0;i<r;i++){
        
                vector<int> vtr;
    
                for(int j=0;j<c;j++){
        
                    if(col1==col) {
        col1=0;row1++;}
    
                    int value=mat[row1][col1++];
                    vtr.push_back(value);
                }
    
                result.push_back(vtr);
            }
    
            return result;
        }
    };

[https_leetcode-cn.com_problems_reshape-the-matrix]: https://leetcode-cn.com/problems/reshape-the-matrix/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwNDY3Njcw_size_16_color_FFFFFF_t_70]: /images/20220829/3aff0a6e409b462a94e297e179275936.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQwNDY3Njcw_size_16_color_FFFFFF_t_70 1]: /images/20220829/e2d61648ee994521a86b574ac25a5bca.png