LeetCode -- 566.重塑矩阵

链接：重塑矩阵

描述：给定一个范围在 1 ≤ a[i] ≤ n ( n = 数组大小 ) 的 整型数组，数组中的元素一些出现了两次，另一些只出现一次。

重构后的矩阵需要将原始矩阵的所有元素以相同的行遍历顺序填充。

要求：如果具有给定参数的reshape操作是可行且合理的，则输出新的重塑矩阵；否则，输出原始矩阵。

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

思路：很清楚的记得当初学习数组的时候，当老师讲到二维数组，他问我们二维数组行和列哪个更重要，答案是：列数。这样的话确定元素个数和列数，格式就唯一。
所以看到这个题目的时候我想到的就是把原二维数组一维化，通过列数和%，/进行枚举。

class Solution {
public:
    vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
		int m = nums.size(); //原矩阵个数 
		int n = nums[0].size(); //原矩阵元素个数 
		vector<vector<int>> ans(r, vector<int>(c));
		if(m * n == r * c) //唯有元素总数相同才可以转换 
		{
			for(int i = 0 ; i < m*n ; i++)
			{
				ans[i / c][i % c] = nums[i / n][i % n];	
			}	
		} 
		else //否则原样输出 
		{
			ans = nums;
		}
		return ans;
    }
};

快速笔记：vector创建二维数组，形式如

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

敦促一下：明天做完最长回文子串,把blog也赶出来。