字典树--java 定义

 
闲着写了一个字典树的模板,包括:生成字典树,查询某个字符串(前缀)是否出现以及次数,打印某个前缀所涉及的字符串等功能
package aaa;

import java.util.*;

public class Main {
    static Scanner in = new Scanner(System.in);
    static Node root;
    static int n;
	public static void main(String[] args) {
		n = in.nextInt();
		String s;
		root = new Node();
		for(int i = 0; i < n;i++) {
			s = in.next();
			root.insert(root, s);
		}
	   String str = in.next();
	   root.printStr(root, str);
	}

}
class Node{
	static int kind = 0;
	int num;//出现次数
	int id;//在树中的数字索引
	boolean isend;//是否到了单词的结束
	Node[] son;//儿子节点
	char val;
	Node(){
		id = 0;
		num = 0;
		son = new Node[26];
		isend = false;
	}
	int insert(Node root,String str) {
		Node t = root;
		int k;
		for(int i = 0; i < str.length();i++) {
			k = str.charAt(i)-'a';
			if(t.son[k]==null) {
				t.son[k]= new Node();
				t.son[k].val = str.charAt(i);
			}
			t.son[k].num++;
			t = t.son[k];
		}
		if(t.isend)
		    return t.id;
		else {
			t.isend = true;
			t.id = kind++;
			return t.id;
		}
		
	 }
    int conut(Node root,String str) {
    	int k;
    	Node t = root;
    	for(int i = 0;i < str.length();i++) {
    		k = str.charAt(i)-'a';
    		if(t.son[k]==null) return 0;
    		else	
    			t = t.son[k];
        }
       return t.num;
    }
    void printStr(Node root,String prefix) {
    	int k;
    	Node t = root;
    	for(int i = 0;i < prefix.length();i++) {
    		k = prefix.charAt(i)-'a';
    		if(t.son[k]==null) return;
    		else	
    			t = t.son[k];
        }
    	preTraverse(t,prefix);
    	return ;
    }
    void preTraverse(Node root,String prefix) {
    	if(root.isend) {
    		 System.out.println(prefix);
    		 return ;
    	  }
        for (Node child : root.son){
               if (child!=null){
                   preTraverse(child, prefix+child.val);
             }
         }
    }
}


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值