leetCode题解之Reshape the Matrix

1、题目描述

2、分析

使用了一个队列。

 

3、代码

vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
        
        if( nums.size() * nums[0].size()  !=  r * c )
            return nums;
        vector<vector<int>> ans;
        queue<int> q;
        for( size_t i = 0; i < nums.size() ;i++ )
            for( size_t j =0; j < nums[0].size() ;j++)
                q.push( nums[i][j] );
        
        vector<int> col;
        while( !q.empty()  )
        {
            col.push_back( q.front() );
            q.pop();
            if( col.size() == c )
            {
                ans.push_back( col );
                col.clear();
            }
        }
        
        return ans;
        
    }