Java树节点遍历变成迭代器

public class Node {
    final Map<String, Node> valueMap = new HashMap<>();
}
public Stream<Node> stream(Order order) {
        //valueMap is children
        Stream<Node> childrenStream = this.valueMap.values().stream().map(i -> i.stream(order)).reduce(Stream::concat).orElse(Stream.empty());
        //this is parent
        switch (order) {
            case PRE:
                return Stream.concat(Collections.singleton(this).stream(), childrenStream);
            case SUFFIX:
                return Stream.concat(childrenStream, Collections.singleton(this).stream());
            default:
                throw new UnsupportedOperationException();
        }
    }

因为children没有区分左右,是多个的情况,所以没有中缀遍历,可以自行根据情况实现

   public Stream<Node> stream(Order order){
      return   stream((i)->true,order);
    }
    public Stream<Node> stream(Predicate<Node> predicate,Order order) {
        //valueMap is children
        Stream<Node> childrenStream = this.valueMap.values().stream().filter(predicate).map(i -> i.stream(predicate,order)).reduce(Stream::concat).orElse(Stream.empty());
        //this is parent
        if (predicate.test(this)){
            switch (order) {
                case PREFIX:
                    return Stream.concat(Collections.singleton(this).stream(), childrenStream);
                case SUFFIX:
                    return Stream.concat(childrenStream, Collections.singleton(this).stream());
                default:
                    throw new UnsupportedOperationException();
            }
        }else {
            return childrenStream;
        }
    }
//过滤特定节点,以及后代

 

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

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值