一段小代码, 计算树状节点的总数

使用zTree来显示文件树, 数据格式类似于:

var data = {
    id: 1,
    children: [
        {
            id: 2,
            children: [
                {
                    id: 3,
                    children: [
                        {
                            id: 8,
                            children: []
                        },
                    ]
                },
                {
                    id: 5,
                    children: []
                }
            ]
        },
        {
            id: 4,
        },
        {
            id: 6,
            children: []
        },
        {
            id: 7,
            children: []
        }
    ],
}

 

 

一则小问题, 如何计算节点总数.

当然是递归了那么, 思路就是 total = 1 + iter(children),  1代表节点本身.

核心代码3行就够了.

function counter(data) {
    // total = self + count(children);
    if (typeof data.children === "undefined") {
        return 1;
    }
    return 1 + data.children.reduce(function (prev, item) {
        return prev + counter(item);
    }, 0);
}