HDU 1251 统计难题 字典树大水题

今天刚看的字典树， 就RE了一发， 字典树原理还是很简单的， 唯一的问题就是不知道一维够不够用， 就开的贼大， 这真的是容易MLE的东西啊， 赶紧去学优化吧。

HDU-1251 统计难题

这道题唯一的问题就是会不会字典树， 2333， 给一个字典树的博客传送门， 话说这个博客一搜就搜到了啊.

代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define ULL unsigned LL
#define fi first
#define se second
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define max3(a,b,c) max(a,max(b,c))
#define min3(a,b,c) min(a,min(b,c))
const int INF = 0x3f3f3f3f;
const LL mod = 1e9+7;
typedef pair<int,int> pll;
const int N = 1e6+10;
int tree[N][26];
int sum[N];
char str[15];
int tot = 2;
void Insert(){
    int rt = 1;
    int len = strlen(str);
    for(int i = 0; i < len; i++){
        int id = str[i] - 'a';
        if(tree[rt][id] == 0) tree[rt][id] = tot++;
        sum[tree[rt][id]]++;
        rt = tree[rt][id];
    }
}
int Find(){
    int rt = 1;
    int len = strlen(str);
    for(int i = 0; i < len; i++){
        int id = str[i] - 'a';
        if(tree[rt][id] == 0) return 0;
        rt = tree[rt][id];
    }
    return sum[rt];
}
int main(){
    while(cin.getline(str, 15)){
        if(!isalpha(str[0])) break;
        Insert();
    }
    while(cin.getline(str, 15)){
        printf("%d\n",Find());
    }
    return 0;
}