1. 程式人生 > >CCF 201412-2 Z字形掃描

CCF 201412-2 Z字形掃描

試題編號: 201412-2
試題名稱: Z字形掃描
時間限制: 2.0s
記憶體限制: 256.0MB
問題描述:

問題描述

  在影象編碼的演算法中,需要將一個給定的方形矩陣進行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的正整數。

註釋處是我一開始寫的程式碼,k總是從1開始,那麼在下三角中第一次輸出的值總是按照奇數輸出的,所以應當以偶數輸出時就會出錯,只要多測試幾個樣例即可避免該問題!

#include <iostream>
#define maxn 505

int a[maxn][maxn];

using namespace std;

int main()
{
    int N;
    cin >> N;
    for(int i = 0;i < N;i++)
    {
        for(int j = 0;j < N;j++)
        {
            cin >> a[i][j];
        }
    }
    for(int k = 1;k <= N;k++)
    {
        for(int i = k - 1,j = 0;i >= 0;i--,j++)
        {
            if(k%2 != 0)
            {
                cout << a[i][j] << " ";
            }
            else
            {
                cout << a[j][i] << " ";
            }
        }
    }
/*
     for(int k = 1;k < N;k++)
     {
        for(int i = N - 1,j = k;j < N; i--,j++)
        {
            if(k % 2 != 0)
            {
                cout << a[i][j] << " ";
            }
            else
            {
                cout << a[j][i] << " ";
            }
        }
    }
*/
    for(int k = N - 1;k > 0;k--)
    {
        for(int i = N - 1,j = N - k;j < N; i--,j++)
        {
            if(k % 2 != 0)
            {
                cout << a[i][j] << " ";
            }
            else
            {
                cout << a[j][i] << " ";
            }
        }
    }
    return 0;
}