js陣列去重、扁平化函式

墨城發表於2021-12-11

1、陣列扁平化(又稱陣列降維)

flat() 方法會按照一個可指定的深度遞迴遍歷陣列,並將所有元素與遍歷到的子陣列中的元素合併為一個新陣列返回

const test = ["a", ["b", "c"], ["d", ["e", ["f"]], "g"]]

// flat不傳引數時,預設扁平化一層
test.flat()
// ["a", "b", "c", "d", ["e", ["f"]], "g"]

// flat傳入一個整數引數,整數即扁平化的層數
test.flat(2)
// ["a", "b", "c", "d", "e", ["f"], "g"]

// Infinity 關鍵字作為引數時,無論多少層巢狀,都會轉為一維陣列
test.flat(Infinity)
// ["a", "b", "c", "d", "e", "f", "g"]

// 傳入 <=0 的整數將返回原陣列,不扁平化
test.flat(0)
test.flat(-1)
// ["a", ["b", "c"], ["d", ["e", ["f"]], "g"]]

// 如果原陣列有空位,flat()方法會跳過空位。
["a", "b", "c", "d",,].flat()
// ["a", "b", "c", "d"]

方法一:使用 reduce 方法

一次性扁平化所有

function flattenDeep(arr) { 
    return Array.isArray(arr)
      ? arr.reduce( (acc, cur) => [...acc, ...flattenDeep(cur)] , [])
      : [arr]
}

// 測試
var test = ["a", ["b", "c"], ["d", ["e", ["f"]], "g"]]
flattenDeep(test)
// ["a", "b", "c", "d", "e", "f", "g"]

實現 flat 函式

function flat(arr, depth = 1) {
    return depth > 0
        ? arr.reduce((acc, cur) => {
        if(Array.isArray(cur)) {
            return [...acc, ...flat(cur, depth-1)]
        }
        return [...acc, cur]
    } , [])
      : arr
}

// 測試
var test = ["a", ["b", "c"], ["d", ["e", ["f"]], "g"]]
// 不傳引數時,預設扁平化一層
flat(test)
// ["a", "b", "c", "d", ["e", ["f"]], "g"]

// 傳入一個整數引數,整數即扁平化的層數
flat(test, 2)
// ["a", "b", "c", "d", "e", ["f"], "g"]

// Infinity 關鍵字作為引數時,無論多少層巢狀,都會轉為一維陣列
flat(test, Infinity)
// ["a", "b", "c", "d", "e", "f", "g"]

// 傳入 <=0 的整數將返回原陣列,不扁平化
flat(test, 0)
flat(test, -10)
// ["a", ["b", "c"], ["d", ["e", ["f"]], "g"]];

// 如果原陣列有空位,flat()方法會跳過空位。
var arr = ["a", "b", "c", "d",,]
flat(arr)
// ["a", "b", "c", "d"]

方法二:棧

一次性降維所有

function flattenDeep(arr) {
  const result = [] 
  // 將陣列元素拷貝至棧,直接賦值會改變原陣列
  const stack = [...arr]
  // 如果棧不為空,則迴圈遍歷
  while (stack.length !== 0) {
    const val = stack.pop() 
    if (Array.isArray(val)) {
      // 如果是陣列再次入棧,並且展開了一層
      stack.push(...val) 
    } else {
      // 如果不是陣列,就用頭插法插入到結果陣列中
      result.unshift(val)
    }
  }
  return result
}

// 測試
var test = ["a", ["b", "c"], ["d", ["e", ["f"]], "g"]]
flattenDeep(animals)
// ["a", "b", "c", "d", "e", "f", "g"]

2、陣列去重

方式一:Set(ES6)

function unique(arr) {
    return Array.from(new Set(arr))
}
// 或者
var unique = arr => [...new Set(arr)]

// 測試
var arr = [1, 2, 2, 3]
unique(arr); // [1, 2, 3]

方式二:reduce

function unique (arr) {
    return arr.sort().reduce((acc, cur) => {
     if (acc.length === 0 || acc[acc.length - 1] !== cur) {
         acc.push(cur);
     }
     return acc
 }, [])}
;

// 測試
var arr = [1, 2, 2, 3]
unique(arr); // [1, 2, 3]

方法三:filter

function unique(arr) { 
    return arr.filter( (element, index, array) => {
     return array.indexOf(element) === index
 })
}

// 測試
var arr = [1, 2, 2, 3]
unique(arr); // [1, 2, 3]

相關文章