CF163E e-Government(AC自动机+树状数组维护fail树的DFS序)

给出n个字符串,每个字符串有两种状态:退役和服役

三种操作:

1)询问当前服役的字符串在询问字符串内的出现次数之和。

2)把一个退役的字符串设为服役。

3)把一个服役的字符串设为退役。

如何处理操作1?

对所有字符串建出AC自动机,建出fail树,求子树和。

每个模式串s_i在t中出现的次数之和就是fail树上根到当前节点的路径的点权之和。

操作2 3的修改操作,就是把某个点权+1或-1。

所以可以用线段树维护 每个节点到根节点的路径点权之和。

单点修改影响的是整个子树。

这里涉及到对fail树的dfs序的区间修改和区间询问,但是有一个巨巨巨恶心的点,就是卡线段树,上线段树一定会MLE。

网上找了个树状数组维护区间修改的板子...

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+500;
int n,q,tr[maxn][26],fail[maxn],tot;
vector<int> g[maxn];
long long cnt[maxn];
string t,s;
int ed[maxn];
int dfn[maxn],id[maxn],tol,sz[maxn];
int insert (string s) {
	int u=0;
	for (char i:s) {
		if (!tr[u][i-'a']) tr[u][i-'a']=++tot;
		u=tr[u][i-'a'];
	}
	cnt[u]++;
	return u;
}
void build () {
	queue<int> q;
	for (int i=0;i<26;i++) {
		if (tr[0][i]) {
			q.push(tr[0][i]);
		}
	}
	while (q.size()) {
		int u=q.front();
		q.pop();
		for (int i=0;i<26;i++) {
			if (tr[u][i]) {
				fail[tr[u][i]]=tr[fail[u]][i];
				q.push(tr[u][i]);
			}
			else {
				tr[u][i]=tr[fail[u]][i];
			}
		}
	}
}
 
 
long long c[maxn<<2],lz[maxn<<2];
void Build (int i,int l,int r) {
	if (l==r) {
		c[i]=cnt[id[l]];
		return;
	}
	int mid=(l+r)>>1;
	Build(i<<1,l,mid);
	Build(i<<1|1,mid+1,r);
	c[i]=c[i<<1]+c[i<<1|1];
}
void pushdown (int i,int l,int r) {
	if (lz[i]) {
		int mid=(l+r)>>1;
		c[i<<1]+=1ll*(mid-l+1)*lz[i];
		lz[i<<1]+=lz[i];
		c[i<<1|1]+=1ll*(r-mid)*lz[i];
		lz[i<<1|1]+=lz[i]; 
		lz[i]=0;
	} 
}
void up (int i,int l,int r,int L,int R,int v) {
	if (l>=L&&r<=R) {
		c[i]+=1ll*(r-l+1)*v;
		lz[i]+=v;
		return;
	}
	pushdown(i,l,r);
	int mid=(l+r)>>1;
	if (L<=mid) up(i<<1,l,mid,L,R,v);
	if (R>mid) up(i<<1|1,mid+1,r,L,R,v);
	c[i]=c[i<<1]+c[i<<1|1];
}
long long query (int i,int l,int r,int L,int R) {
	if (l>=L&&r<=R) return c[i];
	pushdown(i,l,r);
	int mid=(l+r)>>1;
	long long ans=0;
	if (L<=mid) ans+=query(i<<1,l,mid,L,R);
	if (R>mid) ans+=query(i<<1|1,mid+1,r,L,R);
	return ans;
}
 
void dfs (int u,int f) {
	dfn[u]=++tol;
	cnt[u]+=cnt[f];
	id[tol]=u;
	sz[u]=1;
	for (int v:g[u]) {
		if (v==f) continue;
		dfs(v,u);
		sz[u]+=sz[v];
	}
}
int cc[maxn];
int main () {
	ios::sync_with_stdio(false);
	cin>>q>>n;
	if (n==1&&q==1) return printf("1\n"),0;
	for (int i=1;i<=n;i++) {
		cin>>s;
		ed[i]=insert(s);
		cc[ed[i]]++;
	}
	build();
	for (int i=1;i<=tot;i++) {
		g[fail[i]].push_back(i);
		g[i].push_back(fail[i]);
	}
	dfs(0,0);
	Build(1,1,tot+1);
	while (q--) {
		string op;
		cin>>op;
		if (op[0]=='?') {
			long long ans=0;
			int u=0;
			for (int i=1;i<op.size();i++) {
				u=tr[u][op[i]-'a'];
				ans+=query(1,1,tot+1,dfn[u],dfn[u]);
			}
			printf("%lld\n",ans);
		}
		else if (op[0]=='+') {
			int tt=0;
			for (int i=1;i<op.size();i++) {
				tt=tt*10+op[i]-'0';
			}
			if (cc[ed[tt]]) continue; 
			cc[ed[tt]]++;
			up(1,1,tot+1,dfn[ed[tt]],dfn[ed[tt]]+sz[ed[tt]]-1,1);
		}
		else if (op[0]=='-') {
			int tt=0;
			for (int i=1;i<op.size();i++) {
				tt=tt*10+op[i]-'0';
			}
			if (cc[ed[tt]]==0) continue;
			cc[ed[tt]]--;
			up(1,1,tot+1,dfn[ed[tt]],dfn[ed[tt]]+sz[ed[tt]]-1,-1);
		}
	}
}

上一篇:ac自动机


下一篇:tensorflow进阶篇-5(反向传播1)