在图像编码的算法中,需要将一个给定的方形矩阵进行 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>
using namespace std;
const int N = 510;
int a[N][N];
int main()
{int n;cin>>n;for (int i = 1;i<=n;i++)for (int j = 1;j<=n;j++)cin>>a[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<<a[j][i-j]<<" ";}}else{for(int j = i-1;j;j--){if (j>=1 && j<=n && i-j >= 1&& i-j <=n)cout<<a[j][i-j]<<" ";}}}return 0;
}