arylo / from_arylo Goto Github PK

import test from 'ava';

/* eslint-disable no-plusplus,no-continue */

let no = 0;

function fn(g, deep = 1) {
  const arr = [ ];
  const m = g.length || 0;
  const n = (g[0] || []).length || 0;
  const top = deep - 1;
  const bottom = m - deep;
  const left = deep - 1;
  const right = n - deep;
  // console.log(m, n, top, bottom, left, right);
  for (let i = 0; i < n; i++) {
    if (i < left || right < i) {
      continue;
    }
    // console.log('t', g[top][i]);
    arr.push(g[top][i]);
  }
  if (bottom - top > 1) {
    for (let j = top + 1; j < bottom; j++) {
      // console.log('r', g[j][right]);
      arr.push(g[j][right]);
    }
  }
  if (bottom - top > 0) {
    for (let i = 0; i < n; i++) {
      if (i < left || right < i) {
        continue;
      }
      // console.log('b', i, g[bottom][n - i - 1]);
      arr.push(g[bottom][n - i - 1]);
    }
  }
  if (bottom - top > 1 && right - left > 0) {
    for (let j = top + 1; j < bottom; j++) {
      // console.log('l', g[bottom - j][left]);
      arr.push(g[bottom - j][left]);
    }
  }
  if (bottom - top > 1 && right - left > 1) {
    arr.push(...fn(g, deep + 1));
  }
  return arr;
}

test(`#${++no}`, (t) => {
  const g = [];
  t.deepEqual(fn(g), []);
});

test(`#${++no}`, (t) => {
  const g = [
    []
  ];
  t.deepEqual(fn(g), []);
});

test(`#${++no}`, (t) => {
  const g = [
    [1, 2],
  ];
  t.deepEqual(fn(g), [1, 2]);
});

test(`#${++no}`, (t) => {
  const g = [
    [1, 2],
    [3, 4],
  ];
  t.deepEqual(fn(g), [1, 2, 4, 3]);
});

test(`#${++no}`, (t) => {
  const g = [
    [1, 2, 2],
    [3, 4, 4],
  ];
  t.deepEqual(fn(g), [1, 2, 2, 4, 4, 3]);
});

test(`#${++no}`, (t) => {
  const g = [
    [1, 2],
    [3, 4],
    [5, 6],
  ];
  t.deepEqual(fn(g), [1, 2, 4, 6, 5, 3]);
});

test(`#${++no}`, (t) => {
  const g = [
    [1, 2, 3],
    [3, 4, 5],
    [5, 6, 7],
  ];
  t.deepEqual(fn(g), [1, 2, 3, 5, 7, 6, 5, 3, 4]);
});

test(`#${++no}`, (t) => {
  const g = [
    [1, 2, 3],
    [3, 4, 5],
    [5, 6, 7],
    [7, 8, 9],
  ];
  t.deepEqual(fn(g), [1, 2, 3, 5, 7, 9, 8, 7, 5, 3, 4, 6]);
});

test(`#${++no}`, (t) => {
  const g = [
    [1, 2, 3, 4],
    [5, 6, 7, 8],
    [9, 10, 11, 12],
    [13, 14, 15, 16],
  ];
  t.deepEqual(fn(g), [1, 2, 3, 4, 8, 12, 16, 15, 14, 13, 9, 5, 6, 7, 11, 10]);
});