CSP 201412-2 Z字形扫描

一、问题链接

CSP 201412-2 Z字形扫描

二、参考代码

#include <bits/stdc++.h>
using namespace std;
int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	int n;
	cin>>n;
	vector<vector<int>> v(n,vector<int>(n));
	for(int i=0; i<n; i++) {
		for(int j=0; j<n; j++) {
			cin>>v[i][j];
		}
	}
	int i=0,j=0;
	int a;
	while(1) {
		//右上i--，j++
		while(1) {
			a=v[i][j];
			cout<<v[i][j]<<" ";
			if(i==n-1 and j==n-1) return 0;
			i--;
			j++;
			//若到达右上角，即碰右壁又碰上壁，往下走；
			//所以向下走优先 
			if(j==n) { //碰右壁 ,向下走
				j--;
				i+=2;
				break;
			}
			if(i==-1) { //碰上壁,向右走
				i++;
				break;
			}

		}
		//左下i++，j--
		while(1) {
			a=v[i][j];
			cout<<v[i][j]<<" ";
			if(i==n-1 and j==n-1) return 0;
			i++;
			j--;
			//左下角，向右走，右走优先 
			if(i==n) { //碰下底，向右走 
				i--;
				j+=2;
				break;
			}
			if(j==-1) { //碰左壁，向左走 
				j++;
				break;
			}
		}
	}
	return 0;
}