高级前端才会做的一道算法题

前言

群友大周还是小周发了几道陌陌的笔试题，在这里我分享其中一道。

题目

实现一个算法getClosest3Nums，从数组中找出3个元素，使这三个元素之和最接近于给定的数n。

输入：0,[1,-1,-2,-3,0,2,1] 
输出：[[1,-1,0],[1,-3,2],...]

解题思路

1. 找出所有的组合
2. 对所有的组合进行分类
3. 根据要求匹配分类

抛砖

/**
 * @function 找出所有的组合（没有去重）
 * @param {Array} arr 
 * @returns {Array} res
 */
const createAll = (arr) => {
    const len = arr.length
    let res = []
    if (len < 3) {
        return res
    }
    for (let i = 0; i < len; i++) {
        for (let j = 0; j < len; j++) {
            if (j == i) {
                continue
            }
            for (let k = 0; k < len; k++) {
                if (k == i || k == j) {
                    continue
                }
                let tempArr = []
                tempArr[0] = arr[i]
                tempArr[1] = arr[j]
                tempArr[2] = arr[k]
                res.push(tempArr)
            }
        }
    }
    return res
}
/**
 * @function 对所有的组合进行分类
 * @param {Array} res 
 * @returns {Object} json
 */
const createJson = (res) => {
    let json = {}
    for (let i = 0; i < res.length; i++) {
        let item = res[i]
        let num = item.reduce((total, currentValue) => {
            total += currentValue
            return total
        })
        if (typeof json[num] === 'undefined') {
            json[num] = [item]
        } else {
            json[num].push(item)
        }
    }
    return json
}
/**
 * @function 用n去匹配出来key
 * @param {Object} json 
 * @param {Number} n 
 */
const createKey = (json, n) => {
    let key
    let keys = Object.keys(json).sort((v1, v2) => {
        return v1 - v2
    })
    let len = keys.length
    for (let i = 0; i < len; i++) {
        if (keys[i] == n) {
            key = n
            break
        }
        if (i > 0) {
            let tempKeyStart = keys[i - 1]
            let tempKeyEnd = keys[i]
            if (tempKeyStart < n && n < tempKeyEnd) {
                if (n - tempKeyStart > tempKeyEnd - n) {
                    key = tempKeyEnd
                } else {
                    key = tempKeyStart
                }
                break
            }
        }
        if (i === len - 1) {
            if (typeof key === 'undefined') {
                if (Math.abs(keys[0] - n) > Math.abs(keys[len - 1] - n)) {
                    key = keys[len - 1]
                } else {
                    key = keys[0]
                }
            }
        }
    }
    return key
}
/**
 * @function 实现一个算法getClosest3Nums，从数组中找出3个元素，使这三个元素之和最接近于给定的数n。
 * @param {Number} n 
 * @param {Array} arr 
 * @returns {Array} array
 */
const getClosest3Nums = (n, arr) => {
    let array = []
    let res = createAll(arr)
    let json = createJson(res)
    let key = createKey(json, n)
    if (typeof json[key] != 'undefined') {
        array = json[key]
    }
    return array
}

//求值
// const result = getClosest3Nums(0, [1, -1, -2, -3, 0, 2, 1])
const result = getClosest3Nums(0, [1, -1, 0])
console.log(result)

引玉

上面的代码只是我抛砖引玉，本人算法水平也有限。

谢谢阅读！