面试例题:输入n,求一个nXn矩阵,规定矩阵沿45度递增,形成一个zigzag数组(JPEG编码里取像素数据的排列顺序),请问如何用C++实现?
(中国台湾著名硬件公司2007年11月面试题)(自程序员面试宝典第二版p89)
程序实现如下:
分析:例如输入为8的时候得到的zigzag数组为:
由于对同一斜线上的元素,s=i+j为常数
1.对于第n个(n<N)斜线:
每一斜线个数比上一行多一个,则每一斜线的第一个值表示了在该斜线之前元素的个数,即一个累加 :
(s+1)*s/2
斜线中的任意元素可表示为s*(s+1)/2+i
2.同理对于第n个(n>=N)的斜线:
每斜线的元素个数开始减少,等差数组不适用,为方便起见,可以用减法计算元素个数,
剩余元素的斜线数目:s1=(2(N-1)-(i+j))
元素值=(N*N-(s1+1)*s1/2)-(N-i)=下一斜线的第一个元素值-该斜线剩余元素的个数
1.对于第n个(n<N)斜线:
每一斜线个数比上一行多一个,则每一斜线的第一个值表示了在该斜线之前元素的个数,即一个累加 :
(s+1)*s/2
斜线中的任意元素可表示为s*(s+1)/2+i
2.同理对于第n个(n>=N)的斜线:
每斜线的元素个数开始减少,等差数组不适用,为方便起见,可以用减法计算元素个数,
剩余元素的斜线数目:s1=(2(N-1)-(i+j))
元素值=(N*N-(s1+1)*s1/2)-(N-i)=下一斜线的第一个元素值-该斜线剩余元素的个数
程序如下:
#include <iostream>
#include <stdio.h>
using namespace std;
int main()
{
int N;
int s, i, j;
int squa;
scanf("%d", &N);
int **a = (int **)malloc(N * sizeof(int));
if(NULL == a)
{
return 0;
}
//空间分配
for(i = 0; i < N; i++)
{
if(NULL == (a[i] = (int*)malloc(N * sizeof(int))))
{
while(--i >= 0)
{
free(a[i]);
}
free(a);
return 1;
}
}
squa = N * N;
//求对应位置上应填写的值
for(i = 0; i < N; i++)
{
for(j = 0; j < N; j++)
{
s = i + j;
if(s < N)//上三角
{
s = i + j;//为了看的更清楚,这里加了这条与下面对称的语句
a[i][j] = s * (s + 1) / 2 + ((0 == (i + j) % 2) ? i : j);
}
else//下三角
{
s = (N - 1 - i) + (N - 1 - j);
a[i][j] = squa - s * (s + 1) / 2 - (N - ((0 == (i + j) % 2) ? i : j));
}
}
}
//打印输出
for(i = 0; i < N; i ++)
{
for(j = 0; j < N; j++)
{
printf("%6d", a[i][j]);
}
printf("\n");
}
return 0;
}