社区讨论

蒟蒻求助,全WA

P3808AC 自动机(简单版)参与者 3已保存回复 3

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
3 条
当前快照
1 份
快照标识符
@mi7ydd0l
此快照首次捕获于
2025/11/21 05:38
4 个月前
此快照最后确认于
2025/11/21 05:38
4 个月前
查看原帖
CPP
#include<cstdio>
#include<cstring>
#include<queue>
#define N 1000000
using namespace std;
int trie[N+1][31],fail[N+1],cntword[N+1],tot=1;
void insert(char s[])
{
	int len=strlen(s);
	int u=1;
	for(int i=0;i<len;i++)
	{
		int c=s[i]-'a';
		if(trie[u][c]!=0)
			trie[u][c]=++tot;
		u=trie[u][c];
	}
	cntword[u]++;
}
void get_fail()
{
	queue<int> q;
	for(int i=0;i<26;i++)
		if(trie[1][i]!=0)
		{
			fail[trie[1][i]]=1;
			q.push(trie[1][i]);
		}
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		for(int i=0;i<26;i++)
		{
			if(trie[u][i]!=0)
			{
				fail[trie[u][i]]=trie[fail[u]][i];
				q.push(trie[u][i]);
			}
			else
				trie[u][i]=trie[fail[u]][i];
		}
	}
}
int AC_query(char s[])
{
	int len=strlen(s);
	int u=1,ans=0;
	for(int i=0;i<len;i++)
	{
		u=trie[u][s[i]-'a'];
		for(int j=u;j!=1&&cntword[j]!=-1;j=fail[j])
		{
			ans+=cntword[j];
			cntword[j]=-1;
		}
	}
}
char s[N+1];
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s);
        insert(s);
    }
    fail[1]=1;
    get_fail();
    scanf("%s",s);
    printf("%d",AC_query(s));
    return 0;
}

回复

3 条回复,欢迎继续交流。

正在加载回复...