[Ybtoj第9章例1]プレフィックス統計[トライツリー]

---

問題解決のアイデア

知識ポイント：辞書ツリー

（うーん、アムウェイのブログに来たばかりです^ _ ^）

辞書ツリーの基本的な操作を学んだ後、戻ってこの質問、ああ、テンプレートの質問を見てください。。。
最初にNNを置く$N個の$文字列を辞書ツリーに格納します$v\left[p\right]$は、辞書ツリーのノードpで終わる文字ストリングの数を表します。
答えを計算します。文字列Tをトラバースすると、答えは${\sum }_{X\epsilon P\left[T\right]}v\left[p\right]$
（$p\left[T\right]$は、Tが辞書ツリーでトラバースしたノード番号のセットを表します）

---

コード

#include<iostream>
#include<cstdio>
#include<iomanip>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;

char a[1000010],s[1000010];
int n,m,trie[1000010][30],v[1000010],lyx=1;

void insert(){
    
    //往字典树加入一个字符串
	int len=strlen(a+1),p=1,c=0;
	for(int i=1;i<=len;i++)
	{
    
    
		c=a[i]-'a'+1;
		if(!trie[p][c])
			trie[p][c]=++lyx;
		p=trie[p][c];
	}
	v[p]++;
//	cout<<v[p]<<" "<<p;
}

int get(){
    
    //查询
	int len=strlen(s+1),p=1,ans=0,c=0;
	for(int i=1;i<=len;i++)
	{
    
    
		c=s[i]-'a'+1;
		if(!trie[p][c])
			return ans;
		p=trie[p][c];
		ans+=v[p];
	}
	return ans;
}

int main() {
    
    
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
    
    
		scanf("%s",a+1);
		insert();
	}
	for(int i=1;i<=m;i++)
	{
    
    
		scanf("%s",s+1);
		printf("%d\n",get());
	}
}