HDU 1671 Phone List Trie樹

這題是前幾天trie樹專題裏的一道練習題，那天沒有做對。今天重新拿出來，發現是cmp函數寫錯了。直接用了-strcmp導致排序錯誤纔會WA。改正之後AC

思路就是把給的電話號碼按照反字典序排序，再加入trie樹。判斷是否有串被包含即可。用ok[cur]來判斷一個結點被計算了幾次。要是插入時的末尾結點被計算過了那就說明存在被包含的串。

代碼如下

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int trie[100005][15];
int sz;
int ok[100005];
struct node{
    char num[15];
}te[20005];
int cmp(node a,node b){
    return strcmp(a.num,b.num)>0;
}
void init(){
    memset(trie,0,sizeof(trie));
    memset(ok,0,sizeof(ok));
    sz = 1;
}
int in(char *s) {
    int cur = 1;
    for (int i= 0; s[i]; i++) {
        int c = s[i]-'0';
        if (!trie[cur][c]){
            trie[cur][c]= ++sz;
        }
        cur = trie[cur][c];
        ok[cur] ++;
    }
    if(ok[cur]==1)return 1;
    return 0;
}
int main()
{
    int T;
    cin>>T;
    while(T--){
        int n;
        scanf("%d",&n);
        init();
        int t = 1;
        for(int i=0;i<n;i++)
            scanf("%s",te[i].num);
        sort(te,te+n,cmp);
        for(int i=0;i<n;i++){
            if(in(te[i].num)==0)t = 0;
            //cout<<t<<endl;
            if(!t)break;
		}
        if(t)printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}