【acwing 寒假每日一题（入门组）】day37 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

思路

其实题意就是让我们遍历所有对角线，但是给了一点约束，要求先左下->右上，再右上->左下的z字形遍历
预备知识:对角线的性质，这个在八皇后问题中用到过，就是对角线一定满足x+y=c
我们存储的时候是从(0,0)开始的 那么显然 第i条对角线的和是i,如第一条对角线上元素只有(0,0) x+y=0
n * n的矩阵，有2*n-1条对角线，我们每次遍历两条方向不同的对角线，然后遍历n次就好了

代码

#include<bits/stdc++.h>

using namespace std;

const int N=510;

int n;
int a[N][N];

int main()
{
    cin>>n;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++) cin>>a[i][j];
    int cur=0; //第0条对角线
    for(int k=0;k<n;k++)
    {
        //左下->右上
        int i=min(cur,n-1),j=cur-i;
        while(i>=0 && j<n)
        {
            cout<<a[i][j]<<' ';
            i--,j++;
        }
        cur++;
         //右上->左下  因为操作是一摸一样的 只是含义不同了而已
        i=min(cur,n-1),j=cur-i;
        while(i>=0 && j<n)
        {
            cout<<a[j][i]<<' ';
            i--,j++;
        }
        cur++;
    }
    return 0;
}