HDU1251 统计难题 字典树模板

一直觉得字典树是个高大上的东西,今天学了才发现也并没有那么难。。

附上讲解博客Orz:https://www.cnblogs.com/TheRoadToTheGold/p/6290732.html

后来一直WA,应该是读数据的方法不对,,看了网上的博客发现gets是个好东西。。

附上AC代码:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;

const int MAX=500010;
int trie[MAX][26];
int tot;
int sum[MAX];//被访问过的次数

void ins(char s[])
{
    int len=strlen(s);
    int root=0;
    for(int i=0;i<len;i++)
    {
        int id=s[i]-'a';
        if(!trie[root][id])
            trie[root][id]=++tot;
        root=trie[root][id];
        sum[root]++;
    }
}
int Search(char s[])
{
    int len=strlen(s);
    int root=0;
    for(int i=0;i<len;i++)
    {
        int id=s[i]-'a';
        if(!trie[root][id])
            return 0;
        root=trie[root][id];
    }
    return sum[root];
}
int main()
{
    memset(trie,0,sizeof(trie));
    memset(sum,0,sizeof(sum));
    tot=0;
    char s[13];
    while(gets(s))
    {
         if(s[0]==NULL) //空行。gets读入的回车符会自动转换为NULL
             break;
         ins(s);
    }
    while(gets(s))
        printf("%d\n",Search(s));
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Cc_Sonia/article/details/82624461