試題編號：	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");
}