HDU 1671 字典树 查找前缀

题目链接

题意： 若干个电话号码 是否存在某个号码是另一个号码的前缀

将每一个号码都加入一颗字典树中

此时如果这个号码是某个号码的前缀或者某个号码是它的前缀

那么 第一种情况下此号码的节点在某个号码的路径上

第二种情况 在这个号码的路径上存在号码节点

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
const int maxnode = 100000 + 50;
const int sigma_size = 15;

int ch[maxnode][sigma_size];
int value[maxnode];
int get_idx(char x){
return x - '0';
}
int sz;

void init_Trie(){
sz = 0;
memset(value,0,sizeof(value));
memset(ch[0],-1,sizeof(ch[0]));
}

bool Insert(char* num){
int n = strlen(num);
int cur = 0;
for(int i = 0;i < n;++i){
int idx = get_idx(num[i]);
if(ch[cur][idx] == -1){
sz++;
memset(ch[sz],-1,sizeof(ch[sz]));
ch[cur][idx] = sz;
}
cur = ch[cur][idx];
if(value[cur]){
return false;
}
}
value[cur] = 1;
for(int i = 0;i < 10;++i){
if(ch[cur][i] != -1){
return false;
}
}
return true;
}

char N[20];
int main(){
int T;
scanf("%d",&T);
while( T-- ){
init_Trie();
int n;scanf("%d",&n);
bool right = true;
for(int i = 0;i < n;++i){
scanf("%s",N);
if(right){
right = right & Insert(N);
}
}
printf("%s\n",right ? "YES" : "NO");
}
return 0;
}