light OJ -DNA Prefix （字典树）

题目链接：

http://acm.hust.edu.cn/vjudge/contest/view.action?cid=121897#problem/G

题目大意：

寻找所给字符串的 前缀*出现次数 的最大值

题解：

字典树，v[u]为结点出现次数，存字符串后利用搜索不断遍历所有有字母的点，找到最大值。此题因为只有4个字母，所以字典树的宽度应为4，如果为26会超出内存。

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;
}