マトリックスをZ形式で印刷するC++プログラム？

ここでは、行列要素をZ形式で印刷する方法を説明します。したがって、配列が次のようになっている場合-

5 8 7 1
2 3 6 4
1 7 8 9
4 8 1 5

次に、次のように印刷されます：5、8、7、1、6、7、4、8、1、5

アルゴリズム

printMatrixZ（mat）

Begin
   print the first row
   i := 1, j := n-2
   while i < n and j >= 0, do
      print mat[i, j]
      i := i + 1, j := j - 1
   done
   print the last row
End

例

#include<iostream>
#define MAX 4
using namespace std;
void printMatrixZ(int mat[][MAX], int n){
   for(int i = 0; i<n; i++){
      cout << mat[0][i] << " ";
   }
   int i = 1, j = n-2;
   while(i < n && j >= 0){
      cout << mat[i][j] << " ";
      i++;
      j--;
   }
   for(int i = 1; i<n; i++){
      cout << mat[n-1][i] << " ";
   }
}
main() {
   int matrix[][MAX] = {{5, 8, 7, 1},
      {2, 3, 6, 4},
      {1, 7, 8, 9},
      {4, 8, 1, 5}
   };
   printMatrixZ(matrix, 4);
}

出力

5 8 7 1 6 7 4 8 1 5