POJ 3630 Phone List（字典树）

题意：检查是否有某一串数字与其他数字的前缀相同。从而打错电话。

题记：先把所有数字字符串存入字典树中，id[p]记录这个字符串的编号，然后第二次将所有字符串存入字典树（相当于检查一次）。

#include
#include
#include
using namespace std;
#define F first
#define S second
#define lowbit(x) ((x)&-(x))
#define endl "\n"
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
typedef pair<int,int> PII;
typedef long long ll;
const int INF=0x3f3f3f3f;
const int N=1e5+10;
int trie[N][15];
//int num[N];
int id[N];
int pos;
bool flag;
string s[10005];
void Insert(string s,int x){
    int len=s.size();
    int p=0;
    for(int i=0;i<len;i++){
        int c=s[i]-'0';
        if(!trie[p][c])
            trie[p][c]=pos++;
        if(id[p]&&id[p]!=x)
            flag=false;
        p=trie[p][c];
        //num[p]++;
    }
    id[p]=x;
}
int main(){
    IOS;
    int T;
    cin>>T;
    while(T--){
        memset(trie,0,sizeof(trie));
        memset(id,0,sizeof(id));
        flag=true;
        int n;
        cin>>n;
        pos=1;
        for(int i=1;i<=n;i++){
            cin>>s[i];
            Insert(s[i],i);
        }
        if(!flag){
            cout<<"NO"<<endl;
            continue;
        }
        for(int i=1;i<=n;i++){
            Insert(s[i],i);
            if(!flag) break;
        }
        if(flag) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
    return 0;
}