重塑二维矩阵

Input:
nums =
[[1,2],
[3,4]]
r = 1, c = 4
Output:
[[1,2,3,4]]
Explanation:
The row-traversing of nums is [1,2,3,4]. The new reshaped matrix is a 1 * 4 matrix, fill it row by row by using the previous list.

#include<iostream>
#include<unistd.h>
#include<vector>
#include<stack>
#include<queue>
using namespace std;


vector<vector<int>> reshapeMatrix(vector<vector<int>>& input, int r , int c)
{
	int m = input.size();
	int n = input[0].size();

	if(m*n < r*c) return input;

	vector<vector<int>> result(r , vector<int>(c));

	for(int i =0; i< r*c;++i)
	{
		result[i/c][i%c] = input[i/n][i%n];
	}

	return result;
}

int main()
{
	vector<vector<int>> v = {{1,2},
							{3,4}};
	vector<vector<int>> ret = reshapeMatrix(v , 4,1);
	for(auto& v:ret)
	{
		cout<<v.size()<<endl;
	}

}