Leetcode566,高階指針形參的處理

Leetcode566

/**
 * 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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章