对角矩阵转化为一维数组

对角矩阵转为一维数组

 

 

代码：

 #include <stdio.h>
 #include <math.h>
 #define dataType int
 #define n 4
 
int main()
 {
     dataType A[n][n]={
          {1,4,0,0},
         {3,4,1,0},
         {0,2,3,4},
         {0,0,1,3}};
      dataType B[10];
     int i,j,k;

     for(i=0;i<n;i++)
      {
         for(j=0;j<n;j++)
         {
           if(abs(i-j)<2)
              {
                 k=2*i+j;
                B[k]=A[i][j];
             }
        }
   }
for(i=0;i<10;i++)
     {
        printf("%3d",B[i]);
     }
 
    printf("/n");
      getchar();
     return 0;
 }