輸入n,生成一個n*n的矩陣,規定沿着45度方向遞增,形成一個zigzag數組(JPEG編碼裏取像素數據的排列次序),請問如何實現?
例如:
實現如下:
void makeA(int **A,int N){
int sum=0;
for(int i=0;i<N;i++){
for(int j=0;j<N;j++){
sum=i+j;
if(sum<N)
A[i][j]=sum*(sum+1)/2+(sum%2==0?j:i);
else{
A[i][j]=N*N-1-((N-1)*2-sum)*((N-1)*2-sum+1)/2-(((N-1)*2-sum)%2==0?(N-1-j):(N-1-i));
}
}
}
}
int main(int argc, char *argv[])
{
int N=9;
int **A;
int i;
A=(int **)malloc(sizeof(int *)*N);
for(i=0;i<N;i++)
*(A+i)=(int *)malloc(sizeof(int)*N);
makeA(A,N);
for(i=0;i<N;i++){
for(int j=0;j<N;j++){
cout<<setw(3)<<A[i][j]<<" ";
}
cout<<endl;
}
getchar();
for(i=0;i<N;i++)
free(*(A+i));
free(A);
return 0;
}