[BZOJ1174] Toponyms 字典树模板

刚刚学字典树，于是乎打了一个模板。不得不说好像会卡空间。。反正写了一个错误的代码2333
Trie树数组版：

#include<bits/stdc++.h>
#define maxn 500010
using namespace std;
int trie[maxn][53],f[maxn],sz=1,ans,n;
string s;
void add()
{
    int rt=0;
    for(int i=0;s[i];++i)
    {
        int k;
        if(s[i]==' ') k=52;
        else if(s[i]>='A'&&s[i]<='Z') k=s[i]-'A';
        else k=s[i]-'a'+26;
        if(trie[rt][k]) rt=trie[rt][k];
        else rt=trie[rt][k]=++sz;
        f[rt]++;
        ans=max(ans,f[rt]*(i+1));
    }
}
int main()
{
    scanf("%d\n",&n);
    for(int i=1;i<=n;++i)
        getline(cin,s),add();
    cout<<ans;
return 0;   
 } 
 /*
7
Jora de Sus
Orhei
Jora de Mijloc
Joreni
Jora de Jos
Japca
Orheiul Vechi
 */

Trie树链表版：

#include<bits/stdc++.h>
using namespace std;
struct tree{
    int sz;
    tree*so[52];
}*xtr;
string s;
int n,ans;
void add()
{
    tree* p=xtr;
    for(int i=0;s[i];++i)
    {
        int k;
        if(s[i]==' ') k=52;
        else if(s[i]>='A'&&s[i]<='Z') k=s[i]-'A';
        else k=s[i]-'a'+26;
        if(p->so[k]==NULL) 
            p->so[k]=new tree;
        p=p->so[k];
        p->sz++;
        ans=max(ans,p->sz*(i+1));
    }
}
int main()
{
    xtr=new tree;
    scanf("%d\n",&n);
    for(int i=1;i<=n;++i)
        getline(cin,s),add();
    cout<<ans;
    return 0;
}
/*
7
Jora de Sus
Orhei
Jora de Mijloc
Joreni
Jora de Jos
Japca
Orheiul Vechi
*/

Trie树数组封装版：

struct trie{  
    int tot,ch[2010][26];bool ok[maxn];  
    void insert(){  
        int p=0,len=strlen(s);  
        for (int i=0;i<len;p=ch[p][s[i]-'a'],i++) if (!ch[p][s[i]-'a']) ch[p][s[i]-'a']=++tot;  
        ok[p]=1;  
    }  
}T;