zigzag.m

来自「matlab例题。这样的描述就已经很清楚了。」· M 代码 · 共 142 行

function x=zigzag(a)
clear x;
[ro co]=size(a);
x(1)=a(1,1);

if co>=ro
   for n=2:2:ro
      x=[x diag(flipud(a(1:n,1:n)))'];
      if (n+1)<=ro
         x=[x diag(fliplr(a(1:n+1,1:n+1)))'];
      end
   end   
s=rem(ro,2);
if co==ro
   if s==0
      for n=2:2:ro
         x=[x diag(fliplr(a(n:ro,n:ro)))'];
         if (n+1)<=ro
            x=[x diag(flipud(a((n+1):ro,(n+1):ro)))'];
         end
      end
   else 
      for n=2:2:ro
         x=[x diag(flipud(a(n:ro,n:ro)))'];
         if (n+1)<=ro
            x=[x diag(fliplr(a((n+1):ro,(n+1):ro)))'];
         end
      end
   end
else if co>ro
      ss=rem(co,2);
      if s==0
         
            for n=2:2:co-ro+1
               x=[x diag(fliplr(a(:,n:ro+n-1)))'];
               if (n+1)<=co-ro+1
                  x=[x diag(flipud(a(:,n+1:ro+n)))'];
               end
            end
            if ss==0
               for n=2:2:ro
                  x=[x diag(fliplr(a(n:ro,co-ro+n:co)))'];
                  if (n+1)<=ro
                     x=[x diag(flipud(a((n+1):ro,co-ro+(n+1):co)))'];
                  end
               end
            else 
               for n=2:2:ro
                  x=[x diag(flipud(a(n:ro,co-ro+n:co)))'];
                  if (n+1)<=ro
                     x=[x diag(fliplr(a((n+1):ro,co-ro+(n+1):co)))'];
                  end
               end
            end
         else
            for n=2:2:co-ro+1
               x=[x diag(flipud(a(:,n:ro+n-1)))'];
               if (n+1)<=co-ro+1
                  x=[x diag(fliplr(a(:,n+1:ro+n)))'];
               end
            end
            if ss==0
               for n=2:2:ro
                  x=[x diag(fliplr(a(n:ro,co-ro+n:co)))'];
                  if (n+1)<=ro
                     x=[x diag(flipud(a((n+1):ro,co-ro+(n+1):co)))'];
                  end
               end
            else
               for n=2:2:ro
                  x=[x diag(flipud(a(n:ro,co-ro+n:co)))'];
                  if (n+1)<=ro
                     x=[x diag(fliplr(a((n+1):ro,co-ro+(n+1):co)))'];
                  end
               end
            end
            
         end
      end
   end
else if ro>co
      a=a';
      [ro co]=size(a);
      ss=rem(co,2);
      for n=2:2:ro
      x=[x diag(fliplr(a(1:n,1:n)))'];
      if (n+1)<=ro
         x=[x diag(flipud(a(1:n+1,1:n+1)))'];
      end
   end   
   s=rem(ro,2);
   if s==0
         
            for n=2:2:co-ro+1
               x=[x diag(flipud(a(:,n:ro+n-1)))'];
               if (n+1)<=co-ro+1
                  x=[x diag(fliplr(a(:,n+1:ro+n)))'];
               end
            end
            ss=rem(co,2);
            if ss==0
               for n=2:2:ro
                  x=[x diag(flipud(a(n:ro,co-ro+n:co)))'];
                  if (n+1)<=ro
                     x=[x diag(fliplr(a((n+1):ro,co-ro+(n+1):co)))'];
                  end
               end
            else 
               for n=2:2:ro
                  x=[x diag(fliplr(a(n:ro,co-ro+n:co)))'];
                  if (n+1)<=ro
                     x=[x diag(flipud(a((n+1):ro,co-ro+(n+1):co)))'];
                  end
               end
            end
         else
            for n=2:2:co-ro+1
               x=[x diag(fliplr(a(:,n:ro+n-1)))'];
               if (n+1)<=co-ro+1
                  x=[x diag(flipud(a(:,n+1:ro+n)))'];
               end
            end
            if ss==0
               for n=2:2:ro
                  x=[x diag(flipud(a(n:ro,co-ro+n:co)))'];
                  if (n+1)<=ro
                     x=[x diag(fliplr(a((n+1):ro,co-ro+(n+1):co)))'];
                  end
               end
            else
               for n=2:2:ro
                  x=[x diag(fliplr(a(n:ro,co-ro+n:co)))'];
                  if (n+1)<=ro
                     x=[x diag(flipud(a((n+1):ro,co-ro+(n+1):co)))'];
                  end
               end
            end
            
         end
      end
end