数据结构——Trie字符串

文章目录

  • Trie 字符串
    • 模板

Trie 字符串

Trie字符串:快速存储字符串集合
那么它是怎么样的一种结构呢??
特点:字母的个数不会很多,why?
数据结构——Trie字符串_第1张图片

数据结构——Trie字符串_第2张图片

来看代码的实现吧
题目
AcWing835 Trie字符串

维护一个字符串集合,支持两种操作:
“I x”向集合中插入一个字符串x;
“Q x”询问一个字符串在集合中出现了多少次。
共有N个操作,输入的字符串总长度不超过 105,字符串仅包含小写英文字母。

数据结构——Trie字符串_第3张图片

模板

	static int[][] son = new int[N][26];//只存储小写字母
	static int[] cnt = new int[N]; //存取下标
	static int idx; //idx索引,唯一
	
	private static void insert(String s){
     
		//存储一个字符串
		int p = 0;
		for(int i=0;i<s.length();i++){
     
			int u = s.charAt(i) - 'a';//存个差值就ok了
			if(son[p][u]==0){
     
				//如果子节点没有,就新增
				son[p][u] = ++idx;
			}
			
			p = son[p][u];//下一层		
		}
		
		//存储完毕后,记录下标
		cnt[p]++;
	}
	
	private static int query(String s){
     
		//查询一个字符串
		int p = 0;
		for(int i=0;i<s.length();i++){
     
			int u = s.charAt(i)-'a';
			if(son[p][u]==0){
     
				return 0;
			}
			p = son[p][u];
		}
		
		return cnt[p];
	}

你可能感兴趣的:(算法学习,字符串)