poj 3630 找重複前綴 trie樹

題目：點擊打開鏈接

題意：

給你一些電話號碼，如果其中一個電話號碼是另一個電話號碼的前綴，那麼輸出NO，否則YES

分析：

這題就是找最小前綴問題，以前我都是用string然後排序，然後取相等的一段判斷就行。今天主要是學習一下trie樹。

如果是動態節點的話，會TLE，只能初始化數組，但是別忘了一開始trie數組清0.

思路是在建樹的時候，在單詞的最後一個節點標記一些結尾，然後查詢的時候，如果遍歷樹的過程中遇到了這種標記，而且還沒到單詞尾，那麼說明這個單詞有前綴，直接返回0，否則一直遍歷到結尾，返回1.

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int N=1e4+5;
char a[N][11];
struct node
{
    bool ok;
    node *next[11];
    node()
    {
        ok=0;
        memset(next,0,sizeof(next));
    }
}trie[100000]; //最多10000*10個節點
int num;
bool Insert(node *rt,char *s)
{
    int i=0;
    node *p=rt;
    while(s[i]){
        int k=s[i++]-'0';
        if(p->next[k]==NULL)
            p->next[k]=&trie[num++];
        p=p->next[k];
        //if(p->ok)return 0;
    }
    p->ok=1;
    return 1;
}
bool Search(node *rt,char *s)
{
    int i=0;
    node *p=rt;
    while(s[i]){
        int k=s[i++]-'0';
        p=p->next[k];
        if(p->ok&&s[i])return 0;
    }
    //if(cnt>1)return false;
    return 1;
}
int main()
{
    int T;
    //freopen("f.txt","r",stdin);
    scanf("%d",&T);
    while(T--){
        int n;
        memset(trie,0,sizeof(trie)); //別忘了
        node *rt=&trie[0];
        num=1;
        scanf("%d",&n);
        for(int i=0;i<n;i++){
            scanf("%s",a[i]);
            Insert(rt,a[i]);
        }
        bool flag=true;
        for(int i=0;i<n;i++){
           // cout<<i<<endl;
            if(Search(rt,a[i]))continue;
            flag=false;break;
        }
        if(flag)printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}