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;
}