사전 트 리 연습 (1) hihocoder 1014 (같은 접두사 의 수 를 구하 십시오)

제목 링크:
http://hihocoder.com/problemset/problem/1014
제목:
n 개의 단 어 를 정 한 다음 에 우 리 는 사전 트 리 를 구성 한 다음 에 m 개의 문자열 을 줄 것 입 니 다. 몇 개의 단어 가 이 문자열 을 접두사 로 하 는 지 구 합 니 다.
코드 는 다음 과 같 습 니 다:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int maxn = 100010;

const int max_size = 30;

int id(char c){
    return c-'a';
}

struct Trie{
    Trie *ch[max_size];
    int num;
    Trie(){
        num = 0;
        for(int i=0;i<max_size;i++)
            ch[i]=NULL;
    }
}*root;

void insert_str(char *s){
    Trie *p = root;
    p->num++;
    for(int i=0; p&&s[i] ; i++){
        int u = id(s[i]);
        if(p->ch[u]==NULL)
            p->ch[u] = new Trie;
        p=p->ch[u];
        p->num++;
    }
}

int find_str(char *s){
    Trie *p = root;
    for(int i=0;p&&s[i];i++){
        int u = id(s[i]);
        if(p->ch[u]==NULL) return 0;
        p=p->ch[u];
    }
    return p->num;
}

int main()
{
  char s[12];
  int n, m;
  while(~scanf("%d", &n))
  {
    root = new Trie;
    while(n--){
      scanf("%s", s);
      insert_str(s);
    }
    scanf("%d", &m);
    while(m--){
      scanf("%s", s);
      printf("%d
", find_str(s)); } } return 0; }

좋은 웹페이지 즐겨찾기