CCF CSP 编程题目和解答-----试题名称：Z字形扫描-------201412-2

问题描述

试题编号：	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<algorithm>
#include<vector>

using namespace std;



int main()
{
int n;
int flag = 4;//记录方向，1表示向右，2表示左下，3表示下，4表示右上
int sqr[500][500] = { 0 };
cin >> n ;

for (int i = 0; i < n; i++)
{
for (int j = 0; j < n; j++)
{
cin >> sqr[i][j];
}
}
vector<int> result ;
result.push_back(sqr[0][0]);//第一个点放入结果集合
;
int x = 0;
int y = 0;
while (!(x == n - 1 && y == n - 1))
{
if (flag == 1) //如果上一步是向右
{
if (x + 1 < n&& y - 1 >=0)//可以左下
flag = 2;//当前步向左下走
else
flag = 4;//当前向右上走
}
else if (flag == 2)//如果上一步是左下
{
if (y - 1 >= 0 && x + 1 < n)
flag = 2;//继续左下
else if (x+1<n)
flag = 3;
else 
flag = 1;
}
else if (flag == 3)//上一步是向下
{
if (x - 1 >= 0 && y + 1 < n)//如果可以右上
flag = 4;
else
flag = 2;  //当前向左下走
}
else if (flag == 4)//上一步是右上
{
if (x - 1 >= 0 && y + 1 < n)//可以继续右上
flag = 4;
else if (y + 1 < n)//是否可以向右
flag = 1;
else
flag = 3;
}

switch (flag)
{
case 1:
{
y++;
result.push_back(sqr[x][y]);
break;
}
case 2:
{
x++;
y--;
result.push_back(sqr[x][y]);
break;
}
case 3:
{
x++;
result.push_back(sqr[x][y]);
break;
}
case 4:
{
x--;
y++;
result.push_back(sqr[x][y]);
break;
}
default:
break;
}
}

for (int i = 0; i<result.size(); i++)
{
cout << result[i] << " ";
}

//system("pause");
}