洛谷2922 秘密消息（Trie树）

传送门

【题目分析】

既然是读入字符串然后与之前的字符串进行匹配，那么肯定是对之前的字符串建立Trie树，这点毋庸置疑。

那么就是统计有多个字符串作为他的前缀出现以及该字符串是多少个字符串的前缀，那么就与电话簿一题类似了。

我记录经过该点的字符串数量以及以该点作为结束点的数量，那么最后答案就是这条路径上结束点的数量之和+经过最后点的字符串数量（注意我的pass记录的不包括最后一个字符，最后一个打的是end标记）

【代码~】

#include<bits/stdc++.h>
using namespace std;
const int MAXN=1e6+10;
const int LEN=1e4+10;

int n,q;
int siz,k,a[LEN];
struct Trie{
	int pass,end;
	int son[2];
}tr[MAXN];

int Read(){
	int i=0,f=1;
	char c;
	for(c=getchar();(c>'9'||c<'0')&&c!='-';c=getchar());
	if(c=='-')
	  f=-1,c=getchar();
	for(;c>='0'&&c<='9';c=getchar())
	  i=(i<<3)+(i<<1)+c-'0';
	return i*f;
}

void insert(){
	for(int i=1;i<=k;++i){
		a[i]=Read();
	}
	int last=0;
	for(int i=1;i<=k;++i){
		tr[last].pass++;
		if(!tr[last].son[a[i]]){
			tr[last].son[a[i]]=++siz;
			memset(tr[siz].son,0,sizeof(tr[siz].son));
		}
		last=tr[last].son[a[i]];
	}
	tr[last].end++; 
}

int query(){
	int ret=0;
	for(int i=1;i<=k;++i){
		a[i]=Read();
	}
	int last=0;
	for(int i=1;i<=k;++i){
		ret+=tr[last].end;
		if(!tr[last].son[a[i]]){
			return ret;
		}
		last=tr[last].son[a[i]];
	}
	ret+=tr[last].pass+tr[last].end;
	return ret;
}

int main(){
	n=Read(),q=Read();
	for(int i=1;i<=n;++i){
		k=Read();
		insert();
	}
	for(int i=1;i<=q;++i){
		k=Read();
		cout<<query()<<'\n';
	}
	return 0;
}