AcWing 3208. Z字形扫描(模拟、练习一下重载输入输出）

#include<bits/stdc++.h>
using namespace std;
const int N=1010;
vector<int>v[N];
int n,a[N/2][N/2];
bool f(int x,int y){
    return x>=1&&x<=n&&y>=1&&y<=n;
}
ostream & operator <<(ostream & os,vector<int>&v){
    for(auto it:v){
        os<<it<<" ";
    }
    return os;
}
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            cin>>a[i][j];
        }
    }
    for(int i=1;i<=2*n-1;i++){
        int st,ed;
        if(i>=n)st=n,ed=i-n+1;
        else st=i,ed=1;
        while(f(st,ed)){
            v[i].push_back(a[st][ed]);
            st--,ed++;
        }
    }
    int flag=0;
    for(int i=1;i<=2*n-1;i++){
        if(!flag)cout<<v[i];
        else{
            reverse(v[i].begin(),v[i].end());
            cout<<v[i];
        }
        flag^=1;
    }
}