leetcode 566:重塑矩阵

题目：

 

算法思想：先判断不同矩阵的元素个数是否和原矩阵相同，然后遍历构造即可。

 

代码 ：

vector<vector<int> > matrixReshape(vector<vector<int> >& nums, int r, int c) {
        if(r * c != nums.size() * nums[0].size())
                return nums;
        vector<vector<int > > result(r);
        vector<vector<int > > ::iterator iter;
        vector<int> tmp;
        int row = 0;
        int column = 0;
        for(int i = 0;i < nums.size();i++)
        {
            for(int j = 0;j < nums[0].size();j++)
            {
                if(column < c)
                {
                    result[row].push_back(nums[i][j]);
                    column++;
                }
                else
                {
                    row++;
                    column = 1;
                    result[row].push_back(nums[i][j]);
                }
            }
        }
        return result;
}