题意:检查是否有某一串数字与其他数字的前缀相同。从而打错电话。
题记:先把所有数字字符串存入字典树中,id[p]记录这个字符串的编号,然后第二次将所有字符串存入字典树(相当于检查一次)。
#include<iostream>
#include<cstring>
#include<string>
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;
}