hdu 1251 统计难题

最新推荐文章于 2019-08-09 09:05:28 发布

原创最新推荐文章于 2019-08-09 09:05:28 发布 · 187 阅读

0 ·

CC 4.0 BY-SA版权

hdu 同时被 2 个专栏收录

119 篇文章

订阅专栏

trie树

5 篇文章

订阅专栏

本文介绍了一种使用Trie树（字典树）来高效处理字符串集合的方法，特别关注于如何通过Trie树的节点来记录字符串的出现次数，并实现字符串的插入、查找及前缀统计等功能。

主题思想 : Tries , 在每个非空节点，记录经过它的字符串的个数。
核心方法是 Tries 的put ，get 和collect 方法。

const int R=26;

struct Node{

    int val=0;
    Node* next[R];
    Node(){
        val=0;
        //very import 
        memset(next,NULL,sizeof(next));
    }
};
Node* root;

int ans=0;

void put(char key[]){

    Node *x=root;
    int d=0;
    char c;
    while(d<strlen(key)){
        char c=key[d];
        if(x->next[c-'a']==NULL){
            x->next[c-'a']=new Node();
        }
        x=x->next[c-'a'];
        x->val++;
        d++;
    }
}
Node* get(char key[]){

    Node *x=root;
    int d=0;
    char c;

    while(x!=NULL){

        if(d==strlen(key)) {
            return x;
        }
        c=key[d];

        x=x->next[c-'a'];
        d++;
    }

    return NULL;
}
int collect(Node *x){

    ans=0;
    if(x==NULL) return 0;
    ans+=x->val;
}
void prefix(char key[]){

    ans=0;
    collect(get(key));
}

，本来报RE ，，结果使用memest 语句对Node* 数组 next 进行了初始化，就不报错了。

AC代码：

#include <iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<queue>
using namespace std;

const int R=26;

struct Node{

    int val=0;
    Node* next[R];
    Node(){
        val=0;
        //very import 
        memset(next,NULL,sizeof(next));
    }
};
Node* root;

int ans=0;

void put(char key[]){

    Node *x=root;
    int d=0;
    char c;
    while(d<strlen(key)){
        char c=key[d];
        if(x->next[c-'a']==NULL){
            x->next[c-'a']=new Node();
        }
        x=x->next[c-'a'];
        x->val++;
        d++;
    }
}
Node* get(char key[]){

    Node *x=root;
    int d=0;
    char c;

    while(x!=NULL){

        if(d==strlen(key)) {
            return x;
        }
        c=key[d];

        x=x->next[c-'a'];
        d++;
    }

    return NULL;
}
int collect(Node *x){

    ans=0;
    if(x==NULL) return 0;
    ans+=x->val;
}
void prefix(char key[]){

    ans=0;
    collect(get(key));
}


int main()
{
    root=new Node();

    char s[15];
    int n=0;
    while(gets(s)){
        if(strcmp(s,"")==0)break;
        n++;
        put(s);
    }

    while(gets(s)){
        prefix(s);
        printf("%d\n",ans);
    }
    return 0;
}