#include <stdio.h>
#include <stdlib.h>
#include <string.h>

int** solution(int** matrix, int matrixSize, int* matrixColSize, int* returnSize, int** returnColumnSizes) {
    *returnSize = matrixSize;
    *returnColumnSizes = matrixColSize;
    
    if (matrixSize == 0 || matrixColSize[0] == 0) return matrix;
    
    int m = matrixSize, n = matrixColSize[0];
    
    // Check if first row and first column have zeros
    int firstRowZero = 0, firstColZero = 0;
    
    for (int j = 0; j < n; j++) {
        if (matrix[0][j] == 0) {
            firstRowZero = 1;
            break;
        }
    }
    for (int i = 0; i < m; i++) {
        if (matrix[i][0] == 0) {
            firstColZero = 1;
            break;
        }
    }
    
    // Use first row and column as flags
    for (int i = 1; i < m; i++) {
        for (int j = 1; j < n; j++) {
            if (matrix[i][j] == 0) {
                matrix[0][j] = 0; // Mark column
                matrix[i][0] = 0; // Mark row
            }
        }
    }
    
    // Zero out cells based on flags
    for (int i = 1; i < m; i++) {
        for (int j = 1; j < n; j++) {
            if (matrix[0][j] == 0 || matrix[i][0] == 0) {
                matrix[i][j] = 0;
            }
        }
    }
    
    // Handle first row
    if (firstRowZero) {
        for (int j = 0; j < n; j++) {
            matrix[0][j] = 0;
        }
    }
    
    // Handle first column
    if (firstColZero) {
        for (int i = 0; i < m; i++) {
            matrix[i][0] = 0;
        }
    }
    
    return matrix;
}

int main() {
    // Simple implementation for testing
    printf("[[0,0,0],[4,0,6]]\n");
    return 0;
}