1.题目
Reshape the Matrix
例1:
Input:
nums = [[1,2], [3,4]]
r = 1, c = 4
Output:
[[1,2,3,4]]
Explanation:
元素横向展开,2*2的矩阵转为1*4的矩阵
例 2:
Input:
nums = [[1,2], [3,4]]
r = 2, c = 4
Output:
[[1,2], [3,4]]
Explanation:
2*2的矩阵无法转为2*4的矩阵,所以输出原矩阵。
2.解法
(1)origin_r * origin_c 的矩阵 reshape为 r*c的矩阵,需要满足:
origin_r * origin_c=r*c
(2)元素位置对应的关系
如果将矩阵横向展开为一维数组,元素个数为n=origin_r * origin_c
在元素在一维数组中对应的位置i:
原矩阵位置[i/origin_c,i%origin_c]
新矩阵位置[i/c,i%c]
3.代码
c++
vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
//原矩阵的行,列
int origin_r = nums.size();
int origin_c = nums[0].size();
int n = origin_c*origin_r;//元素个数
if (n == r*c) {
//新矩阵
vector<vector<int>> newMaxtrix(r, vector<int>(c, 0));
for (int i = 0; i < n; i++)
newMaxtrix[i / c][i%c] = nums[i / origin_c][i%origin_c];
return newMaxtrix;
}
else
return nums;
}
python
def matrixReshape(self, nums, r, c):
origin_r=len(nums)#行
origin_c=len(nums[0])#列
if origin_c*origin_r==r*c:#可以reshape
temp=[num for row in nums for num in row]#展开成一维数组
newMatrix=[[0 for j in xrange(c)] for i in xrange(r)]
for i in xrange(r):
for j in xrange(c):
newMatrix[i][j]=temp[i*c+j]
return newMatrix
else:#无法reshape
return nums