【BZOJ2434】【NOI2011】阿狸的打字机（AC自动机，树状数组）

时间：2023-03-09 09:33:02
【BZOJ2434】【NOI2011】阿狸的打字机（AC自动机，树状数组）

【BZOJ2434】阿狸的打字机（AC自动机，树状数组）

先写个暴力：

每次打印出字符串后，就插入到\(Trie\)树中

搞完后直接搭\(AC\)自动机

看一看匹配是怎么样的：

每次沿着\(AC\)自动机走，在每一个节点都跳\(fail\)指针

如果有\(x\)串的末节点，就给答案\(+1\)

这样的话没有必要存下每个串

只要给\(AC\)自动机存一个父亲节点

记录一下每个串的结束位置

倒着往上跳就可以了

这样能够拿到\(40\)分

Update2018.1.25:这份代码对于重复串的处理会有问题，感谢 @超级大蒟蒻

#include<iostream>

#include<cstdio>

#include<cstdlib>

#include<cstring>

#include<cmath>

#include<algorithm>

#include<set>

#include<map>

#include<vector>

#include<queue>

using namespace std;

#define MAX 200000

inline int read()

{

	int x=0,t=1;char ch=getchar();

	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();

	if(ch=='-')t=-1,ch=getchar();

	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();

	return x*t;

}

char ss[MAX];

int nd[MAX],n,tot;

struct Node

{

	int vis[26];

	int fail,fa;

	int lt;

}t[MAX];

void GetFail()

{

	queue<int> Q;

	for(int i=0;i<26;++i)

		if(t[0].vis[i])Q.push(t[0].vis[i]);

	while(!Q.empty())

	{

		int u=Q.front();Q.pop();

		for(int i=0;i<26;++i)

			if(t[u].vis[i])

				t[t[u].vis[i]].fail=t[t[u].fail].vis[i],Q.push(t[u].vis[i]);

			else t[u].vis[i]=t[t[u].fail].vis[i];

	}

}

int Query(int x,int y)

{

	int ret=0;

	int now=nd[y];

	while(now)

	{

		for(int i=now;i;i=t[i].fail)

			if(t[i].lt==x){++ret;break;}

		now=t[now].fa;

	}

	return ret;

}

int main()

{

	scanf("%s",ss+1);

	int now=0;

	for(int i=1,l=strlen(ss+1);i<=l;++i)

	{

		if(ss[i]>='a'&&ss[i]<='z')

		{

			if(!t[now].vis[ss[i]-'a'])t[now].vis[ss[i]-'a']=++tot,t[tot].fa=now;

			now=t[now].vis[ss[i]-'a'];

		}

		if(ss[i]=='B')now=t[now].fa;

		if(ss[i]=='P'){nd[++n]=now;t[now].lt=n;}

	}

	int Q=read();

	GetFail();

	while(Q--)

	{

		int x=read(),y=read();

		printf("%d\n",Query(x,y));

	}

	return 0;

}

这样子对于每一个询问都会要暴跳

如果对于某个串有重复的多次询问

那么就会多很多次没有任何意义的计算

所以，可以离线把所有询问都按照\(y\)排序

每次跳的时候开个桶一起计算

这样的话可以拿到\(70\)分

#include<iostream>

#include<cstdio>

#include<cstdlib>

#include<cstring>

#include<cmath>

#include<algorithm>

#include<set>

#include<map>

#include<vector>

#include<queue>

using namespace std;

#define MAX 200000

inline int read()

{

	int x=0,t=1;char ch=getchar();

	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();

	if(ch=='-')t=-1,ch=getchar();

	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();

	return x*t;

}

char ss[MAX];

int nd[MAX],n,tot;

int ans[MAX];

struct Node

{

	int vis[26];

	int fail,fa;

	int lt;

}t[MAX];

struct Question{int x,y,id,ans;}q[MAX];

bool operator<(Question a,Question b){return a.y<b.y;}

int sum[MAX];

void GetFail()

{

	queue<int> Q;

	for(int i=0;i<26;++i)

		if(t[0].vis[i])Q.push(t[0].vis[i]);

	while(!Q.empty())

	{

		int u=Q.front();Q.pop();

		for(int i=0;i<26;++i)

			if(t[u].vis[i])

				t[t[u].vis[i]].fail=t[t[u].fail].vis[i],Q.push(t[u].vis[i]);

			else t[u].vis[i]=t[t[u].fail].vis[i];

	}

}

int Query(int y)

{

	int ret=0;

	int now=nd[y];

	while(now)

	{

		for(int i=now;i;i=t[i].fail)

			if(t[i].lt)sum[t[i].lt]++;

		now=t[now].fa;

	}

	return ret;

}

int main()

