[CCF] Z字形扫描

CCF Z字形扫描

感觉和LeetCode中的ZigZag还是有一些不一样的。

题目描述

在图像编码的算法中，需要将一个给定的方形矩阵进行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字形扫描的结果。

输入格式

1. 输入的第一行包含一个整数n，表示矩阵的大小。
2. 输入的第二行到第n+1行每行包含n个正整数，由空格分隔，表示给定的矩阵。

输出格式

1. 输出一行，包含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的正整数。

代码

这个题还是比较有意思的，代码如下：

# include <stdio.h>
# define MAX 510

int n;
int i,j;
int m[MAX][MAX];
typedef struct{
  int x,y;
} pos; // define a poit struct

void show(pos p)
{
  printf("%d ",m[p.x][p.y]);
}

int main()
{
  pos current = {0,0};
  scanf("%d",&n);
  for(i = 0;i < n; ++i){
    for(j = 0; j < n;++j){
      scanf("%d", &m[i][j]);
    }
  }//end of for i
  // data collect end
  show(current);
  while(1){
    if (current.x == current.y && current.x == n-1){
      // the end condition
      break;
    }
    if (current.x > n - 1 || current.y > n-1 ) break;
    // right or down
    if (current.y == n-1){
      // down
      current.x += 1;
      show(current);
    }else{
      // right
      current.y += 1;
      show(current);
    }
    // then go left down
    while(current.x < n - 1 && current.y > 0){
      current.x +=1;
      current.y -=1;
      show(current);
    }
    // go down or right
    if (current.x == n - 1 && current.y < n-1){
      // can not go down then right
      current.y += 1;
      show(current);
    }else{
      // go down
      current.x += 1;
      show(current);
    }
    // up and right
    while(current.x > 0 && current.y < n -1){
      current.x -=1;
      current.y +=1;
      show(current);
    }
  }
  printf("\n");
  return 0;
}