【ccf认证】201412-2 Z字形扫描（c/c++）

试题编号：	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的正整数。

 

基本思路：

根据    可得，基本扫描路线为：从左上角出发，右→左下→下→右上→右，这样的一个循环。其中，右和下每次走一格，右上和左上走到不能走为止。

需要注意的几点是：（以上图为例）

1、左下完后应该是下，但如果不能往下走（即a52的情况），那么应该直接往右走，然后是右上。

2、右上完后应该是右，但如果不能往右走（即a15的情况），那么应该直接往下走，然后是左下。

 

故，对于右和下的情况，因为只需走一格，不单独拿出来讨论。

对于左下，分为3种情况：继续往左下走，往下走再往右上走，往右走再往右上走。

对于右上，分为3种情况：继续往右上走，往右走再往左下走，往下走再往左下走。

因此，对于n*n个数，用flag记录下一个的走法，输出当前位置的数即可。

 

AC代码：

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

int main(){
    int n,flag;//右1，左下2，下3，右上4
    int a[505][505];

    while(scanf("%d",&n)!=EOF){
        memset(a,0,sizeof(a));
        flag=1;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                scanf("%d",&a[i][j]);

        int r=1,c=1;
        for(int i=0;i<n*n;i++){
            printf("%d ",a[r][c]);
            //向右走，初始位置
            if(flag==1){
                c++;
                flag=2;
//                printf("flag=%d, a[%d][%d]\n",flag,r,c);
            }
            //向左下走
            else if(flag==2){
                //继续往左下走
                if(c!=1&&r!=n){
                    r++;
                    c--;
                }
                //往右走再往右上走
                else if(r==n){
                    c++;
                    flag=4;
                }
                //往下走再往右上走
                else{
                    r++;
                    flag=4;
                }
//                printf("flag=%d, a[%d][%d]\n",flag,r,c);
            }
            //向右上走
            else if(flag==4){
                //继续往右上走
                if(r!=1&&c!=n){
                    r--;
                    c++;
                }
                //往右走再往左下走
                else if(c!=n){
                    c++;
                    flag=2;
                }
                //往下走再往左下走
                else{
                    r++;
                    flag=2;
                };
//                printf("flag=%d, a[%d][%d]\n",flag,r,c);
            }
        }
        printf("\n");

    }


    return 0;
}