统计难题 HDU - 1251(trie树)

这道题当时用map水过了,但733ms而用trie树只需要62ms。

TRIE树

#include <bits/stdc++.h>
using namespace std;

const int maxn=1e6+10;
char s[20];
int sum[maxn]={0};
int trie[maxn][26]={0};
int root;
int tot;

void inser()
{
    root=0;
    for(int i=0;s[i];i++)
    {
        int id=s[i]-'a';
        if(!trie[root][id])
            trie[root][id]=++tot;   //开辟一个新的节点,节点编号是tot+1
        root=trie[root][id];          
        sum[root]++;
    }
}

void srch()
{
    root=0;
    for(int i=0;s[i];i++)
    {
        int x=s[i]-'a';
        if(trie[root][x]==0) {printf("0\n"); return;}
        root=trie[root][x];
    }
    printf("%d\n", sum[root]);
}

int main()
{
    //freopen("input.txt", "r", stdin);
    tot=0;
    while(gets(s)&&strcmp(s, ""))
    {
        inser();
    }
    while(gets(s)!=NULL)
    {
        srch();
    }
    return 0;
}

水map代码

#include <set>
#include <map>
#include <stack>
#include <queue>
#include <deque>
#include <cmath>
#include <cstdio>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
#define mem(a,n) memset(a,n,sizeof(a))
#define FRER() freopen("input.txt", "r", stdin)
#define FREW() freopen("output.txt", "r", stdin)
typedef long long LL;
using namespace std;
const int inf = 0x3f3f3f3f, maxn = 100000 + 10;

map<string, int> m;

int main()
{
   // FRER();
    string s;
    string t;
    while(getline(cin,s)&&s!="") {
        t="";
        for(int i=0;i<s.length();i++){
            t+=s[i];
            m[t]++;
        }
    }
    while(cin >> s) {
        printf("%d\n",m[s]);
    }
    return 0;
}
    原文作者:Trie树
    原文地址: https://blog.csdn.net/sunmaoxiang/article/details/81866583
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