AcWing寒假每日一题——Day36Z字型扫描

在图像编码的算法中，需要将一个给定的方形矩阵进行 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 字形扫描后的结果。

数据范围
$1\le n\le 500，$
矩阵元素为不超过 1000 的正整数。

输入样例：

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

分析： 判断每次行走的前一个是否越界，越界的话就更换方向

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=505;//z:0下 1上 h:0下 1右
int n,a[N][N],zx[]={
    
    1,-1},zy[]={
    
    -1,1},hx[]={
    
    1,0},hy[]={
    
    0,1};
int main(){
    
    
    cin>>n;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            cin>>a[i][j];
    int i=1,j=2,flagh=1,flagz=0;
    cout<<a[1][1]<<" ";
    if(n==1) j=1;
    else cout<<a[i][j]<<" ";
    while(i!=n||j!=n){
    
    
        int x=i+zx[flagz],y=j+zy[flagz];
        if(a[x][y]) i+=zx[flagz],j+=zy[flagz];
        else{
    
    
            if(y<1&&x<=n&&x>=1) flagz=1,flagh=0;
            else if(x<1&&y>=1&&y<=n) flagz=0,flagh=1;
            else if(x<=n&&x>=1&&y>n||y>n&&x<1) flagz=0,flagh=0;
            else if(x>n&&y>=1&&y<=n||y<1&&x>n) flagz=1,flagh=1;
            i+=hx[flagh],j+=hy[flagh];
        }
        cout<<a[i][j]<<" ";
    }
}