light OJ -DNA Prefix （字典树）

题目链接：

题目大意：

题解：

AC代码：

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cctype>
#include <iostream>
using namespace std;

const int maxnode = 2500500;
const int sigma   = 4;
int ch[maxnode][sigma],v[maxnode],sz,ans;
int u;
int maxx;
int numb[maxnode];
struct trie
{
trie()
{

sz=1;
memset(v,0,sizeof v);
memset(ch[0],0,sizeof ch[0]);
}
int idx(char c)//如果不这样存的话会爆内存
{
if(c == 'C')
return 1;
else if(c == 'G')
return 2;
else if(c == 'T')
return 3;
return 0;
}
void insert(char* s)
{
int l =strlen(s);
u=0;
for(int i=0;i<l;i++)
{
int c = idx(s[i]);
if(!ch[u][c])
{
v[sz] = 0;
memset(ch[sz],0,sizeof ch[sz]);
ch[u][c] = sz++;
//cout << ch[u][c] <<endl;
}
u = ch[u][c];
v[u] ++;
}
}
void solve(int u, int d)//搜索寻找最大值，d为前缀长度
{
ans = max(ans, d * v[u]);
for (int i = 0; i < 4; i++)
{
if (ch[u][i])
solve(ch[u][i], d + 1);
}
}
}T;

int n,m;
char c[105];

int main()
{
int TTT;
int cnt = 1;
cin >> TTT;
while(TTT--)
{
int nnn;
cin >> nnn;
while(nnn--)
{
scanf("%s" , c);
T.insert(c);
}
ans = 0;
T.solve(0,0);
printf("Case %d: %d\n", cnt++, ans);
sz=1;
memset(v,0,sizeof v);
memset(ch[0],0,sizeof ch[0]);
}
return 0;
}