hihocoder 1014

hihocoder 1014

题意是去后缀是输入字符串s的的个数。

首先用结构体做，结构体更实现，而且通俗易懂些。一直建立数，每次建立都往子节点加1，查找的时候找到最后那个字母时，输出它的子节点的数字就可以了。

#include <iostream>
#include <stdio.h>
#include <malloc.h>
using namespace std;
struct Nod{
    int num;
    Nod* next[26];
    Nod(){
        int num = 0;
        for(int i = 0; i < 26; i ++)
            next[i] = NULL;
    }
}t;
int n,m;
char s[11], ss[11];
void mkTrie(char* c){
    Nod* p = &t;
    for(int i = 0; c[i]; i ++){
        int a = c[i]-'a';
        if(p->next[a] == NULL){
            struct Nod* b = (struct Nod*)malloc(sizeof(struct Nod));
            p->next[a] = b;
            //p->next[a] = new Nod;
        }
        p = p->next[a];
        p->num++;
    }
}
int find(char *c){
    Nod* p = &t;
    for(int i = 0; c[i]; i ++){
        int a = c[i]-'a';
        if(p->next[a] == NULL) return 0;
        p = p->next[a];
    }
    return p->num;
}
int main(){
    scanf("%d",&n);
    for(int i = 0; i < n; i ++) scanf("%s",s),mkTrie(s);
    scanf("%d",&m);
    while(m--){
        scanf("%s",ss);
        printf("%d\n",find(ss));
    }
    return 0;
}

也可以用数组实现，不过更麻烦些。

#include <iostream>
#include <stdio.h>
using namespace std;
int sum[100000*26][26], sz = 1,n, m, trie[100000*26];
char s[11];

void mkTrie(){
    int root = 0;
    for(int i = 0; s[i]; i ++){
        int a = s[i]-'a';
        if(!sum[root][a]) sum[root][a] = sz++;
        trie[root]++;
        root = sum[root][a];
    }
    trie[root]++;
}
int find(){
    int root = 0;
    for(int i = 0; s[i]; i ++){
        int a = s[i] - 'a';
        if(sum[root][a])root=sum[root][a];
        else return 0;
    }
    return trie[root];
}
int main(){
    scanf("%d",&n);
    for(int i = 0; i < n; i ++){
        scanf("%s",s);
        mkTrie();
    }
    scanf("%d",&m);
    while(m--){
        scanf("%s",s);
        printf("%d\n",find());
    }
    return 0;
}