Сума головної діагоналі

Вихідний код

static int diagonalSum(int[][] a, int n) {
    int sum = 0;
    for (int i = 0; i < n; i++) {
        sum += a[i][i];
    }
    return sum;
}