统计难题 HDU - 1251(trie树)

最新推荐文章于 2021-11-09 00:20:03 发布

原创最新推荐文章于 2021-11-09 00:20:03 发布 · 227 阅读

0 ·

CC 4.0 BY-SA版权

本文通过一个具体的实例对比了使用Trie树和Map在字符串查找效率上的区别。通过对比发现，在相同任务下，Trie树的时间消耗远低于Map。文章提供了完整的代码实现，并详细解释了两种方法的具体操作。

这道题当时用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;
}