CCF认证-2014-12-2 Z字形扫描

问题描述
　　在图像编码的算法中，需要将一个给定的方形矩阵进行Z字形扫描(Zigzag Scan)。给定一个n×n的矩阵，Z字形扫描的过程如下图所示：
　　
对于下面的4×4的矩阵，
　　1 5 3 9
　　3 7 5 6
　　9 4 6 4
　　7 3 1 3
　　对其进行Z字形扫描后得到长度为16的序列：
　　1 5 3 9 7 3 9 5 4 7 3 6 6 4 1 3
　　请实现一个Z字形扫描的程序，给定一个n×n的矩阵，输出对这个矩阵进行Z字形扫描的结果。

输入格式：
　　输入的第一行包含一个整数n，表示矩阵的大小。
　　输入的第二行到第n+1行每行包含n个正整数，由空格分隔，表示给定的矩阵。

输出格式：
　　输出一行，包含n×n个整数，由空格分隔，表示输入的矩阵经过Z字形扫描后的结果。

样例输入
4
1 5 3 9
3 7 5 6
9 4 6 4
7 3 1 3

样例输出
1 5 3 9 7 3 9 5 4 7 3 6 6 4 1 3
评测用例规模与约定
　　1≤n≤500，矩阵元素为不超过1000的正整数。

思路：找出本题的规律，只有当row ==n-1 or col == 0 往右上方递进输出；当row == 0 || col == n - 1 往左下方递进输出；

一开始的90分是因为没有考虑一个元素的矩阵，开始我是把第一个元素先输出，然后在进入循环；后来改动：先判断右上方递进，然后再执行左下方递进；这样第一个元素就能直接进入循环而不用一开始单独输出；
code：

#include <bits/stdc++.h>
using namespace std;
#define Max_n 510

int a[Max_n][Max_n];

int main()
{
    int n, row, col;
    cin >> n;
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
        {
            cin >> a[i][j];
        }
    }
    row=0;
    col=0;
    while (1)
    {
        if (row == n - 1 && col == n - 1) //判断是否到达终点
        {
            cout << a[row][col]<<" ";
            break;
        }
        if (row == n - 1 || col == 0)//往右上推
        {
            for (int i = row, j = col; i >= 0 && j < n; --i, ++j)
            {
                cout << a[i][j] << " ";
                row = i;
                col = j;
            }
            if (col == n - 1)
                row++;
            else col++;
        }
        if (row == 0 || col == n - 1) //往左下推
        {
            for (int i = row, j = col; i < n && j >= 0; ++i, --j)
            {
                cout << a[i][j] << " ";
                row = i;
                col = j;
            }
            if (row == n - 1)
                col++;
            else row++;
        }   
        
    }
    return 0;
}