CCF201412-2 Z字形扫描

sort函数排序时，区间是左闭右开的，所以右边要加一

using namespace std;
#include<bits/stdc++.h>
int main(){
	int n,i,j;
	int a[505][505];
	cin>>n;
	for(i=1;i<=n;i++){
		for(j=1;j<=n;j++){
			cin>>a[i][j];
		}
	}
	int x=1,y=1;
	for(i=0;i<n;i++){
		if(i%2 == 0){
			for(j=0;j<i;j++){
				cout<<a[x--][y++]<<" ";
			}
			cout<<a[x][y++]<<" ";
		}
		else{
			for(j=0;j<i;j++){
				cout<<a[x++][y--]<<" ";
			}
			cout<<a[x++][y]<<" ";
		}
	}
	if(n%2 == 0){
		y++;
		x--;
	}
	else{
		x++;
		y--;
	}
	for(i=n-2;i>0;i--){
		if(i%2 == 0){
			for(j=0;j<i;j++){
				cout<<a[x--][y++]<<" ";
			}	
			cout<<a[x++][y]<<" ";
		}
		else{
			for(j=0;j<i;j++){
				cout<<a[x++][y--]<<" ";
			}
			cout<<a[x][y++]<<" ";
		}
	}
	if(n!=1) cout<<a[n][n];
	return 0;
}