HDU 1671 Phone List (字典树)

#include<bits/stdc++.h>
using namespace std;
int s[100005][26];  //数组字典树（数组开太大会MLE）
int lazy[100005];   //标记字符串最后一个元素的编号
int k;
int flag;
void insert(string &ss){    //构造字典树
    int len=ss.length();
    int p=0;
    for(int i=0;i<len;i++){
        if(!s[p][ss[i]-'0']){
            s[p][ss[i]-'0']=++k;
        }
        p=s[p][ss[i]-'0'];
        if(lazy[p]){    //如果出现被标记的位置则前缀存在 输出NO
            flag=1;
            break;
        }
    }
    lazy[p]++;  //标记最后一个元素
}
vector<string>ve;
void init(){    //初始化
    memset(s,0,sizeof(s));
    memset(lazy,0,sizeof(lazy));
    ve.clear();
    flag=0;
    k=0;
}
int main(){
    ios::sync_with_stdio(false);    //流加速
    int t,n;
    cin>>t;
    while(t--){
        init();
        cin>>n;
        string str;
        for(int i=0;i<n;i++){
            cin>>str;
            ve.push_back(str);
        }
        sort(ve.begin(),ve.end());  //排序解决输入的先后问题
        for(auto i:ve){
            insert(i);
        }
        if(flag)cout<<"NO"<<endl;
        else cout<<"YES"<<endl;
    }
    return 0;
}