freeCodeCamp/curriculum/challenges/chinese/22-rosetta-code/rosetta-code-challenges/zig-zag-matrix.md

id, title, challengeType, forumTopicId, dashedName

id	title	challengeType	forumTopicId	dashedName
594810f028c0303b75339ad8	之字形矩阵	1	302348	zig-zag-matrix

--description--

A 'zig-zag' array is a square arrangement of the first N^2 integers, where the numbers increase sequentially as you zig-zag along the array's anti-diagonals.

例如，对于输入 5，应该产生以下结果：

 0  1  5  6 14
 2  4  7 13 15
 3  8 12 16 21
 9 11 17 20 22
10 18 19 23 24

--instructions--

编写一个函数，该函数采用 zig-zag 矩阵的大小，并以二维数组的形式返回相应的矩阵。

--hints--

ZigZagMatrix 应该是一个函数。

assert.equal(typeof ZigZagMatrix, 'function');

ZigZagMatrix 应返回数组。

assert.equal(typeof ZigZagMatrix(1), 'object');

ZigZagMatrix 应返回嵌套数组的数组。

assert.equal(typeof ZigZagMatrix(1)[0], 'object');

ZigZagMatrix(1) 应返回 0。

assert.deepEqual(ZigZagMatrix(1), zm1);

ZigZagMatrix(2) 应该返回 0, 1], [2, 3。

assert.deepEqual(ZigZagMatrix(2), zm2);

ZigZagMatrix(5) 应返回指定的矩阵。

assert.deepEqual(ZigZagMatrix(5), zm5);

--seed--

--after-user-code--

const zm1 = [[0]];
const zm2 = [[0, 1], [2, 3]];
const zm5 = [
  [0, 1, 5, 6, 14],
  [2, 4, 7, 13, 15],
  [3, 8, 12, 16, 21],
  [9, 11, 17, 20, 22],
  [10, 18, 19, 23, 24]
];

--seed-contents--

function ZigZagMatrix(n) {

  return [[], []];
}

--solutions--

function ZigZagMatrix(n) {
  const mtx = [];
  for (let i = 0; i < n; i++) {
    mtx[i] = [];
  }

  let i = 1;
  let j = 1;
  for (let e = 0; e < n * n; e++) {
    mtx[i - 1][j - 1] = e;
    if ((i + j) % 2 === 0) {
      // Even stripes
      if (j < n) j++;
      else i += 2;
      if (i > 1) i--;
    } else {
      // Odd stripes
      if (i < n) i++;
      else j += 2;
      if (j > 1) j--;
    }
  }
  return mtx;
}