CSP201412-2：Z字形扫描

引言：CSP（http://www.cspro.org/lead/application/ccf/login.jsp）是由中国计算机学会（CCF）发起的"计算机职业资格认证"考试，针对计算机软件开发、软件测试、信息管理等领域的专业人士进行能力认证。认证对象是从事或将要从事IT领域专业技术与技术管理人员，以及高校招考研究生的复试对象。

• 问题描述

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

• 源代码

# include <stdio.h> # include <stdlib.h> # include <memory.h> int main(void) { int n; //矩阵规模 scanf("%d", &n); //创建矩阵（二维数组）并分配空间 int **pInput = (int **)malloc(sizeof(int *) * n); for (int i = 0; i < n; i++) { pInput[i] = (int *)malloc(sizeof(int) * n); memset(pInput[i], 0, sizeof(int) * n); } //创建输出数组 int *pOutput = (int *)malloc(sizeof(int) * n * n); memset(pOutput, 0, sizeof(int) * n * n); //矩阵输入 for (int i = 0; i < n; i++) { for (int j = 0; j < n; j++) { scanf("%d", &pInput[i][j]); } } int j = 0; int k = 0; int state = 1; pOutput[0] = pInput[0][0]; for (int i = 1; i < n*n; i++) { switch (state) { case 1:{ //向右 k += 1; pOutput[i] = pInput[j][k]; if (j == 0) { state = 2; } else { state = 4; } break; } case 2:{ //向左下 j += 1; k -= 1; pOutput[i] = pInput[j][k]; if (k == 0 || j == n - 1) { if (j == n - 1) { state = 1; } else { state = 3; } } else { state = 2; } break; } case 3:{ //向下 j += 1; pOutput[i] = pInput[j][k]; if (k == 0) { state = 4; } else { state = 2; } break; } case 4:{ //向右上 j -= 1; k += 1; pOutput[i] = pInput[j][k]; if (j == 0 || k == n - 1) { if (k == n - 1) { state = 3; } else { state = 1; } } else { state = 4; } break; } default:{ break; } } } for (int i = 0; i < n * n - 1; i++) { printf("%d ", pOutput[i]); } printf("%d\n", pOutput[n * n - 1]); //释放矩阵（二位数组）空间 for (int i = 0; i < n; i++) { free(pInput[i]); } free(pInput); //释放输出数组空间 free(pOutput); return 0; }