Z字形扫描( 第03次CCF计算机软件能力认证)

Z字形扫描
找规律即可

#include <bits/stdc++.h>

using namespace std;

const int N = 550;

int a[N][N];
int n;

signed 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 <= n * 2; ++i){
        if(i & 1){
            int row = i, col = 1;
            for(int j = 1; j <= i; ++j){
                if(a[row][col]) cout << a[row][col] << " ";
                --row, ++col;
            }
        }else{
            int row = 1, col = i;
            for(int j = 1; j <= i; ++j){
                if(a[row][col]) cout << a[row][col] << " ";
                ++row, --col;
            }
        }
    }
    
    return 0;
}