自己编写的matlab zigzag排序算法

%%%%%%%%%%%%%%

%%%Copyright：wzr1201  

%%%Data：2013-05-08

%%%%%%%%%%%%%%

clear;

clc;
N=512;
A=randint(N,N,[2,10]); %N-by-N, random array of integers


len=N*N;
B=zeros(1,len);%zigzag排序后的数组
B(1)=A(1);
k=1; 
i=1;j=1;%起始坐标
d=1;%排序的方向标示。值为1表示下一个元素在右上方向；值为-1表示下一个元素在左下方向
flag=1;%上下三角标示。值为1表示当前元素在上三角；值为0表示当前元素在下三角
while k<len
   
    if flag==1%当前元素为上三角
        if i~=1&&j~=1%非边界元素
            if d==1%下一个元素在右上方向
                i=i-1;j=j+1;
            else%下一个元素在左下方向
                i=i+1;j=j-1;
            end
        else
            if i==1%上边界                
                if d==1%需要变向
                    j