关于CCF中的Z字形扫描

之前学习了折叠方阵并试着做出了蛇形折叠，

最近在想螺旋折叠应怎么做。

今天看到了一个CCF 真题，感觉和之前类型差不多，就试着写一写。

题目如下：在图像编码的算法中，需要将一个给定的方形矩阵进行Z字形扫描(Zigzag Scan)。
给定一个n×n的矩阵，Z字形扫描的过程如下图所示
对于下面的4×4的矩阵              1 5 3 9 
                                                 3 7 5 6   
9 4 6 4   
7 3 1 3    
对其进行Z字形扫描后得到长度为16的序列：   1 5 3 9 7 3 9 5 4 7 3 6 6 4 1 3    
请实现一个Z字形扫描的程序，给定一个n×n的矩阵，输出对这个矩阵进行Z字形扫描的结果。
输入格式    输入的第一行包含一个整数n，表示矩阵的大小。    
输入的第二行到第n+1行每行包含n个正整数，由空格分隔，表示给定的矩阵。 
输出格式    输出一行，包含n×n个整数，由空格分隔，表示输入的矩阵经过Z字形扫描后的结果。

代码如下：

#include<iostream>using namespace std;int main(){int n,i,j,k=0;cin>>n;int a[n][n],b[n*n];for(i=0;i<n;i++){for(j=0;j<n;j++){cin>>a[i][j];}}for(i=1;i<n;i++)                //上半部三角形 {for(j=0;j<=i;j++){k++;if(i%2==1)              //分奇数偶数斜向进行讨论 {b[k]=a[j][i-j];}else{b[k]=a[i-j][j];}}}for(i=n;i<2*n-2;i++)                  //下半部三角形 {for(j=n-1;j>=1;j--){k++;    if(i%2==0)    {    b[k]=a[j][i-j];}else{b[k]=a[i-j][j];}}}b[0]=a[0][0];b[n*n-1]=a[n-1][n-1];for(i=0;i<n*n;i++){cout<<b[i]<<' ';}}