hihoCoder #1014 : Trie树

来源:互联网 发布:asp php 编辑:程序博客网 时间:2024/06/01 15:18

题目:

Trie树
原题链接:https://hihocoder.com/problemset/problem/1014

题解思路:
题目上边已经讲的比较清楚了,建立一棵trie树,同时在创建树的时候,每经过一个节点就给它值加一,也就是说在建立树的时候就顺便记录下了以当前为前缀的单词的个数。

代码如下:

#include<bits/stdc++.h>#define length 11using namespace std;struct Node{    int num;    Node* s[26];    Node(){        num = 0;        for(int i=0;i<26;i++){            s[i] = NULL;        }    }}node;void insertIn(char ch[]){    Node *n = &node;    for(int i=0;ch[i];i++){        int index = ch[i] - 'a';        if(n->s[index] == NULL){            n->s[index] = new Node();        }        n = n->s[index];        n->num++;    }}int findNods(char ch[]){    Node *n = &node;    int i;    for(i=0;ch[i];i++){        int index = ch[i] -'a';        if(n->s[index] == NULL){            cout<<0<<endl;            return 0;        }        n = n->s[index];    }    cout<<n->num<<endl;    return 0;}int main(){    int s;    char ch[length];    cin>>s;    for(int i=0;i<s;i++){        cin>>ch;        insertIn(ch);    }    cin>>s;    for(int i=0;i<s;i++){        cin>>ch;        findNods(ch);    }    return 0;}

也有好久没写算法题了,当时没坚持acm确实是最大的失误,现在上手写树还是相当不熟练的,c++也用的不习惯了,还得坚持!