201412-2z字形扫描

#include <bits/stdc++.h>
using namespace std;
int main()
{
     int n;
     cin >> n;
    int a[n+1][n+1];
    for(int i=0; i<=n; i++)
    {
        for(int j=0; j<=n; j++)
        {
            a[i][j] = 0;
        }
    }
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<n; j++)
        {
            cin >> a[i][j];
        }
    }
    int i=0, j=0;
    cout << a[i][j] <<" ";
    while(i<n&&j<n)
    {
        if(i==0)
        {
            j++;
            if(j==n)break;
        }
//        cout << "i=" << i << "j=" << j << endl;
        cout << a[i][j] << " ";
        while(j!=0)
        {
            cout << a[++i][--j] << " ";
            if(i==n)
                break;
//            cout << "i=" << i << "j=" << j << endl;
        }
        if(j==0)
        {
            i++;
            if(i==n)break;
        }
//        cout << "i=" << i << "j=" << j << endl;
        cout << a[i][j] << " ";
        while(i!=0)
        {
            cout << a[--i][++j] << " ";
            if(j==n)break;
//             cout << "i=" << i << "j=" << j << endl;
        }
    }
    if(i==n)
        i--;
    else if(j==n)
        j--;
//        cout << "i=" << i << "j=" << j << endl;
    while(i<n&&j<n)
    {
        if(i==n-1)
        {
            j++;
            if(j==n)break;
        cout << a[i][j] << " ";
        }
//         cout << "i=" << i << "j=" << j << endl;

        if(i!=0)
        {
        while(j!=n-1)
        {
            cout << a[--i][++j] << " ";
            if(j==n)break;
//             cout << "i=" << i << "j=" << j << endl;
        }
        }

        if(j==n-1)
        {
            i++;
            if(i==n)break;
            cout << a[i][j] << " ";
        }
//        cout << "i=" << i << "j=" << j << endl;
        if(j!=0)
        {
        while(i!=n-1)
        {
               cout << a[++i][--j] << " ";
            if(i==n)
                break;
//            cout << "i=" << i << "j=" << j << endl;
        }
        }
    }
    cout << endl;

    return 0;
}