/**
* Return an array of arrays of size *returnSize.
* The sizes of the arrays are returned as *returnColumnSizes array.
* Note: Both returned array and *columnSizes array must be malloced, assume caller calls free().
*/
int** matrixReshape(int** nums, int numsSize, int* numsColSize, int r, int c, int* returnSize, int** returnColumnSizes){
if(r*c != numsSize*(*numsColSize)){
*returnSize = numsSize;
*returnColumnSizes = numsColSize;
return nums;
}
int** returnMat = (int **)malloc(r*sizeof(int *));
for(int i=0;i<r;i++){
returnMat[i] = (int *)malloc(c*sizeof(int));
}
*returnSize = r;
(*returnColumnSizes) = (int *)malloc(r*sizeof(int));
for(int i=0;i<r;i++){
(*returnColumnSizes)[i] = c;
}
int line_index = 0;
for(int row=0;row<numsSize;row++){
for(int col=0;col<(*numsColSize);col++){
line_index = (*numsColSize)*row+col;
returnMat[line_index/c][line_index%c] = nums[row][col];
}
}
return returnMat;
}
Leetcode566,高阶指针形参的处理
最新推荐文章于 2024-03-23 18:38:56 发布