{

	scanf("%s",ss+1);

	int now=0;

	for(int i=1,l=strlen(ss+1);i<=l;++i)

	{

		if(ss[i]>='a'&&ss[i]<='z')

		{

			if(!t[now].vis[ss[i]-'a'])t[now].vis[ss[i]-'a']=++tot,t[tot].fa=now;

			now=t[now].vis[ss[i]-'a'];

		}

		if(ss[i]=='B')now=t[now].fa;

		if(ss[i]=='P'){nd[++n]=now;t[now].lt=n;}

	}

	int Q=read();

	GetFail();

	for(int i=1;i<=Q;++i)

	{

		q[i].x=read(),q[i].y=read();

		q[i].id=i;

	}

	sort(&q[1],&q[Q+1]);

	for(int i=1,pos=1;i<=Q;i=pos)

	{

		Query(q[i].y);

		while(q[pos].y==q[i].y)q[pos].ans=sum[q[pos].x],pos++;

		memset(sum,0,sizeof(sum));

	}

	for(int i=1;i<=Q;++i)ans[q[i].id]=q[i].ans;

	for(int i=1;i<=Q;++i)

		printf("%d\n",ans[i]);

	return 0;

}

再来想想我们每次在干什么？？

跳\(fail\)

显然每个节点有且仅有一个\(fail\)指针

所以，这就是一棵树？？

把这个\(fail\)反过来看

现在的问题是什么？

原来是\(y\)的某个节点往上跳能不能到达\(x\)

现在反过来：

\(x\)往下跳能够到达几个\(y\)的节点

那，不就是求子树和？？？

如果把所有\(y\)的节点全部打上一个\(1\)的标记

那么，每次就变成了求\(x\)末节点的子树和

而一个点的子树在\(dfs\)序上一定是连续的一段

这样还是可以拿到\(70\)分

#include<iostream>

#include<cstdio>

#include<cstdlib>

#include<cstring>

#include<cmath>

#include<algorithm>

#include<set>

#include<map>

#include<vector>

#include<queue>

using namespace std;

#define MAX 200000

inline int read()

{

	int x=0,t=1;char ch=getchar();

	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();

	if(ch=='-')t=-1,ch=getchar();

	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();

	return x*t;

}

char ss[MAX];

int nd[MAX],n,tot;

int ans[MAX];

int c[MAX];

int dfn[MAX],low[MAX],tim;

inline int lowbit(int x){return x&(-x);}

void Modify(int x,int w){while(x<=tim)c[x]+=w,x+=lowbit(x);}

int getsum(int x){int ret=0;while(x)ret+=c[x],x-=lowbit(x);return ret;}

struct Node

{

	int vis[26];

	int fail,fa;

	int lt;

}t[MAX];

struct Question{int x,y,id,ans;}q[MAX];

bool operator<(Question a,Question b){return a.y<b.y;}

void GetFail()

{

	queue<int> Q;

	for(int i=0;i<26;++i)

		if(t[0].vis[i])Q.push(t[0].vis[i]);

	while(!Q.empty())

	{

		int u=Q.front();Q.pop();

		for(int i=0;i<26;++i)

			if(t[u].vis[i])

				t[t[u].vis[i]].fail=t[t[u].fail].vis[i],Q.push(t[u].vis[i]);

			else t[u].vis[i]=t[t[u].fail].vis[i];

	}

}

struct Line{int v,next;}e[MAX<<1];

int h[MAX],cnt=1;

inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}

void dfs(int u)

{

	dfn[u]=++tim;

	for(int i=h[u];i;i=e[i].next)dfs(e[i].v);

	low[u]=tim;

}

int main()

{

	scanf("%s",ss+1);

	int now=0;

	for(int i=1,l=strlen(ss+1);i<=l;++i)

	{

		if(ss[i]>='a'&&ss[i]<='z')

		{

			if(!t[now].vis[ss[i]-'a'])t[now].vis[ss[i]-'a']=++tot,t[tot].fa=now;

			now=t[now].vis[ss[i]-'a'];

		}

		if(ss[i]=='B')now=t[now].fa;

		if(ss[i]=='P'){nd[++n]=now;t[now].lt=n;}

	}

	int Q=read();

	GetFail();

	for(int i=1;i<=tot;++i)Add(t[i].fail,i);

	dfs(0);

	for(int i=1;i<=Q;++i)

	{

		q[i].x=read(),q[i].y=read();

		q[i].id=i;

	}

	sort(&q[1],&q[Q+1]);

	for(int i=1,pos=1;i<=Q;i=pos)

	{

		for(int now=nd[q[i].y];now;now=t[now].fa)

			Modify(dfn[now],1);

		while(q[pos].y==q[i].y)

		{

			int v=nd[q[pos].x];

			q[pos].ans=getsum(low[v])-getsum(dfn[v]-1);

			pos++;

		}

		memset(c,0,sizeof(c));

	}

	for(int i=1;i<=Q;++i)ans[q[i].id]=q[i].ans;

	for(int i=1;i<=Q;++i)

		printf("%d\n",ans[i]);

	return 0;

}

