AcWing每日一题--Z字形扫描

https://www.acwing.com/problem/content/3211/

直接看成两个三角形（略显复杂）

#include<iostream>
using namespace std;
const int N=500;
int a[N][N];
int main(void){
    int n;
    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;i++){
        if(i%2){
            int x=i,y=1;
            while(x>0){
                cout<<a[x][y]<<" ";
                x--,y++;
            }
        }else{
            int x=1,y=i;
            while(y>0){
                cout<<a[x][y]<<" ";
                x++,y--;
            }
        }
    }  
    for(int i=n+1;i<=2*n-1;i++){
        if(i%2){
            int x=n,y=i-n+1;
            while(y<=n){
                cout<<a[x][y]<<" ";
                x--,y++;
            }
        }else{
            int x=i-n+1,y=n;
            while(x<=n){
                cout<<a[x][y]<<" ";
                x++,y--;
            }
        }
    }
    
    return 0;
}

将矩阵扩展到2n*2n，如此便不需要将矩阵分为两个三角形来考虑了。

#include<iostream>
using namespace std;
const int N=500;
int a[N][N];
int main(void){
    int n;
    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;i++){
        if(i%2){
            int x=i,y=1;
            while(x>0){
                if(x<=n&&y<=n){
                    cout<<a[x][y]<<" ";
                }
                x--,y++;
            }
        }else{
            int x=1,y=i;
            while(y>0){
                if(x<=n&&y<=n){
                    cout<<a[x][y]<<" ";
                }
                x++,y--;
            }
        }
    }
    
    return 0;
}