freeCodeCamp/curriculum/challenges/chinese/10-coding-interview-prep/rosetta-code/element-wise-operations.md

3.0 KiB

id title challengeType forumTopicId dashedName
599c333915e0ea32d04d4bec Element-wise operations 5 302252 element-wise-operations

--description--

Implement basic element-wise matrix-matrix and scalar-matrix operations.

Implement:

  • addition
  • subtraction
  • multiplication
  • division
  • exponentiation

The first parameter will be the operation to be performed, for example, "m_add" for matrix addition and "s_add" for scalar addition. The second and third parameters will be the matrices on which the operations are to be performed.

--hints--

operation should be a function.

assert(typeof operation === 'function');

operation("m_add",[[1,2],[3,4]],[[1,2],[3,4]]) should return [[2,4],[6,8]].

assert.deepEqual(
  operation(
    'm_add',
    [
      [1, 2],
      [3, 4]
    ],
    [
      [1, 2],
      [3, 4]
    ]
  ),
  [
    [2, 4],
    [6, 8]
  ]
);

operation("s_add",[[1,2],[3,4]],2) should return [[3,4],[5,6]].

assert.deepEqual(
  operation(
    's_add',
    [
      [1, 2],
      [3, 4]
    ],
    2
  ),
  [
    [3, 4],
    [5, 6]
  ]
);

operation("m_sub",[[1,2],[3,4]],[[1,2],[3,4]]) should return [[0,0],[0,0]].

assert.deepEqual(
  operation(
    'm_sub',
    [
      [1, 2],
      [3, 4]
    ],
    [
      [1, 2],
      [3, 4]
    ]
  ),
  [
    [0, 0],
    [0, 0]
  ]
);

operation("m_mult",[[1,2],[3,4]],[[1,2],[3,4]]) should return [[1,4],[9,16]].

assert.deepEqual(
  operation(
    'm_mult',
    [
      [1, 2],
      [3, 4]
    ],
    [
      [1, 2],
      [3, 4]
    ]
  ),
  [
    [1, 4],
    [9, 16]
  ]
);

operation("m_div",[[1,2],[3,4]],[[1,2],[3,4]]) should return [[1,1],[1,1]].

assert.deepEqual(
  operation(
    'm_div',
    [
      [1, 2],
      [3, 4]
    ],
    [
      [1, 2],
      [3, 4]
    ]
  ),
  [
    [1, 1],
    [1, 1]
  ]
);

operation("m_exp",[[1,2],[3,4]],[[1,2],[3,4]]) should return [[1,4],[27,256]].

assert.deepEqual(
  operation(
    'm_exp',
    [
      [1, 2],
      [3, 4]
    ],
    [
      [1, 2],
      [3, 4]
    ]
  ),
  [
    [1, 4],
    [27, 256]
  ]
);

operation("m_add",[[1,2,3,4],[5,6,7,8]],[[9,10,11,12],[13,14,15,16]]) should return [[10,12,14,16],[18,20,22,24]].

assert.deepEqual(
  operation(
    'm_add',
    [
      [1, 2, 3, 4],
      [5, 6, 7, 8]
    ],
    [
      [9, 10, 11, 12],
      [13, 14, 15, 16]
    ]
  ),
  [
    [10, 12, 14, 16],
    [18, 20, 22, 24]
  ]
);

--seed--

--seed-contents--

function operation(op, arr1, arr2) {

}

--solutions--

function operation(op, arr1, arr2) {
  const ops = {
    add: ((a, b) => a + b),
    sub: ((a, b) => a - b),
    mult: ((a, b) => a * b),
    div: ((a, b) => a / b),
    exp: ((a, b) => Math.pow(a, b))
  };
  const ifm = op.startsWith('m');
  const doOp = ops[op.substring(2)];
  for (let i = 0; i < arr1.length; i++) {
    for (let j = 0; j < arr1[0].length; j++) {
      arr1[i][j] = doOp(arr1[i][j], (ifm) ? (arr2[i][j]) : (arr2));
    }
  }
  return arr1;
}