LeetCode 566. 重塑矩阵

示例 1:
输入: 
nums = 
[[1,2],
 [3,4]]
r = 1, c = 4
输出: 
[[1,2,3,4]]
解释:
行遍历nums的结果是 [1,2,3,4]。
新的矩阵是 1 * 4 矩阵, 用之前的元素值一行一行填充新矩阵。

示例 2:
输入: 
nums = 
[[1,2],
 [3,4]]
r = 2, c = 4
输出: 
[[1,2],
 [3,4]]
解释:
没有办法将 2 * 2 矩阵转化为 2 * 4 矩阵。 所以输出原矩阵。

注意：
给定矩阵的宽和高范围在 [1, 100]。
给定的 r 和 c 都是正数。

class Solution {
public:
    vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
        int r0 = nums.size(), c0 = nums[0].size();
        if(r*c != r0*c0)
        	return nums;
        vector<int> row;
        vector<vector<int>> matrix;
        int i, j, k = 0;
        for(i = 0; i < r0; ++i)
        {
        	for(j = 0; j < c0; ++j)
        	{
        		row.push_back(nums[i][j]);
        		k++;
        		if(k == c)
        		{
        			matrix.push_back(row);
        			k = 0;
        			row.clear();
        		}
        	}
        }
        return matrix;
    }
};

class Solution {
public:
    vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c){
        int r0 = nums.size(), c0 = nums[0].size();
        if(r*c != r0*c0)
        	return nums;
        vector<vector<int>> matrix(r,vector<int>(c));
        int i, j, x = 0, y = 0;
        for(i = 0; i < r0; ++i)
        {
        	for(j = 0; j < c0; ++j)
        	{
        		matrix[x][y++] = nums[i][j];
                if(y==c)
                {
                    x++;
                    y=0;
                }
        	}
        }
        return matrix;
    }
};