leetcode 1572.矩阵对角线元素的和

1.题目要求:

给你一个正方形矩阵 mat，请你返回矩阵对角线元素的和。

请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和

2.题目代码:

int diagonalSum(int** mat, int matSize, int* matColSize){
    int i = 0;
    int col = *matColSize;
    int j = 0;
    int sum = 0;
    for(i = 0;i < matSize;i++){
        for(j = 0;j < col;j++){
            if(i == j || i + j == matSize - 1){
                sum += mat[i][j];
            }
        }
    }
    return sum;
}