找对角线元素下标之间的关系（转为一维）

#include<stdio.h>
int main(){
    int N;
    int matrix;
    int sum=0;
    scanf("%d", &N);
    for(int i=0; i<N*N; i++){
        scanf("%d", &matrix);
        if(i==0||i%(N-1)==0||i%(N+1)==0){
            sum += matrix;
        }
    }
    printf("%d\n", sum);
    return 0;
}