【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;

}

【BZOJ2434】【NOI2011】阿狸的打字机（AC自动机，树状数组）的更多相关文章

  1. BZOJ2434: [Noi2011]阿狸的打字机(AC自动机 树状数组)

    Time Limit: 10 Sec  Memory Limit: 256 MBSubmit: 4140  Solved: 2276[Submit][Status][Discuss] Descript ...

  2. [BZOJ2434][Noi2011]阿狸的打字机 AC自动机+树状数组+离线

    题目链接:http://www.lydsy.com/JudgeOnline/problem.php?id=2434 题目中这种多个串匹配的问题,一下子就想到了AC自动机.然后发现如果要建立AC自动机, ...

  3. 【BZOJ】2434: [Noi2011]阿狸的打字机 AC自动机+树状数组+DFS序

    [题意]阿狸喜欢收藏各种稀奇古怪的东西,最近他淘到一台老式的打字机.打字机上只有28个按键,分别印有26个小写英文字母和'B'.'P'两个字母. 经阿狸研究发现,这个打字机是这样工作的: l 输入小写 ...

  4. [NOI2011]阿狸的打字机 --- AC自动机 + 树状数组

    [NOI2011] 阿狸的打字机 题目描述: 阿狸喜欢收藏各种稀奇古怪的东西,最近他淘到一台老式的打字机. 打字机上只有28个按键,分别印有26个小写英文字母和'B'.'P'两个字母.经阿狸研究发现, ...

  5. BZOJ.2434.[NOI2011]阿狸的打字机(AC自动机 树状数组 DFS序)

    题目链接 首先不需要存储每个字符串,可以将所有输入的字符依次存进Trie树,对于每个'P',记录该串结束的位置在哪,以及当前节点对应的是第几个串(当前串即根节点到当前节点):对于'B',只需向上跳一个 ...

  6. BZOJ2434[Noi2011]阿狸的打字机——AC自动机+dfs序+树状数组

    题目描述 阿狸喜欢收藏各种稀奇古怪的东西,最近他淘到一台老式的打字机.打字机上只有28个按键,分别印有26个小写英文字母和'B'.'P'两个字母. 经阿狸研究发现,这个打字机是这样工作的: l 输入小 ...

  7. BZOJ2434:[NOI2011]阿狸的打字机(AC自动机,线段树)

    Description 阿狸喜欢收藏各种稀奇古怪的东西,最近他淘到一台老式的打字机.打字机上只有28个按键,分别印有26个小写英文字母和'B'.'P'两个字母. 经阿狸研究发现,这个打字机是这样工作的 ...

  8. bzoj 2434 阿狸的打字机 - Aho-Corasick自动机 - 树状数组

    题目传送门 传送站I 传送站II 题目大意 阿狸有一个打字机,它有3种键: 向缓冲区追加小写字母 P:打印当前缓冲区(缓冲区不变) B:删除缓冲区中最后一个字符 然后多次询问第$x$个被打印出来的串在 ...

  9. 洛谷P2414 阿狸的打字机 [NOI2011] AC自动机+树状数组/线段树

    正解:AC自动机+树状数组/线段树 解题报告: 传送门! 这道题,首先想到暴力思路还是不难的,首先看到y有那么多个,菜鸡如我还不怎么会可持久化之类的,那就直接排个序什么的然后按顺序做就好,这样听说有7 ...

  10. 【BZOJ2434】[NOI2011]阿狸的打字机 AC自动机+DFS序+树状数组

    [BZOJ2434][NOI2011]阿狸的打字机 Description 阿狸喜欢收藏各种稀奇古怪的东西,最近他淘到一台老式的打字机.打字机上只有28个按键,分别印有26个小写英文字母和'B'.'P ...

随机推荐

  1. Java获取指定包名下的所有类的全类名的解决方案

        最近有个需求需要获取一个指定包下的所有类的全类名,因此特意写了个获取指定包下所有类的全类名的工具类.在此记录一下,方便后续查阅 一.思路         通过ClassLoader来查找指定包 ...

  2. 解决RobotFramework的关键字不能高亮的问题

    一个可能的原因:路径中存在汉字,RobotFramework对这方面运行的不太好.

  3. 【强化学习】python 实现 q-learning 例一

    本文作者:hhh5460 本文地址:https://www.cnblogs.com/hhh5460/p/10134018.html 问题情境 -o---T# T 就是宝藏的位置, o 是探索者的位置 ...

  4. Dubbo（四） Dubbo-Admin项目 Dubbo管理台

    前言 在dubbo项目中,有注册中心,消费者,提供者就足以构成一个完整的项目了.但是仅仅有这三个角色,很难对整个项目状态有直观的了解,以及对项目操作. 因此早有前辈对此原因作出了贡献——一个通用的du ...

  5. 安卓开发helloworld

    https://blog.csdn.net/tangjie134/article/details/79495204

  6. left join 右表数据不唯一的情况解决方法

    https://blog.csdn.net/u010089432/article/details/52165026

  7. Word frequency program ver.1

    先吐槽一下自己 做这个作业的时候完全没有提前考虑好时间 导致要提交前一天晚上通宵写代码 而且还满满的bug TAT 缓一口气 程序还在继续写 先把有点累 想着先把博客写了 PART I 预计时间: 语 ...

  8. #个人作业Week2——结对编程对象代码复审

    General 代码能够正确运行,能够正确生成指定数量的题目和答案,并且能够对给出的题目和答案文件进行比对,输出结果. 代码没有非常复杂的逻辑,比较容易理解,但是在缺少注释的情况下有部分代码需要较长时 ...

  9. 《Linux内核设计与实现》第八周读书笔记——第四章 进程调度

    <Linux内核设计与实现>第八周读书笔记——第四章 进程调度 第4章 进程调度35 调度程序负责决定将哪个进程投入运行,何时运行以及运行多长时间,进程调度程序可看做在可运行态进程之间分配 ...

  10. Leetcode 546. Remove Boxes

    题目链接: https://leetcode.com/problems/remove-boxes/description/ 问题描述 若干个有序排列的box和它们的颜色,每次可以移除若干个连续的颜色相 ...