在一个奇怪的村子中,很多人的名字都很长,比如aaaaa, bbb and abababab。
Output
Sample Input
Sample Output
名字这么长,叫全名显然起来很不方便。所以村民之间一般只叫名字的前缀。比如叫'aaaaa'的时候可以只叫'aaa',因为没有第二个人名字的前三个字母是'aaa'。不过你不能叫'a',因为有两个人的名字都以'a'开头。村里的人都很聪明,他们总是用最短的称呼叫人。输入保证村里不会有一个人的名字是另外一个人名字的前缀(作为推论,任意两个人的名字都不会相同)。
如果村里的某个人要叫所有人的名字(包括他自己),他一共会说多少个字母?
Input输入第一行为数据组数T (T<=10)。每组数据第一行为一个整数n(1<=n<=1000),即村里的人数。以下n行每行为一个人的名字(仅有小写字母组成)。输入保证一个村里所有人名字的长度之和不超过1,000,000。
对于每组数据,输出所有人名字的字母总数。
1
3
aaaaa
bbb
abababab
5
利用字典树可以在规定时间内求解
#include<iostream>
#include<cstring>
using namespace std;
#define max 26
struct tree
{
tree *next[max];
int v;
}*root,*t;
int l,i,sum;
char s[1001][1001];
void creat(int k,tree *p)
{
p->v++;
if (k==l) return;
if (p->next[s[i][k]-'a']!=NULL) creat(k+1,p->next[s[i][k]-'a']);
else
{
int m,n;
for (n=k;n<l;n++)
{
t=new tree;
t->v=1;
for (m=0;m<max;m++) t->next[m]=NULL;
p->next[s[i][n]-'a']=t;
p=t;
}
}
}
void count(int k,int i,tree *p)
{
sum++;
if (p->v==1) return;
count(k+1,i,p->next[s[i][k+1]-'a']);
}
void delet(tree *p)
{
int i;
for (i=0;i<max;i++)
if (p->next[i]!=NULL) delet(p->next[i]);
delete p;
}
int main()
{
int T,ans,n;
cin>>T;
while (T--)
{
cin>>n;
root=new tree;
root->v=0;
for (i=0;i<max;i++) root->next[i]=NULL;
for (i=0;i<n;i++)
{
cin>>s[i];
l=strlen(s[i]);
creat(0,root);
}
ans=0;
for (i=0;i<n;i++)
{
sum=0;
count(0,i,root->next[s[i][0]-'a']);
ans=ans+sum;
}
cout<<ans<<endl;
delet(root);
}
}