HTML Code

<!DOCTYPE html>
<html lang="en">
<head>
  <meta charset="utf-8">
  <title>Matrix subtraction — shape-safe</title>
</head>
<body>
  <h1>Shape-safe matrix subtract</h1>
  <p>Edit <code>goodA</code>, <code>goodB</code>, <code>badA</code>, <code>badB</code>, then <strong>Run</strong>.</p>
  <pre id="out"></pre>
  <script>
    (function () {
      function sameShape(a, b) {
        if (!a.length || !b.length) {
          return false;
        }
        if (a.length !== b.length) {
          return false;
        }
        var cols = a[0].length;
        if (cols === 0) {
          return false;
        }
        var all = a.concat(b);
        var i;
        for (i = 0; i < all.length; i++) {
          if (all[i].length !== cols) {
            return false;
          }
        }
        return true;
      }
      function subtractSafe(a, b) {
        if (!sameShape(a, b)) {
          return null;
        }
        var rows = a.length;
        var cols = a[0].length;
        var result = [];
        var i, j;
        for (i = 0; i < rows; i++) {
          result[i] = [];
          for (j = 0; j < cols; j++) {
            result[i][j] = a[i][j] - b[i][j];
          }
        }
        return result;
      }
      var goodA = [[5, 8, 2], [7, 4, 9], [3, 6, 1]];
      var goodB = [[3, 1, 7], [6, 9, 2], [8, 5, 4]];
      var badA = [[1, 2], [3, 4]];
      var badB = [[5, 6, 7]];
      var lines = [];
      lines.push(JSON.stringify(subtractSafe(goodA, goodB)));
      lines.push(String(subtractSafe(badA, badB)));
      document.getElementById("out").textContent = lines.join("\n");
    })();
  </script>
</body>
</html>

Lines: 0 | Characters: 0

Live Preview Auto-refresh on Run

JavaScript Matrix Subtraction — Shape-safe (Try It)