Элементы, расположеные на главной диагонали, в первых 3 строках выше диагонали и в последних 2 строках ниже диагонали переместить в одномерный массив - C++ - Ответ 4490019

#include <stdio.h>
#include <conio.h>
 
void main()
{
 
int a[10][10];
int b[50];
int n,i,j,k=0;
clrscr();
printf("\n Vvedite ramzer");
scanf("%d",&n);
for (i=0;i<n;i++)
 for (j=0;j<n;j++)
 {
  printf("\n A[%d][%d]= ",i+1,j+1);
  scanf("%d",&a[i][j]);
 }
 printf("\n Icxodnaya matriza\n");
for (i=0;i<n;i++)
 {
 for (j=0;j<n;j++)
    printf("%6d ",a[i][j]);
  printf("\n");
 }
 
for (i=0;i<n;i++)
{
 b[i]=a[i][i];
 k++;
}
 
 for(j=1;j<n;j++)
 {
  b[k]=a[0][j];
  k++;
 }
 
for (j=2;j<n;j++)
{
 b[k]=a[1][j];
 k++;
}
 
if (n<4) printf("Nety 3 stroki nad gl.diagonal\n");
else
{
for (j=2;j<n;j++)
 {
  b[k]=a[2][j+1];
  k++;
 }
}
 
 
for(j=0;j<n;j++)
{
b[k]=a[n][j];
k++;
}
 
  for (j=0;j<n-1;j++)
  {
   b[k]=a[n-1][j];
   k++;
 }
for(i=0;i<k;i++)
printf("\nB[%d]=%d ",i,b[i]);
getch();
}