寒假每日一题题解(2.16)Z字形扫描（模拟）

Z字形扫描（模拟）

在图像编码的算法中，需要将一个给定的方形矩阵进行 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×n×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 <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 510;

int n;
int q[N][N];

int main()
{
    cin >> n;
    for (int i = 1; i <= n; i ++ )
        for (int j = 1; j <= n; j ++ )
            cin >> q[i][j];

    for (int i = 2; i <= n * 2; i ++ )
    {
        if (i % 2)
        {
            for (int j = 1; j <= i; j ++ )
                if (j >= 1 && j <= n && i - j >= 1 && i - j <= n)
                    cout << q[j][i - j] << ' ';
        }
        else
        {
            for (int j = i; j; j -- )
                if (j >= 1 && j <= n && i - j >= 1 && i - j <= n)
                    cout << q[j][i - j] << ' ';
        }
    }
    return 0;
}