构建字符树

有如下字符数组:"A","B","AB","AC","ABC","ACD"。需要构建如下字符树

 代码如下:

package test;
import com.alibaba.fastjson.JSON;
import org.apache.commons.lang3.StringUtils;
import java.util.*;

public class TestTree {
    public static void main(String[] args) {
        List<String> strList = Arrays.asList("A","B","AB","AC","ABC","ACD");
        Collections.shuffle(strList);

        Node root = new Node("");
        for(String str:strList){
            Node.insertNode(root,str);
        }

        System.out.println(JSON.toJSONString(root));
        System.out.println(JSON.toJSONString(Node.getDataByLayer(root)));
    }

    static class Node{
        //private Node prev;
        private String val;
        private List<Node> subNodes;

        public Node(String val){
            this.val = val;
        }
        /*public Node getPrev() {
            return prev;
        }

        public void setPrev(Node prev) {
            this.prev = prev;
        }*/

        public String getVal() {
            return val;
        }

        public void setVal(String val) {
            this.val = val;
        }

        public List<Node> getSubNodes() {
            if(this.subNodes == null){
                this.subNodes = new ArrayList<Node>();
            }
            return subNodes;
        }

        public void setSubNodes(List<Node> subNodes) {
            this.subNodes = subNodes;
        }

        @Override
        public String toString() {
            return "Node{" + "val='" + val + '\'' + '}';
        }

        public static void insertNode(Node node, String val){
            if(StringUtils.isEmpty(val)){
                return;
            }
            if(node.getSubNodes().size() == 0){
                Node nNode = new Node(val);

                node.getSubNodes().add(nNode);
                //nNode.setPrev(node);

                return;
            }
            boolean isContain = false;
            for(int i = node.getSubNodes().size() - 1;i >= 0;i--){
                if(node.getSubNodes().get(i).getVal().contains(val)){  //val被包含
                    isContain = true;

                    Node nNode = new Node(val);
                    Node curNode = node.getSubNodes().get(i);

                    node.getSubNodes().set(i,nNode);
                    //nNode.setPrev(node);
                    //curNode.setPrev(nNode);
                    nNode.getSubNodes().add(curNode);
                }else if(val.contains(node.getSubNodes().get(i).getVal())){ //val包含
                    isContain = true;
                    insertNode(node.getSubNodes().get(i),val);
                }
            }
            if(!isContain){
                Node nNode = new Node(val);

                node.getSubNodes().add(nNode);
                //nNode.setPrev(node);
            }
        }
        //按照层获取数据
        public static Collection<String> getDataByLayer(Node node){
            Deque<Node> queue=new ArrayDeque<>();
            queue.add(node);

            Collection<String> col = new LinkedHashSet<>();
            while(!queue.isEmpty()){
                Node temp = queue.remove();
                if(!StringUtils.isEmpty(temp.getVal())){
                    col.add(temp.getVal());
                }

                queue.addAll(temp.getSubNodes());
            }

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值