这是一道经典笔试题,题目如下:
输入矩阵的行M和列N,然后打印出如下矩阵:
或者是输入一个M*N的矩阵,按之字形将矩阵元素提取/打印出来,虽然表述不同,但其实现都类似——需要按之字形遍历矩阵。
在《程序员面试宝典》一书中也有类似的题,不过那道题是N阶方阵,不具有普遍性。
我的思路如下:容易知道,对角线方向的行数(如<1,2>、<5,4,3>)为M+N,由于元素在对角线方向的奇数行和偶数行增大的方向正好相反,可由此入手,用一个变量s(0<s<M+N)记录对角线行数的序号,在奇偶行上分别控制序列移动和边界条件。需要注意的是到达对角线行终点后元素的移动方向,有两种情况——向下和向右,此处边界条件的设置比较关键。
代码实现如下:
// zigzag_array.cpp : 定义控制台应用程序的入口点。
//
#include "stdafx.h"
#include "math.h"
#include <iostream>
using namespace std;
int _tmain(int argc, _TCHAR* argv[])
{
int i=0,j=0,s=0,m,n,k=0; //s为斜线数目
int a[100][100]={0};
cin>>m>>n;
for(s=0;s<m+n-1;s++)
{
/*奇数行*/
if((i+j)%2==0)
{
while(i<m&&j>=0)
a[i++][j--]=k++;
if(i==m) //向右移动
j+=2;i=m-1;
else if(j<0)
j=0; //向下移动
}
/*偶数行*/
else
{
while(j<n&&i>=0)
a[i--][j++]=k++;
if(j==n)
i+=2;j=n-1; //向下移动
else if(i<0)
i=0; //向右
}
}
/*输出*/
for(i=0;i<m;i++)
{
for(j=0;j<n;j++)
cout<<a[i][j]<<'\t';
cout<<endl;
}
return 0;
}