Trie是一种高级的数据结构,用来统计多个字符串的查找,由于查找方式和内容和字典很想,所以被称为字典树
所以——只要你会查字典,你就会trie
来看一张图
我们现在这里有一个字典如上图,那么在字典里我们想把一个词给插进去,要怎么做呢,比如我们插入“清华园”,那么我们就先找“清”字,再找“华”,但是没有园,所以我们在华下面连到园
所以trie的插入和这个差不多,每次去找下一个字符,如果没有就建一个点
上代码:
inline void add(char s[]){
int len=strlen(s);
int u=1;
for(int i=0;i<len;i++){
int fin=s[i]-'a';
if(!trie[u][fin])trie[u][fin]=++num;
u=trie[u][fin];
}
end[u]++;
}
查询也很简单
inline int ask(int* s){
int u=1;
int ans=0;
Rep(i,1,s[0]){
if(!trie[u][s[i]])return ans;
u=trie[u][s[i]];
ans+=end[u];
}
return ans-end[u]+cnt[u];
}
因为每个题要查询的内容不太一样,所以trie的写法也不一样
对于这道题P2922 [USACO08DEC]秘密消息Secret Message
大概是这个样子的(结构体版)
struct dictionary_tree{
int trie[N][2];
int end[N],cnt[N];
int sum;
dictionary_tree(){
memset(trie,0,sizeof(trie));
memset(end,0,sizeof(end));
memset(cnt,0,sizeof(end));
}
inline void add(int* s){
int u=1;
Rep(i,1,s[0]){
if(!trie[u][s[i]])trie[u][s[i]]=++sum;
u=trie[u][s[i]];
cnt[u]++;
}
end[u]++;
}
inline int ask(int* s){
int u=1;
int ans=0;
Rep(i,1,s[0]){
if(!trie[u][s[i]])return ans;
u=trie[u][s[i]];
ans+=end[u];
}
return ans-end[u]+cnt[u];
}
}Trie;