<讲解链接>
http://dongxicheng.org/structure/trietree/
<板子>
题目来源:
HUD 1671
题目描述:
给出一份电话号码列表,如果不存在有一个号码是另一个号码的前缀,我们就说这份电话号码列表是合法的。让我们看看如下号码列表:
1. Emergency 911
2. Alice 97625999
3. Bob 91125426
在这组号码中,我们不能拨通 Bob 的电话,因为当你按下 Bob 电话号码的前 3 个数字“911”时,电话局会把你的拨号连接到 Emergency 的线路。
所以这组号码是不合法的。
输入格式:
有多组输入数据。
第一行输入一个正整数 t(1<=t<=40),表示数据组数。
每组数据第一行是一个正整数 n(1<=n<=10000),表示电话号码的数量。
接下来有 n 行,每行一个电话号码,每个电话号码是不超过 10 位的连续数字。
输出格式:
对每组数据,如果电话号码列表合法,则输出“YES”,不合法则输出“NO”。
样例数据 1
输入:
2
3
911
97625999
91125426
5
113
12340
123440
12345
98346
输出:
NO
YES
代码:
#include<bits/stdc++.h>
using namespace std;
int t,n;
int flag,cnt,tot;
int trie[100015][15],num[100015],a[100015];
string s;
inline void add(string s)
{
int x=0;
for(int i=0;i<s.size();++i)
{
if(!trie[x][s[i]-'0'])
trie[x][s[i]-'0']=++tot;
x=trie[x][s[i]-'0'];
++num[x];
}
a[++cnt]=x;
}
int main()
{
cin>>t;
while(t--)
{
flag=true,cnt=0,tot=0;
memset(trie,0,sizeof(trie));
memset(num,0,sizeof(num));
cin>>n;
for(int i=1;i<=n;++i)
cin>>s,add(s);
for(int i=1;i<=n;++i)
if(num[a[i]]>1)
{
flag=false;
cout<<"NO"<<endl;
break;
}
if(flag)
cout<<"YES"<<endl;
}
return 0;
}