數據結構 C/C++ 對角矩陣

#include<stdio.h>
int main()
{
    int i,j,N,t=0,n,m,o,p=0;
    printf("請輸入矩陣階數N：");
    scanf("%d",&N);
A:  printf("請輸入對角矩陣的帶寬n：");
    scanf("%d",&n);
    if(n%2==0)
    {
        printf("帶寬不能爲奇數，請重新輸入\n");
        goto A;
    }
    o=(n-1)/2;
    m=N*N-((N-o)*(N-o-1));
    int a[N][N],k[ m];
    printf("請以行優先順序輸入%d個元素\n",m);
    for(i=0;i<N;i++)
    {
        for(j=0;j<N;j++)
        {
            if(i>j+o||j>i+o) a[i][j]=0;
            else
            {
                scanf("%d",&a[i][j]);
                k[t]=a[i][j];
                t++;
            }
        }
    }
    t=0;
     for(i=0;i<N;i++)
    {
        for(j=0;j<N;j++)
        {
            if(i>j+o||j>i+o) printf("%-5d",p);
            else
            {
                printf("%-5d",k[t]);
                t++;
            }
        }
        printf("\n");
    }
    return 0;
}