C++实现CCF201412-2 Z字形扫描

问题描述

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

using namespace std;
int main()
{
	// 标记数组，visited[0]==1表右下，visited[1]==1表左上，
	int visited[2] = {0}; 
	int n;
	cin>>n;
	int arr[n][n];  // 记录原始数据 
	// 原始数据初始化 
	for(int i = 0; i < n; i++)
	{
		for(int j = 0; j < n; j++)
			cin>>arr[i][j];
	} 
	// 打印arr[0][0]元素 
	int row = 0, col = 1;
	cout<<arr[0][0] << " "; 
	visited[0] = 1;  // 右下 
	
	for(int i = 1; i < n*n; i++)
	{
		// 打印结果 
		cout<<arr[row][col];
		if(i != n*n-1)
			cout<<" ";
		else
			cout<<endl;
		
		if(visited[0] == 1)
		{ // 右下，注意有3种情况 
			if(row == n-1) // 注意右下这里应该先讨论行数，然后讨论列数 
			{ // 先水平向右，然后左上 
				col++;
				visited[0] = 0;
				visited[1] = 1; 
			}
			else if(col == 0)
			{ // 先竖直向下，然后左上 
				row++; 
				visited[0] = 0;
				visited[1] = 1; 
			}
			
			else
			{ // 继续右下 
				row++;
				col--;
			}
		}
		else if(visited[1] == 1)
		{ // 左上，注意有3种情况 
			if(col == n-1) // 注意先讨论列数，然后讨论行数 
			{// 先竖直向下，然后右下 
				row++;
				visited[1] = 0;
				visited[0] = 1; 
			} 
			else if(row == 0)
			{// 先水平向右，然后右下 
				col++;
				visited[1] = 0;
				visited[0] = 1; 
			}
			else
			{// 继续左上 
				row--;
				col++; 
			}
		}
	} 
	 
	return 0;
}