566. Reshape the Matrix

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