使用for循环代替递归实现树结构数据

大深度的树结构数据容易导致堆的内存溢出

package org.amaidan.common.utils;

import cn.hutool.system.SystemUtil;
import com.alibaba.fastjson.JSON;
import lombok.Data;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 树结构数据处理
 * @author Joen
 * @since 2023-03-28 10:19:01
 */
public class TreeUtils<T> {
    public static void main(String[] args) {
        AtomicInteger i = new AtomicInteger(10);
        TreeUtils<Test1> utils = TreeUtils.getInstance(seed -> {
            int i1 = i.getAndDecrement();
            if (i1 > 0){
                i.set(i1 - 1);
                List<Test1> test1s = new ArrayList<>();
                for (int j = 11 - i1;j > 0;j--){
                    Test1 test1 = new Test1();
                    test1.setId(SystemUtil.getCurrentPID());
                    test1s.add(test1);
                }
                return test1s;
            }
            return null;
        }, Test1::setTest1s);

        System.out.println(JSON.toJSONString(utils.tree(null)));
    }

    private U<T> u;
    private C<T> c;

    public static <T> TreeUtils<T> getInstance(U<T> u, C<T> c) {
        TreeUtils<T> utils = new TreeUtils<>();
        utils.u = u;
        utils.c = c;
        return utils;
    }

    public List<T> tree(T seed) {
        List<T> list = u.iti(seed);
        T curve;
        Cache<T> curveCa = new Cache<>(list, null);
        for (; ; ) {
            if (curveCa == null || (curve = curveCa.next()) == null) {
                return list;
            }
            List<T> iti = u.iti(curve);
            c.setChild(curve, iti);
            if (iti != null && iti.size() > 0) {
                curveCa = new Cache<>(iti,curveCa);
            }else {
                curveCa = curveCa.getParent();
            }
        }
    }

    @Data
    public static class Test1 {
        private Long id;
        private List<Test1> test1s;
    }

    @Data
    public static class Cache<T> {
        public Cache(List<T> list, Cache<T> parent) {
            this.list = list;
            this.parent = parent;
        }

        private List<T> list;

        private Integer index = 0;

        public T next() {
            if (list == null || list.size() == index){
                return null;
            }
            return list.get(index++);
        }

        private Cache<T> parent;
    }

    @FunctionalInterface
    public interface U<T> {
        List<T> iti(T seed);
    }

    @FunctionalInterface
    public interface C<T> {
        void setChild(T t, List<T> list);
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值