201412-2_Z字形扫描

试题编号： 201412-2
试题名称： Z字形扫描
时间限制： 2.0s
内存限制： 256.0MB
问题描述
　　在图像编码的算法中，需要将一个给定的方形矩阵进行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字形扫描后的结果。
样例输入
4
1 5 3 9
3 7 5 6
9 4 6 4
7 3 1 3
样例输出
1 5 3 9 7 3 9 5 4 7 3 6 6 4 1 3
评测用例规模与约定
　　1≤n≤500，矩阵元素为不超过1000的正整数。

int m[510][510];
int main(){
    int n;
    cin>>n;
    for (int i = 1; i <= n; ++i) {//边界为0
        for (int j = 1; j <= n; ++j) {
            cin>>m[i][j];
        }
    }
    
    int i = 1,j = 1;
    cout<<m[i][j]<<" ";
    //将扫看成每四步一个周期（直，斜，直，斜）
    while (i+j != 2*n){//终点时i+j=2n
    	//第一步
        if(j < n)//如果没到右边界，则向右，否则向下
            ++j;
        else ++i;
        //第二步
        while (m[i][j]){//走斜线，直到到达边界，退出循环时已在边界外
            cout<<m[i][j]<<" ";
            ++i;
            --j;
        }
        --i;++j;//回退到边界上
        
        if(i < n)//没到下边界，则向下，否则向右
            ++i;
        else ++j;
        
        while (m[i][j]){//斜线
            cout<<m[i][j]<<" ";
            --i;
            ++j;
        }
        ++i;--j;
        
      }
    }