《ybtoj高效进阶》第二部分第五章例题2 单词频率

题目大意

有 n 个单词,然后问你每个单词在这些单词中出现了多少次。

思路

显然的AC自动机,这时我们考虑每一个节点到根的字符串是多少字符的前缀即可。
code:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue> 
using namespace std;
int trie[1000006][26],que[1000006],s[1000006][2],tot=1,tot2,ed[1000006],nxt[1000006],n,m;
string x[210];
void insert(string x)
{
    
    
	int p=1;
	for (int i=0;i<x.size();i++)
	{
    
    
		if (!trie[p][x[i]-'a']) trie[p][x[i]-'a']=++tot;
		p=trie[p][x[i]-'a'];
		s[p][0]++;
	}
	ed[++tot2]=p;
	return;
}
void bfs()
{
    
    
	queue<int> o;
	for (int i=0;i<26;i++) trie[0][i]=1;
	o.push(1);
	nxt[1]=0;
	int r=0;
	que[r++]=1;
	while (o.size())
	{
    
    
		int u=o.front();
		o.pop();
		for (int i=0;i<26;i++)
		{
    
    
			if (!trie[u][i]) trie[u][i]=trie[nxt[u]][i];
			else
			{
    
    
				o.push(trie[u][i]);
				que[r++]=trie[u][i];
				nxt[trie[u][i]]=trie[nxt[u]][i];
			}
		}
	}
	for (int i=r-1;i>=0;i--) s[que[i]][1]=s[que[i]][0];
	for (int i=r-1;i>=0;i--) s[nxt[que[i]]][1]+=s[que[i]][1];
	return;
}
int ACQuantAsk(string x)
{
    
    
	int ans=0,u=1,l=x.size();
	for (int i=0;i<l;i++)
	{
    
    
		int k=trie[u][x[i]-'a'];
		while (k>1&&ed[k]!=-1)
		{
    
    
			ans+=ed[k];
			ed[k]=-1;
			k=nxt[k];
		}
		u=trie[u][x[i]-'a'];
	}
	return ans;
}
string xxx;
int main()
{
    
    
	for (int i=0;i<26;i++) trie[0][i]=1,trie[1][i]=0;
	tot=1;
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
	{
    
    
		cin>>x[i];
		insert(x[i]);
	}
	bfs();
	for (int i=1;i<=n;i++) cout<<s[ed[i]][1]<<endl;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_49843717/article/details/115254544
今日推荐