AC自动机模板

const int kind=26;
struct AC_Automata{
	struct node{
		node *child[kind];
		node *fail;
		int count;
		node(){
			for(int i=0;i<kind;++i)	child[i]=NULL;
			count=0;
			fail=NULL;
		}
	};
	node *root;
	AC_Automata(){
		 root=new node();
	}
	inline int id(char c){return c-'a';}

	void insert(char *s){
		node *now;
		now=root;
		for(int i=0;s[i];++i){
			int t=id(s[i]);
			if(now->child[t]!=NULL)	now=now->child[t];
			else {
				now->child[t]=new node();
				now=now->child[t];
			}
		}
		now->count++;
	}
	
	void build(){
		queue<node*>q;
		while(!q.empty()) q.pop();
		q.push(root);
		while(!q.empty()){
			node *now=q.front(),*p=NULL; q.pop();
			for(int i=0;i<kind;++i){
				if(now->child[i]==NULL)	continue;
				if(now==root)	now->child[i]->fail=root;
				else {
					p=now->fail;
					while(p!=NULL){
						if(p->child[i]!=NULL){
							now->child[i]->fail=p->child[i];
							break;
						}
						p=p->fail;
					}
					if(p==NULL)	now->child[i]->fail=root;
				}
				q.push(now->child[i]);
			}
		}
	}

	int query(char *s){
		int ans=0;
		node *now=root,*temp;
		for(int i=0;s[i];++i){
			int t=id(s[i]);
			while(now->child[t]==NULL && now!=root)	now=now->fail;
			now=now->child[t];
			if(now==NULL)	now=root;
			temp=now;
			while(temp!=root && temp->count!=-1){
				ans+=temp->count;
				temp->count=-1;
				temp=temp->fail;
			}
		}
		return ans;
	}
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值