CCF-CSP201412-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的正整数。

AC代码：
  

#include<iostream>
#include<sstream>
#include<algorithm>
#include<string>
#include<cstring>
#include<iomanip>
#include<vector>
#include<cmath>
#include<ctime>
#include<stack>
#include<queue>
#include<map>
#define mem(a,b) memset(a,b,sizeof(a))
#define e 2.71828182
#define Pi 3.141592654
using namespace std;
int main()
{
    int n,a[502][502];
    cin>>n;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            cin>>a[i][j];
            
    for(int i=1;i<=n;i++)
    {
        if(i&1)
            for(int p=i,q=1;p>=1&&q<=i;p--,q++)
                cout<<a[p][q]<<' ';
        else 
            for(int p=1,q=i;p<=i&&q>=1;p++,q--)
                cout<<a[p][q]<<' ';
    }
    
    for(int i=2;i<=n;i++)
    {
        if(n&1)
        {
            if(i&1)
                for(int p=n,q=i;p>=i&&q<=n;p--,q++)
                    cout<<a[p][q]<<' ';
                else for(int p=i,q=n;p<=n&&q>=i;p++,q--)
                    cout<<a[p][q]<<' ';
        }
        else 
        {
            if(i&1) 
            for(int p=i,q=n;p<=n&&q>=i;p++,q--)
                    cout<<a[p][q]<<' ';
                else for(int p=n,q=i;p>=i&&q<=n;p--,q++)
                    cout<<a[p][q]<<' ';
        }
            
    }
}