Question

2797. Partial Function with Placeholders

中文文档

Description

Given a function fn and an array args, return a function partialFn. 

Placeholders "_" in the args should be replaced with values from restArgs starting from index 0. Any remaining values in the restArgs should be added at the end of the args.

partialFn should return a result of fn. fn should be called with the elements of the modified args passed as separate arguments.

 

Example 1:

Input: fn = (...args) => args, args = [2,4,6], restArgs = [8,10]
Output: [2,4,6,8,10]
Explanation: 
const partialFn = partial(fn, args)
const result = partialFn(...restArgs) 
console.log(result) // [2,4,6,8,10]

There are no placeholders "_" in args therefore restArgs is just added at the end of args. Then the elements of the args are passed as separate arguments to fn, which returns passed arguments as an array.

Example 2:

Input: fn = (...args) => args, args = [1,2,"_",4,"_",6], restArgs = [3,5]
Output: [1,2,3,4,5,6]
Explanation: 
const partialFn = partial(fn, args) 
const result = partialFn(...restArgs) 
console.log(result) // [1,2,3,4,5,6] 

Placeholders "_" are replaced with values from the restArgs. Then the elements of the args are passed as separate arguments to fn, which returns passed arguments as an array.

Example 3:

Input: fn = (a, b, c) => b + a - c, args = ["_", 5], restArgs = [5, 20]
Output: -10
Explanation: 
const partialFn = partial(fn, args)
const result = partialFn(...restArgs)
console.log(result) // -10

Placeholder "_" is replaced with 5 and 20 is added at the end of args. Then the elements of the args are passed as separate arguments to fn, which returns -10 (5 + 5 - 20).

 

Constraints:

• fn is a function
• args and restArgs are valid JSON arrays
• 1 <= args.length <= 5 * 104
• 1 <= restArgs.length <= 5 * 104
• 0 <= number of placeholders <= restArgs.length

Solutions

Solution 1

function partial(fn: Function, args: any[]): Function {
  return function (...restArgs) {
    let i = 0;
    for (let j = 0; j < args.length; ++j) {
      if (args[j] === '_') {
        args[j] = restArgs[i++];
      }
    }
    while (i < restArgs.length) {
      args.push(restArgs[i++]);
    }
    return fn(...args);
  };
}

/**
 * @param {Function} fn
 * @param {Array} args
 * @return {Function}
 */
var partial = function (fn, args) {
  return function (...restArgs) {
    let i = 0;
    for (let j = 0; j < args.length; ++j) {
      if (args[j] === '_') {
        args[j] = restArgs[i++];
      }
    }
    while (i < restArgs.length) {
      args.push(restArgs[i++]);
    }
    return fn.apply(this, args);
  };
};