现在大致的方向已经没有问题了

看看我们重复算在哪里？

每次把串插入进树状数组！

因为很多的串会有重复

所以会反反复复把很多东西给重复插进去

这样就很慢了

于是，我们把\(Trie\)树\(dfs\)遍历一遍

我搞Fail指针的时候会把原来的Trie数给搞掉，还要备份。。

访问到的时候打一个\(+1\)

结束的时候打一个\(-1\)

每次访问到一个结束节点的时候，

一定是有且仅有这个串的节点被打了标记

这样就可以直接回答这个串的相关询问了

#include<iostream>

#include<cstdio>

#include<cstdlib>

#include<cstring>

#include<cmath>

#include<algorithm>

#include<set>

#include<map>

#include<vector>

#include<queue>

using namespace std;

#define MAX 200000

inline int read()

{

	int x=0,t=1;char ch=getchar();

	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();

	if(ch=='-')t=-1,ch=getchar();

	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();

	return x*t;

}

char ss[MAX];

int nd[MAX],n,tot;

int ans[MAX];

int c[MAX];

int dfn[MAX],low[MAX],tim;

int ql[MAX],qr[MAX];

inline int lowbit(int x){return x&(-x);}

void Modify(int x,int w){while(x<=tim)c[x]+=w,x+=lowbit(x);}

int getsum(int x){int ret=0;while(x)ret+=c[x],x-=lowbit(x);return ret;}

struct Node

{

	int vis[26];

	int Vis[26];

	int fail,fa;

	int lt;

}t[MAX];

struct Question{int x,y,id,ans;}q[MAX];

bool operator<(Question a,Question b){return a.y<b.y;}

void GetFail()

{

	queue<int> Q;

	for(int i=0;i<26;++i)

		if(t[0].vis[i])Q.push(t[0].vis[i]);

	while(!Q.empty())

	{

		int u=Q.front();Q.pop();

		for(int i=0;i<26;++i)

			if(t[u].vis[i])

				t[t[u].vis[i]].fail=t[t[u].fail].vis[i],Q.push(t[u].vis[i]);

			else t[u].vis[i]=t[t[u].fail].vis[i];

	}

}

struct Line{int v,next;}e[MAX<<1];

int h[MAX],cnt=1;

inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}

void dfs(int u)

{

	dfn[u]=++tim;

	for(int i=h[u];i;i=e[i].next)dfs(e[i].v);

	low[u]=tim;

}

void DFS(int u)

{

	Modify(dfn[u],1);

	if(t[u].lt)

		for(int i=ql[t[u].lt];i<=qr[t[u].lt];++i)

			q[i].ans=getsum(low[nd[q[i].x]])-getsum(dfn[nd[q[i].x]]-1);

	for(int i=0;i<26;++i)

		if(t[u].Vis[i])

			DFS(t[u].Vis[i]);

	Modify(dfn[u],-1);

}

int main()

{

	scanf("%s",ss+1);

	int now=0;

	for(int i=1,l=strlen(ss+1);i<=l;++i)

	{

		if(ss[i]>='a'&&ss[i]<='z')

		{

			if(!t[now].vis[ss[i]-'a'])t[now].vis[ss[i]-'a']=++tot,t[tot].fa=now;

			now=t[now].vis[ss[i]-'a'];

		}

		if(ss[i]=='B')now=t[now].fa;

		if(ss[i]=='P'){nd[++n]=now;t[now].lt=n;}

	}

	for(int i=0;i<=tot;++i)

		for(int j=0;j<26;++j)

			t[i].Vis[j]=t[i].vis[j];

	int Q=read();

	GetFail();

	for(int i=1;i<=tot;++i)Add(t[i].fail,i);

	dfs(0);

	for(int i=1;i<=Q;++i)

	{

		q[i].x=read(),q[i].y=read();

		q[i].id=i;

	}

	sort(&q[1],&q[Q+1]);

	for(int i=1,pos=1;i<=Q;i=pos)

	{

		ql[q[i].y]=i;

		while(q[pos].y==q[i].y)pos++;

		qr[q[i].y]=pos-1;

	}

	DFS(0);

	for(int i=1;i<=Q;++i)ans[q[i].id]=q[i].ans;

	for(int i=1;i<=Q;++i)

		printf("%d\n",ans[i]);

	return 0;

}

相关文章