求一个 3*3 矩阵对角线元素之和

int main() {
  int a[3][3],sum=0;
 printf("输入留6个数:");
 for (int i = 0;i < 3;i++) {
     for (int j = 0;j < 3;j++) {
        scanf_s("%d", &a[i][j]);
    }
   }
   for (int i = 0;i < 3;i++) {
       for (int j = 0;j < 3;j++) {
            if (i == j) {
               sum += a[i][j];
           }
       }
    }
    printf("%d", sum);
}