ZOJ - 3228 Searching the String(AC自动机求不重复子串出现次数)

题目链接:点击查看

题目大意:给出一个匹配串 str ,再给出 n 个长度不大于 6 的匹配串 s ,问每个匹配串出现的次数,分可以重复或不可以重复两种情况

题目分析:因为是匹配串在模式串中出现的次数,可以重复的情况就是AC自动机的最简单情况了,对于不允许重复的情况,我们可以对于每个节点记录一下最后一次出现的位置,与其长度比较一下就好了,为了简化代码,我们可以给 cnt 数组,也就是记录每个节点出现次数的数组多开一维,表示是否允许重复

代码:

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
    
typedef long long LL;
   
typedef unsigned long long ull;
    
const int inf=0x3f3f3f3f;
    
const int N=6e5+100;
 
char s[10],str[N];
 
int fail[N],pos[N],trie[N][26],type[N],deep[N],num[N][2],cnt;//type:操作类型 0:可重复 1:不可重复 deep:字符串长度 pos:字符串结尾节点编号

int last[N];//最后一次出现的位置 
 
int insert_word()
{
	int len=strlen(s);
	int pos=0;
	for(int i=0;iq;
	for(int i=0;i<26;i++)
	{
		if(trie[0][i])
		{
			fail[trie[0][i]]=0;
			q.push(trie[0][i]);
		}
	}
	while(!q.empty())
	{
		int cur=q.front();
		q.pop();
		for(int i=0;i<26;i++)
		{
			if(trie[cur][i])
			{
				fail[trie[cur][i]]=trie[fail[cur]][i];
				q.push(trie[cur][i]);
			}
			else
				trie[cur][i]=trie[fail[cur]][i];
		}
	}
}

void search_word()
{
	int len=strlen(str),pos=0;
	for(int i=0;i=deep[k])
			{
				num[k][1]++;
				last[k]=i;
			}
			k=fail[k];
		}
	}
}
 
void init()
{
	cnt=0;
	memset(pos,0,sizeof(pos));
	memset(trie,0,sizeof(trie));
	memset(num,0,sizeof(num));
	memset(last,-1,sizeof(last));
}
 
int main()
{
//#ifndef ONLINE_JUDGE
//  freopen("input.txt","r",stdin);
//    freopen("output.txt","w",stdout);
//#endif
//  ios::sync_with_stdio(false);
	int kase=0;
    while(scanf("%s",str)!=EOF)
    {
    	init();
    	int n;
    	scanf("%d",&n);
    	for(int i=1;i<=n;i++)
    	{
    		scanf("%d%s",type+i,s);
    		pos[i]=insert_word();
		}
		getfail();
		search_word();
		printf("Case %d\n",++kase);
		for(int i=1;i<=n;i++)
			printf("%d\n",num[pos[i]][type[i]]);
		putchar('\n');
	}
    
    
     
     
     
     
     
       
       
       
       
       
       
       
    return 0;
}

 

你可能感兴趣的:(字符串处理)