字符串 I - Trie 字典树

Trie字典树又叫单词查找树，是一种树形结构，可以统计、排序、保存大量字符串

它的优点：利用前缀减少查找时间

struct trie{ //以小写字母Trie为例
	int ch[100010][26]; //ch[i][0~25]表示结点i的孩子编号 ch[i][0]表示子结点A的编号
	int value[100010]; //结点i的值 value
	int sz; //结点个数(开始有一个结点)
	trie() {sz = 1; memset(ch[0], 0, sizeof(ch[0]));}//初始化
	int idx(char c) {return c - 'a';} //字母映射到编号 
	void insert(char *s, int v) { //插入算法
		int len = strlen(s), c, now = 0;
		for(int i=0; i<len; i++) {
			c = idx(s[i]);
			if(!ch[now][c]) {
				memset(ch[sz], 0, sizeof(ch[sz]));
				value[sz] = 0;
				ch[now][c] = sz ++;
			}
			now = ch[now][c];
		}
		value[now] = v; //
	}
	int find(char *c, int Node) { //查询算法。调用时Node应为0(根结点)
		int len = strlen(c), next = ch[Node][idx(c[0])];
		if(len == 1) return value[next] == 0 ? -1 : value[next];
		return find(&c[1], next);
	}
} Trie;

例1 Luogu2580 于是他错误的点名开始了

题目背景

XS中学化学竞赛组教练是一个酷爱炉石的人。

他会一边搓炉石一边点名以至于有一天他连续点到了某个同学两次，然后正好被路过的校长发现了然后就是一顿欧拉欧拉欧拉（详情请见已结束比赛CON900）。

题目描述

这之后校长任命你为特派探员，每天记录他的点名。校长会提供化学竞赛学生的人数和名单，而你需要告诉校长他有没有点错名。（为什么不直接不让他玩炉石。）

输入输出格式

输入格式：

第一行一个整数 n，表示班上人数。接下来 n 行，每行一个字符串表示其名字（互不相同，且只含小写字母，长度不超过 50）。第 n+2 行一个整数 m，表示教练报的名字。接下来 m 行，每行一个字符串表示教练报的名字（只含小写字母，且长度不超过 50）。

输出格式：

对于每个教练报的名字，输出一行。如果该名字正确且是第一次出现，输出“OK”，如果该名字错误，输出“WRONG”，如果该名字正确但不是第一次出现，输出“REPEAT”。（均不加引号）

输入输出样例

输入样例#1： 

5  
a
b
c
ad
acd
3
a
a
e

输出样例#1： 

OK
REPEAT
WRONG

说明

对于 40%的数据，n≤1000,m≤2000；

对于 70%的数据，n≤10000,m≤20000；

对于 100%的数据， n≤10000，m≤100000。

【解法】字典树。根据结点的值可以判断是否已经点名或未点名或错误.

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;

struct trie {
	int ch[500010][26];
	short value[500010];
	int sz;
	trie() {
		sz = 1;
		memset(ch[0], 0, sizeof(ch[0]));
	}
	int idx(char c) {return c - 'a';}
	void insert(char *s) {
		int len = strlen(s), c, now = 0;
		for(int i=0; i<len; i++) {
			c = idx(s[i]);
			if(!ch[now][c]) {
				memset(ch[sz], 0, sizeof(ch[sz]));
				value[sz] = 1;
				ch[now][c] = sz ++;
			}
			now = ch[now][c];
		}
		value[now] = 2;
	}
	int find(char *c, int Node) {
		int len = strlen(c), next = ch[Node][idx(c[0])];
		if(len == 1) {
			if(value[next] < 2) return -1;
			return value[next] == 2 ? value[next] ++ : 3;
		}
		return find(&c[1], next);
	}
} Trie;

int main() {
	char s[100];
	int value, n, q;
	scanf("%d", &n);
	for(int i=1; i<=n; i++) {
		scanf("%s", s);
		Trie.insert(s);
	}
	scanf("%d", &q);
	for(int i=1; i<=q; i++) {
		scanf("%s", s);
		value = Trie.find(s, 0);
		if(value == 2) printf("OK\n");
		else if(value == 3) printf("REPEAT\n");
		else printf("WRONG\n");
	}
}