【蓝桥】基础练习 回形取数

参考文章原地址
问题描述
　　回形取数就是沿矩阵的边取数，若当前方向上无数可取或已经取过，则左转90度。一开始位于矩阵左上角，方向向下。
　　
输入格式
　　输入第一行是两个不超过200的正整数m, n，表示矩阵的行和列。接下来m行每行n个整数，表示这个矩阵。
输出格式
　　输出只有一行，共m*n个数，为输入矩阵回形取数得到的结果。数之间用一个空格分隔，行末不要有多余的空格。
　　
样例输入
3 3
1 2 3
4 5 6
7 8 9
样例输出
1 4 7 8 9 6 3 2 5
样例输入
3 2
1 2
3 4
5 6
样例输出
1 3 5 6 4 2

#include<iostream>
using namespace std;

int main()
{
	int m,n;
	cin>>m>>n;
	int a[m][n],flag[m][n];  // a[m][n]矩阵数组，flag[m][n]遍历标记 
	for(int i=0;i<m;i++){  // 输入矩阵元素 
		for(int j=0;j<n;j++){
			cin>>a[i][j];
			flag[i][j] = 0;
		}
	}
	int k=0,dir[4][2]={1,0,   // 向下  , k控制方向 
	                   0,1,   // 向右 
					  -1,0,   // 向上 
					   0,-1}; // 向左 
	int x=-1,y=0,num=0;  // x,y为遍历下标起点,num已遍历元素个数
	while(num<m*n){
		x += dir[k%4][0];  // 改变元素下标 x,y 
		y += dir[k%4][1];
	    if(x>=0 && x<=m-1 && y>=0 && y<=n-1 && !flag[x][y]){ // 该元素在矩阵范围内且未遍历 
	    	cout<<a[x][y]<<" ";
	    	num++;
	    	flag[x][y] = 1;
	    }
		else{  // 若超出范围，则后退一步，k++, 转向 
			x -= dir[k%4][0];
			y -= dir[k%4][1]; 
			k++;
		}	
	} 
					    
	return 0;
}