Z字形掃描(CSP認證)
阿新 • • 發佈:2021-01-07
問題描述
在影象編碼的演算法中,需要將一個給定的方形矩陣進行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的正整數。
#include<bits/stdc++.h>
using namespace std;
const int N=1010;
int a[N][N];
int n;
int main(){
cin>>n;
for(int i=1;i<= n;i++)
for(int j=1;j<=n;j++)
cin>>a[i][j];
int x=0,y=1;
int t=n+n-1;//迴圈這麼多次
for(int i=1;i<=t;i++){
if(i&1){ //i為奇數
//起始點座標
if(x+1<=n&&y<=n)x++; //有位置向下走走一格
else y++;//否則向右
while(x-1>0&&y+1<=n) {
cout<<a[ x][y]<<' ';
x--;
y++;
}
cout<<a[x][y]<<' ';
}
else{
//起始點座標
if(x<=n&&y+1<=n)y++;//有位置向右走一格
else x++;//否則向下
while(x+1<=n&&y-1>0) {
cout<<a[x][y]<<' ';
x++;
y--;
}
cout<<a[x][y]<<' ';
}
}
return 0;
}