I-最短的名字
在一个奇怪的村子中,很多人的名字都很长,比如aaaaa, bbb and abababab。
名字这么长,叫全名显然起来很不方便。所以村民之间一般只叫名字的前缀。比如叫’aaaaa’的时候可以只叫’aaa’,因为没有第二个人名字的前三个字母是’aaa’。不过你不能叫’a’,因为有两个人的名字都以’a’开头。村里的人都很聪明,他们总是用最短的称呼叫人。输入保证村里不会有一个人的名字是另外一个人名字的前缀(作为推论,任意两个人的名字都不会相同)。
如果村里的某个人要叫所有人的名字(包括他自己),他一共会说多少个字母?
Input
输入第一行为数据组数T (T<=10)。每组数据第一行为一个整数n(1<=n<=1000),即村里的人数。以下n行每行为一个人的名字(仅有小写字母组成)。输入保证一个村里所有人名字的长度之和不超过1,000,000。
Output
对于每组数据,输出所有人名字的字母总数。
Sample Input
1
3
aaaaa
bbb
abababab
Sample Output
5
分析:
正确的解题姿势是字典树,但是题目的时间限制是5s,可以用map划船过去,这也是让我很不理解的事,为什么好多前缀后缀的题时间限制都那么长;
代码:
字典树:
#include<iostream> #include<cstring> #include<cstdio> #include<string> #include<map> using namespace std; const int N = 1000 + 5; const int MaxSize = 10000 + 5; char name[N][N]; int ch[MaxSize][N]; int val[MaxSize],sz; int lx(char c){return c-'a';} void Insert(char *str){ int len = strlen(str),u=0,c; for(int i=0;i<len;i++){ c=lx(str[i]); if(!ch[u][c]) ch[u][c] = sz++; u = ch[u][c]; val[u]++; } } int Query(char *str){ int len = strlen(str),u=0,c; for(int i=0;i<len;i++){ c = lx(str[i]); u = ch[u][c]; if(val[u] == 1) return i+1; } return 0; } void Init(){ memset(ch,0,sizeof(ch)); memset(val,0,sizeof(val)); sz = 1; } int main(){ int T; scanf("%d",&T); while(T--){ int n; Init(); scanf("%d",&n); for(int i=0;i<n;i++){ scanf("%s",name[i]); Insert(name[i]); } int sum=0; for(int i=0;i<n;i++) sum+=Query(name[i]); printf("%d\n",sum); } }
map:
#include<iostream> #include<cstring> #include<cstdio> #include<string> #include<map> using namespace std; const int N = 1000 + 5; char name[N][N]; map<string,int> mp; void Insert(char *str){ int len = strlen(str); for(int i=1;i<=len;i++){ char c = str[i]; str[i] = '\0'; mp[str]++; str[i] = c; } } int Query(char *str){ int len = strlen(str); for(int i=1;i<=len;i++){ char c = str[i]; str[i] = '\0'; if(mp[str] == 1) return i; str[i] = c; } } int main(){ int T; string str; scanf("%d",&T); while(T--){ int n; mp.clear(); scanf("%d",&n); for(int i=0;i<n;i++){ cin >> name[i]; Insert(name[i]); } int sum = 0; for(int i=0;i<n;i++) sum += Query(name[i]); printf("%d\n",sum); } }