hiho-1014 Trie树 (Trie的建树与查询)

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<stack>
#include<queue>
#include<map>
#define RG register
#define IL inline
#define pi acos(-1.0)
#define ll long long 
using namespace std;

int gi() {
  char ch=getchar(); int x=0;
  while(ch<'0' || ch>'9') ch=getchar();
  while(ch>='0' && ch<='9') {x=10*x+ch-'0';ch=getchar();}
  return x;
}

const int maxn = 1000010;

int id=1;
char s[maxn];
int ch[maxn][27];
int cnt[maxn],val[maxn];

void build(char *x) {
  int u=0,n=strlen(x);
  for(int i=0; i<n; i++) {
    int c=x[i]-'a';
    if(!ch[u][c]) {
      val[id]=0;//非单词结点
      ch[u][c]=id++;
    }
    u=ch[u][c];//往下走
    cnt[u]++;//经过一次u就加一次
  }
  val[u]=1;//单词结点
}

int query(char *x) {
  int u=0,n=strlen(x);
  for(int i=0; i<n; i++) {
    int c=x[i]-'a';
    if(!ch[u][c]) return 0;
    u=ch[u][c];
  }
  return cnt[u];
}

int main() {
  int n=gi();
  for(int i=1; i<=n; i++) {
    scanf("%s", s);
    build(s);
  }
  int m=gi();
  for(int i=1; i<=m; i++) {
    scanf("%s", s);
    printf("%d\n", query(s));
  }
  return 0;
}