算法 - js 实现对象扁平化

620 阅读1分钟

题目:实现一个 flatten 函数,实现对象的扁平化

// 实现一个 flatten 函数,实现如下的转换功能
const obj = {
    a: 1,
    b: [1, 2, { c: true }],
    c: { e: 2, f: 3 },
    g: null,
};
// 输出
let objRes = {
    a: 1,
    "b[0]": 1,
    "b[1]": 2,
    "b[2].c": true,
    "c.e": 2,
    "c.f": 3,
    // g: null, //null或undefined 都删去
};

解决方法 难点在于注意key值的传递,还有就是递归的思想


const flatten = (obj) => {
    let result = {}
    let process = (key, value) => {
        // 首先判断是基础数据类型还是引用数据类型
        if (Object(value) !== value) {
            // 基础数据类型
            if (value !== null && value !== undefined) {
                result[key] = value
            }
        } else if (Array.isArray(value)) {
            // 数组类型
            for (let i = 0; i < value.length; i++) {
                process(`${key}[${i}]`, value[i])
            }
            if (value.length === 0) {
                result[key] = []
            }
        } else {
            // 对象类型
            let objArr = Object.keys(value)
            objArr.forEach((item) => {
                process(key ? `${key}.${item}` : `${item}`, value[item])
            })
            if (objArr.length === 0 && key) {
                result[key] = {}
            }
        }
    }
    process('', obj)
    return result
}
flatten(obj)