首页 > 技术文章 > [TJOI2013]单词

Troverld 2020-04-26 17:31 原文

[TJOI2013]单词

近乎模板,只要暴力建树,再把每个字符串丢进去匹配就可以了。

我一开始写了非拓扑的代码,结果最后一个点T了。

不得不改成了拓扑。

另外,这题需要保存字符串信息,为了避免MLE,我们把所有东西全都压成一个大串。

代码:

#include<bits/stdc++.h>
using namespace std;
int n,cnt=1,occ[210],S[210];
char s[1000001];
struct AC_automaton{
	int ch[26],fail,in,tms;
	vector<int>fin;
}t[1001000];
inline void ins(int id){
	int x=1;
	for(register int i=S[id-1];i<S[id];i++){
		if(!t[x].ch[s[i]-'a'])t[x].ch[s[i]-'a']=++cnt;
		x=t[x].ch[s[i]-'a'];
	}
	t[x].fin.push_back(id);
}
queue<int>q;
inline void build(){
	for(register int i=0;i<26;i++){
		if(t[1].ch[i])t[t[1].ch[i]].fail=1,q.push(t[1].ch[i]),t[1].in++;
		else t[1].ch[i]=1;
	}
	while(!q.empty()){
		register int x=q.front();q.pop();
		for(register int i=0;i<26;i++){
			if(t[x].ch[i])t[t[x].ch[i]].fail=t[t[x].fail].ch[i],q.push(t[x].ch[i]),t[t[t[x].fail].ch[i]].in++;
			else t[x].ch[i]=t[t[x].fail].ch[i];
		}
	}
}
inline void merge(int id){
	int x=1;
	for(register int i=S[id-1];i<S[id];i++){
		x=t[x].ch[s[i]-'a'];
		t[x].tms++;
	}
}
inline void topo(){
	for(int i=1;i<=cnt;i++)if(!t[i].in)q.push(i);
	while(!q.empty()){
		int x=q.front();q.pop();
		t[t[x].fail].in--;
		t[t[x].fail].tms+=t[x].tms;
		if(!t[t[x].fail].in)q.push(t[x].fail);
		for(int i=0;i<t[x].fin.size();i++)occ[t[x].fin[i]]=t[x].tms;
	}
}
int main(){
	scanf("%d",&n);
	for(register int i=1;i<=n;i++)scanf("%s",s+S[i-1]),S[i]=S[i-1]+strlen(s+S[i-1]),ins(i);
	build();
	for(register int i=1;i<=n;i++)merge(i);
	topo();
	for(register int i=1;i<=n;i++)printf("%d\n",occ[i]);
	return 0;
}

推荐阅读