Zigzag逆扫描代码 MATLAB版

即将一个1 * k 长度的向量，逆扫描为 M * N 的矩阵(如果 k < M * N，则在 M * N 的矩阵中用 0 补齐；若 k > M * N，则向量会被截断)。

代码如下，仅供参考。

function mtr = izigzagScan(vec, m, n)

% izigzagScan(vec, m, n) Inverse zigzag scan.

% mtr = izigzagScan(vec, m, n) inverse zigzag scan of vec, and the target

% matrix is m-by-n. The input parameter vec is an 1-by-k vector. If the

% length of vector is smaller than the size of mtr (i.e. k < m*n), then, the

% pargram will use 0 to fill the matrix. If k > m*n, the vector will be truncated.

%

% version: v1.0

% author: tianlan

% time: Dec 16, 2016

% CHECK INPUT ARGUMENTS

if nargin < 3

error('Not enough input arguments!');

elseif nargin > 3

error('Too many input arguments!');

end

% Get length of vector.

[mv,nv,tv] = size(vec);

if tv ~= 1 || mv ~=1

error('The input vector needs to be 1-by-k vector!');

end

if nv < m*n % if nv < m*n, then, use 0 to fill the matrix.

vec = [vec, zeros(1,m*n-nv)];

end

% REVERSE SCAN A VECTOR BY USING ZIGZAG ORDER.

mtr = zeros(m,n);

% zigzag scan.

x = 1;

y = 1;

vecNum = 0;

vecNum = vecNum + 1;

mtr(x,y) = vec(1,vecNum);

while 1

% arrive right-up border, needs to change direction.

if (y+1) <= n && x == 1 % towards right one step.

y = y + 1;

elseif (y+1) > n && (x+1) <= m % towards down one step.

x = x + 1;

else

break;

end

vecNum = vecNum + 1;

mtr(x,y) = vec(1,vecNum) ;

% judge the scan process is arrived the right-down corner (i.e. mtr(m,n)) or not.

if x == m && y == n

break;

end

% scan the matrix towards with left-down direction.

while (y-1) >= 1 && (x+1) <= m

y = y - 1;

x = x + 1;

vecNum = vecNum + 1;

mtr(x,y) = vec(1,vecNum);

end

% arrive the left-down border, needs to change direction.

if (x+1) <= m && y == 1% towards down one step.

x = x + 1;

elseif (y+1) <= n && x == m % towards right one step.

y = y + 1;

else

break;

end

vecNum = vecNum + 1;

mtr(x,y) = vec(1,vecNum);

% judge the scan process is arrived the right-down corner (i.e. mtr(m,n)) or not.

if x == m && y == n

break;

end

% scan the matrix towards with right-up direction.

while (x-1) >= 1 && (y+1) <= n

x = x - 1;

y = y + 1;

vecNum = vecNum + 1;

mtr(x,y) = vec(1,vecNum);

end

end