《ybtoj高效进阶》第二部分第五章例题3 前缀匹配

题目大意

给n个子串和一个母串,求每个子串前缀在母串上的最大匹配长度

思路

先把母串能够匹配的节点标记一次(AC自动机),再把所有子串跑一次自动机,若跑到的节点有标记,则至少匹配到现在枚举的那一位。
code:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue> 
using namespace std;
int trie[1000006][26],tot=1,nxt[1000006],n,m;
string x[100005];
bool book[1000006];
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,memset(trie[tot],0,sizeof(trie[tot]));
		p=trie[p][x[i]-'A'];
	}
	return;
}
void bfs()
{
    
    
	queue<int> o;
	for (int i=0;i<26;i++) trie[0][i]=1;
	o.push(1);
	nxt[1]=0;
	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
			{
    
    
				nxt[trie[u][i]]=trie[nxt[u]][i];
				o.push(trie[u][i]);
			}
		}
	}
	return;
}
int ACQuantAsk(string x,bool y)
{
    
    
	int ans=0,u=1,l=x.size();
	if (y==0)
	{
    
    
		for (int i=0;i<l;i++)
		{
    
    
			u=trie[u][x[i]-'A'];
			if (book[u]) ans=i+1;
		}
		return ans;
	}
	for (int i=0;i<l;i++)
	{
    
    
		int k=trie[u][x[i]-'A'];
		while (k>1)
		{
    
    
			book[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;
	cin>>n>>n;
	cin>>xxx;
	for (int i=1;i<=n;i++)
	{
    
    
		cin>>x[i];
		insert(x[i]);
	}
	bfs();
	ACQuantAsk(xxx,1);
	for (int i=1;i<=n;i++) cout<<ACQuantAsk(x[i],0)<<endl;
    return 0;
}

猜你喜欢

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