Javascript系列——对象元素的数组去重实现

概要

这是一篇记录文,记录数组操作对象去重的实现。

需求

有这样一个数组


[{
    _id: 123,
    name: '张三'
},{
    _id: 124,
    name: '李四'
},{
    _id: 123,
    name: '张三'
}]

实际上我们只需要


[{
    _id: 123,
    name: '张三'
},{
    _id: 124,
    name: '李四'
}]

去重

简单数组的去重


Array.from(new Set([1,1,2,3,4,4])) // [1,2,3,4]

以对象为元素的数组去重

和数组相关的算法多种多样,在你以为自己已经掌握数组之后,会发现很多和数组相关的算法仍旧很复杂。

下面我将讲述一个入门等级的数组算法,解决上面提出的需求。

1、定义一个函数removeRepeat,它需要传入2个参数,arr表示需要去重的数组,field表示需要比较的key。比如我们的需求是比较 _id 是否有重复。


function removeRepeat(arr, field){
    return arr
}

2、需要一个空数组,用来存储每个对象元素中field对应的value。


let s = []
for(let v of arr){
    s.push(v[field])
}
//s = [123, 124, 123]

3、将所有field的值存到数组之后,它们的下标一一对应原数组的下标(这点很重要),接着我们需要2个对象集合,result用来存储s里遍历出来的元素,如果有重复,就将重复的元素丢到reSet对象里面。


let result = {}, reSet = {}
for(let i=0,len=s.length;i<len;i++){
    if(!result[s[i]] && result[s[i]] !== 0) {
    //如果result不存在当前的key并且它不为0时
       result[s[i]] = i
    } else {
      reSet[s[i]] = i
    }
}
// result = {123: 0, 124: 1} 这是去重重复后的元素
// reSet = {123: 2} 我们将重复的元素123作为key,它的下标2作为value。

4、上一步得到了result和reSet2个对象,那么,我们该用哪个对象来处理原数组的去重呢?只需要reSet,reSet记录了要去重的对象所在的下标,那么可以直接用splice干掉它。


for(let key in reSet){
    arr.splice(reSet[key], 1)
}
/* 
arr = [{
    _id: 123,
    name: '张三'
},{
    _id: 124,
    name: '李四'
}]
*/

5、说明

关键的第3和4步,都是用对象来处理,这样做的好处是时间复杂度可以达到O(1),如果用数组来保存,则需要2个for循环,时间复杂度变成了O(n²)。

完整源码


function removeRepeat(arr, field){
    let s = [], result = {}, reSet = {}
    for(let v of arr){
        s.push(v[field])
    }
    for(let i=0,len=s.length;i<len;i++){
        if(!result[s[i]] && result[s[i]] !== 0) {
           result[s[i]] = i
        } else {
          reSet[s[i]] = i
        }
    }
    for(let key in reSet){
        arr.splice(reSet[key], 1)
    }
    return arr
}

// removeRepeat(arr, '_id')

补充

受到各路大神的解法影响,我也针对上面代码的不足做了修改。

1、更简洁的代码。

2、支持多个重复对象的去重,缺点是会改变原来的排序。


const removeRepeat = (arr, field) => {
    let s = {}
    for(let i=0,len=arr.length;i<len;i++){
        s[arr[i][field]] = arr[i]
    }
    return Object.values(s)
}

总结

数组还有各种有趣的操作,也经常作为各大公司考察的题型之重,多练习和数组相关的算法会很有帮助。

原文地址:https://segmentfault.com/a/1190000012873968

猜你喜欢

转载自www.cnblogs.com/lalalagq/p/9959426.html
今日推荐