hihocoder1014(trie树)

题目链接

        http://hihocoder.com/problemset/problem/1014?sid=1355923

题意

        先给n个字符串,再给m个字符串,对于m个字符串的每一个,计算之前的n个字符串有多少是以其为前缀的,输出结果。

题解

         裸的Trie树,n个字符串先建立Trie树,然后m的每一个都放在m里面跑就行了。

         与模板不一样的是,这里需要维护每一个节点被遍历的次数(之前只记录结尾的点),所以用cnt数组来记录。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
using namespace std;

#define INIT(x) memset(x,0,sizeof(x))
#define eps 1e-8

typedef long long ll;
const int inf = 0x3f3f3f3f;
const int maxn = 1000005;

int trie[maxn][26];
int cnt[maxn];

int k = 1;

void insert (char *w) {
	int len = strlen(w);
	int p = 0;
	for(int i=0;i<len;i++) {
		int c = w[i]-'a';
		if(!trie[p][c]) {
			trie[p][c] = k;
			k++;
		}
		p = trie[p][c];
		cnt[p]++;
 	}
}

int search(char *s) {
	int len = strlen(s);
	int p = 0;
	for(int i=0;i<len;i++) {
		int c = s[i]-'a';
		if(!trie[p][c])
			return 0;
		p = trie[p][c];
	}
	return cnt[p];
}

int n,m;
char s[maxn];

int main()
{
	cin>>n;
	for(int i=0;i<n;i++) {
		scanf("%s",s);
		insert(s);
	}
	int ans = 0;
	cin>>m;
	for(int i=0;i<m;i++) {
		scanf("%s",s);
		ans = search(s);
		cout<<ans<<endl;
	}
	return 0;
}


        

    原文作者:Trie树
    原文地址: https://blog.csdn.net/zxwsbg/article/details/81706794
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