js5種方式實現數組扁平化

數組扁平化概念

數組扁平化是指將一個多維數組變為一維數組

[1, [2, 3, [4, 5]]]  ------>    [1, 2, 3, 4, 5]

實現

1. reduce

遍歷數組每一項，若值為數組則遞歸遍歷，否則concat。

function flatten(arr) {  
    return arr.reduce((result, item)=> {
        return result.concat(Array.isArray(item) ? flatten(item) : item);
    }, []);
}

reduce是數組的一種方法，它接收一個函數作為累加器，數組中的每個值（從左到右）開始縮減，最終計算為一個值。

reduce包含兩個參數：回調函數，傳給total的初始值

// 求數組的各項值相加的和： 
arr.reduce((total, item)=> {  // total為之前的計算結果，item為數組的各項值
    return total + item;
}, 0);

2. toString & split

調用數組的toString方法，將數組變為字符串然后再用split分割還原為數組

function flatten(arr) {
    return arr.toString().split(',').map(function(item) {
        return Number(item);
    })
} 

因為split分割后形成的數組的每一項值為字符串，所以需要用一個map方法遍歷數組將其每一項轉換為數值型

3. join & split

和上面的toString一樣，join也可以將數組轉換為字符串

function flatten(arr) {
    return arr.join(',').split(',').map(function(item) {
        return parseInt(item);
    })
}

4. 遞歸

遞歸的遍歷每一項，若為數組則繼續遍歷，否則concat

function flatten(arr) {
    var res = [];
    arr.map(item => {
        if(Array.isArray(item)) {
            res = res.concat(flatten(item));
        } else {
            res.push(item);
        }
    });
    return res;
}

5. 擴展運算符

es6的擴展運算符能將二維數組變為一維

[].concat(...[1, 2, 3, [4, 5]]);  // [1, 2, 3, 4, 5]

根據這個結果我們可以做一個遍歷，若arr中含有數組則使用一次擴展運算符，直至沒有為止。

function flatten(arr) {
    while(arr.some(item=>Array.isArray(item))) {
        arr = [].concat(...arr);
    }
    return arr;
}

總結

雖然說寫了5種方法，但是核心也只有一個：

遍歷數組arr，若arr[i]為數組則遞歸遍歷，直至arr[i]不為數組然后與之前的結果concat。