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

#include<iostream>
#include<cmath>
using namespace std;
int n,x,y,a[505][505];
void up(){
	while(x>=0&&y<=n-1){
		//cout<<"x="<<x<<"  y="<<y<<" ";
		printf("%d ",a[x][y]);x--;y++;//要注意函数执行后x和y要恢复原状！
	}
}
void down(){
	while(y>=0&&x<=n-1){		
		//cout<<"x="<<x<<"  y="<<y<<" ";
		printf("%d ",a[x][y]);y--;x++;		
	}
}
int main(){
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++)
		    scanf("%d",&a[i][j]);
	}
	/*for(int i=0;i<n;i++){
		for(int j=0;j<n;j++)
		    printf("%d",a[i][j]);
	}*/
	
	if(n==1){
		printf("%d",a[0][0]);
		return 0;
	}
	if(n==2){
		printf("%d %d %d %d",a[0][0],a[0][1],a[1][0],a[1][1]);
		return 0;
	}
	
	int endcol=n&1==1?n-1:n-2;
	x=0;y=0;
	//cout<<"endcol "<<endcol<<endl;
	printf("%d ",a[0][0]);
	while(y!=endcol){
		y++;
		down();y++;x--;
		x++;
		up();x++;y--;
	}
	if(n&1){
		x++;
	}
	else{
		y++;
	}
	down();y++;x--;
	y++;
	while(x!=n-1||y!=n-1){
		up();x++;y--;
		x++;
		down();y++;x--;
		y++;
	}
	printf("%d ",a[n-1][n-1]);
	return 0;
}

错误：up() 和down()函数执行完之后会使得x和y超出边界，需要恢复（因为多执行了一次上移或下移）。以及还有很多细节需要注意