openjudge 4089:电话号码 字典树

很明显是一道裸的字典树，但是好久不刷，我忘了。。。

这种情况还是在看一下比较好。

大概看了下，自己写的，插入的时候直接查找，找到就return0

#include <iostream>#include <queue>#include <iomanip>#include <vector>#include <queue>#include <map>#include <stack>#include <cstdio>#include <algorithm>#include <cmath>#include <set>#include <sstream>#include <cstring>using namespace std;struct trie{    int ch[100001][11];    int val[100001];    int sz;    void init()    {        sz=0;        memset(ch,0,sizeof(ch));        memset(val,0,sizeof(val));    }    int insert_str(string str)    {        int u=0;        for(int i=0;i<str.size();i++)        {           int index=str[i]-'0';           if(!ch[u][index])           {               ch[u][index]=++sz;               memset(ch[sz],0,sizeof(ch[sz]));           }           else           {               if(val[ch[u][index]])                    return 0;                if(i==str.size()-1)                    return 0;           }           u=ch[u][index];        }        val[u]=1;        return 1;    }};trie T;int main(){    int t;    cin>>t;    for(int i=1;i<=t;i++)    {        int n;        cin>>n;        T.init();        string str;        vector<string> V;        for(int j=1;j<=n;j++)        {            cin>>str;            V.push_back(str);        }        int flag=0;        for(int j=0;j<V.size();j++)        {            if(T.insert_str(V[j])==0)            {                flag=1;                break;            }        }        if(flag)            cout<<"NO"<<endl;        else            cout<<"YES"<<endl;    }    return 0;}