初识前缀树 详细的记录一下各点!
1.节点数:单词数*单词长
2.节点定义,初始化,插入单词,询问
下面模拟一下前缀树的建立过程
看下面建好的前缀树:
插入911时我们记录三号结点有单词生成
插入972625999时我们记录11号结点有单词生成
插入91125426时我们发现3号结点已经有单词生成,因此是NO
AC代码:
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxnode = 10000*11;
char str[10010][11];
int cnt;
struct Trie{
bool Isword;
int next[15];
}node[maxnode];
void Init(){
cnt = 1;//记录结点数
for(int i=0; i<maxnode; i++){
node[i].Isword = false;
memset(node[i].next, -1, sizeof(node[i].next));
}
}
void Insert(char *s){//p记录结点下标
int p = 0, n = strlen(s);
for(int i=0; i<n; i++){
if(node[p].next[s[i]-'0'] == -1)//加入新的结点
node[p].next[s[i]-'0'] = cnt++;
p = node[p].next[s[i]-'0'];//沿着原来的结点继续向下走
}
node[p].Isword = true;//记录哪个结点有前缀生成
}
bool Query(char *s){
int p = 0, n = strlen(s);
for(int i=0; i<n; i++){
if(node[p].Isword) return 0;
else p = node[p].next[s[i]-'0'];
}
return true;
}
int main(){
int cas; cin >> cas;
while(cas--){
Init();
int n; cin >> n;
int flag = 1;
for(int i=0; i<n; i++){
scanf("%s", str[i]);
Insert(str[i]);
}
for(int i=0; i<n; i++){
if(!Query(str[i])){
flag = 0;
break;
}
}
if(flag) puts("YES");
else puts("NO");
}
return 0;
}