CSU - 1115 最短的名字(字典树)

点我看题

题意:找出每个字符串的公共前缀的下一个位置,计算这些长度之和.

分析;基础字典树.写的时候一直在re,数组开的太大,但是没有给出name的最长,只能估计了.

参考代码:

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<iostream>

using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
const int maxn = 1e3+5;
const int maxm = 1e6;
int n;
//char name[10005];
char name[1005][10005];
struct TrieTree{
int cnt;//当前字母出现的次数
int child[26];//子树
};
TrieTree tt[500005];
int tsize;
int ans;

void init()
{
tt[0].cnt = 0;
mem(tt[0].child,0);
tsize = 1;
}

void TrieInsert( char name[])
{
int p = 0;
int rt = 0;
while( name[p] != '\0')
{
int id = name[p]-'a';
if( tt[rt].child[id] == 0)//子树结点不存在
{
tt[tsize].cnt = 0;
mem(tt[tsize].child,0);
tt[rt].child[id] = tsize++;
}
tt[rt].cnt++;
rt = tt[rt].child[id];
p++;
}
}
/*
int TrieSearch( int rt, int deep)
{
for( int i = 0; i < 26; i++)
{
if( tt[rt].child[i] != 0)
{
if( tt[rt].cnt == 1)
ans += deep;
else
TrieSearch(tt[rt].child[i],deep+1);
}
}
}
*/

int TrieSearch( char name[])
{
int p = 0;
int rt = 0;
while( name[p] != '\0')
{
int id = name[p]-'a';
if( tt[rt].cnt == 1)
return p;
rt = tt[rt].child[id];
p++;
}
return 0;
}

int main()
{
int T;
scanf("%d",&T);
while( T--)
{
init();
scanf("%d",&n);
/* for( int i = 1; i <= n; i++)
{
scanf("%s",name);
TrieInsert(name);
}

ans = 0;
TrieSearch(0,0);
*/
for( int i = 1; i <= n; i++)
{
scanf("%s",name[i]);
TrieInsert(name[i]);
}

ans = 0;
for( int i = 1; i <= n; i++)
ans += TrieSearch(name[i]);
printf("%d\n",ans);
}

}