Matrix

int checkSame(int** matrix, int sr, int sc, int maxr, int maxc) {  // (1)
    int step = 0;
    while(1) {
        if(sr + step >= maxr) {
            break;                                                 // (2)
        }
        if(sc + step >= maxc) {
            break;                                                 // (3)
        }
        if(matrix[sr+step][sc+step] != matrix[sr][sc]) {           // (4)
            return false;
        }
        ++step;                                                    // (5)
    }
    return true;                                                   // (6)
}

bool isToeplitzMatrix(int** matrix, int matrixSize, int* matrixColSize){
    int r = matrixSize;
    int c = matrixColSize[0];
    int i;
    for(i = 0; i < c; ++i) {
        if( !checkSame(matrix, 0, i, r, c) ) {                     // (7)
            return false;  
        }        
    }
    for(i = 0; i < r; ++i) {
        if( !checkSame(matrix, i, 0, r, c) ) {                     // (8)
            return false;
        }    
    }
    return true;
}

int min(int a, int b) {
    return a < b ? a : b;
}
int max(int a, int b) {
    return a > b ? a : b;
}

int* luckyNumbers (int** matrix, int matrixSize, int* matrixColSize, int* returnSize){
    int i, j;
    int r = matrixSize;
    int c = matrixColSize[0];
    int rmin[100];
    int cmax[100];
    int *ret = (int *)malloc( sizeof(int) * r * c );

    for(i = 0; i < r; ++i) {
        rmin[i] = 1000000;
        for(j = 0; j < c; ++j) {
            rmin[i] = min(rmin[i], matrix[i][j]);             // (1)
        }
    }
    for(j = 0; j < c; ++j) {
        cmax[j] = 0;
        for(i = 0; i < r; ++i) {
            cmax[j] = max(cmax[j], matrix[i][j]);             // (2)
        }
    }
    *returnSize = 0;
    for(i = 0; i < r; ++i) {
        for(j = 0; j < c; ++j) {
            if(matrix[i][j] == rmin[i] && matrix[i][j] == cmax[j]) {
                ret[ (*returnSize)++ ] = matrix[i][j];       // (3)
            }
        }
    }
    return ret;
}

posted @ 2022-05-12 00:10  越菜越自信  阅读(483)  评论(0)    收藏  举报