关闭

hihocoder 1014 trie树

标签: trie树
98人阅读 评论(0) 收藏 举报
分类:
    [题目链接](http://hihocoder.com/problemset/problem/1014)
#include <iostream>
#include <string>
using namespace std;

struct trie{
    char key;
    int num;
    trie *next[26];
    trie(){
        num = 0;
        for(int i = 0; i < 26; i++) next[i] = NULL;
    }
};

trie root;

void ins(string s) {
    trie *head = &root;

    for(int i = 0; i < s.size(); i++) {
        int k = s[i] - 'a';

        if(head->next[k] == NULL) {
            trie *node = new trie;
            node->key = s[i];
            head->next[k] = node;
        }
        head = head->next[k];
        head->num++;
    }
}
int ser(string s) {
    trie *head = &root;
    for(int i = 0; i < s.size(); i++) {
        int k = s[i] - 'a';
        if(head->next[k] == NULL ) return 0;
        else {
            head = head->next[k];
        }
    }
    return head->num;
}

int main(){

    string s;
    int n,m;
    cin >> n;
    while(n--) {
        cin >> s;
        ins(s);
    }
    cin >> m;
    while(m--) {
        cin >> s;
        cout << ser(s) << endl;
    }

    return 0;
}
0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:9453次
    • 积分:400
    • 等级:
    • 排名:千里之外
    • 原创:32篇
    • 转载:3篇
    • 译文:0篇
    • 评论:0条
    文章分类