ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

洛谷 P2414 / LOJ 2444 「NOI2011」阿狸的打字机

2022-06-22 08:01:41  阅读:132  来源: 互联网

标签:结点 ch P2414 2444 阿狸 int st maxn fail


洛谷传送门

LOJ 传送门

思路

首先套路地建 AC 自动机,并存下每一行的终止结点,记为 \(a_x\)。

考虑在 AC 自动机上匹配的过程,\(x\) 在 \(y\) 中出现的次数就相当于在 Trie 树上 \(a_y\) 到根结点的链上,每个结点都不断跳 fail,有多少个结点是 \(a_x\),也就是在 fail 树上有多少个结点在 \(a_x\) 的子树内。

因此我们先将询问离线,将每一组询问 \((x,y)\) 挂到 \(a_y\) 上,然后建出 fail 树并求出每个结点的 \([st_x,ed_x]\)(即子树 \(dfn\) 序的范围)。在原 Trie 上 dfs 并处理所有 \(a_y\) 的询问。每次递归开始时将 \(st_u\) 设为 \(1\),回溯时再减回去。这样当遍历到结点 \(k\) 时,只有 \(k\) 到根结点的链上的点为 \(1\)。于是可以处理每个挂在 \(k\) 上的询问,树状数组求出。

代码

点击查看代码
/*

p_b_p_b txdy
AThousandMoon txdy
AThousandSuns txdy
hxy txdy

*/

#include <bits/stdc++.h>
#define pb push_back
#define fst first
#define scd second

using namespace std;
typedef long long ll;
typedef pair<int, int> pii;

const int maxn = 100100;

int n, m, a[maxn], ans[maxn];
int c[maxn], st[maxn], ed[maxn], times;
int head[maxn], len;
char s[maxn];
vector<pii> qq[maxn];

struct edge {
	int to, next;
} edges[maxn << 1];

void add_edge(int u, int v) {
	edges[++len].to = v;
	edges[len].next = head[u];
	head[u] = len;
}

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

inline void update(int x, int d) {
	for (int i = x; i <= times; i += lowbit(i)) {
		c[i] += d;
	}
}

inline int query(int x) {
	int res = 0;
	for (int i = x; i; i -= lowbit(i)) {
		res += c[i];
	}
	return res;
}

struct AC {
	int ch[maxn][26], och[maxn][26], fa[maxn];
	int fail[maxn], tot;
	
	void insert(char *s) {
		int p = 0;
		for (int i = 0; s[i]; ++i) {
			if (islower(s[i])) {
				if (!ch[p][s[i] - 'a']) {
					ch[p][s[i] - 'a'] = ++tot;
					fa[tot] = p;
				}
				p = ch[p][s[i] - 'a'];
			} else if (s[i] == 'B') {
				p = fa[p];
			} else {
				a[++n] = p;
			}
		}
		for (int i = 0; i <= tot; ++i) {
			for (int j = 0; j < 26; ++j) {
				och[i][j] = ch[i][j];
			}
		}
	}
	
	void build() {
		queue<int> q;
		for (int i = 0; i < 26; ++i) {
			if (ch[0][i]) {
				q.push(ch[0][i]);
			}
		}
		while (q.size()) {
			int u = q.front();
			q.pop();
			for (int i = 0; i < 26; ++i) {
				if (ch[u][i]) {
					fail[ch[u][i]] = ch[fail[u]][i];
					q.push(ch[u][i]);
				} else {
					ch[u][i] = ch[fail[u]][i];
				}
			}
		}
	}
	
	void dfs(int u) {
		st[u] = ++times;
		for (int i = head[u]; i; i = edges[i].next) {
			int v = edges[i].to;
			dfs(v);
		}
		ed[u] = times;
	}
	
	void dfs2(int u) {
		update(st[u], 1);
		for (pii p : qq[u]) {
			ans[p.scd] = query(ed[p.fst]) - query(st[p.fst] - 1);
		}
		for (int i = 0; i < 26; ++i) {
			if (och[u][i]) {
				dfs2(och[u][i]);
			}
		}
		update(st[u], -1);
	}
	
	void solve() {
		for (int i = 1; i <= tot; ++i) {
			add_edge(fail[i], i);
		}
		dfs(0);
		dfs2(0);
		for (int i = 1; i <= m; ++i) {
			printf("%d\n", ans[i]);
		}
	}
} ac;

void solve() {
	scanf("%s%d", s, &m);
	ac.insert(s);
	ac.build();
	for (int i = 1; i <= m; ++i) {
		int x, y;
		scanf("%d%d", &x, &y);
		qq[a[y]].pb(make_pair(a[x], i));
	}
	ac.solve();
}

int main() {
	int T = 1;
	// scanf("%d", &T);
	while (T--) {
		solve();
	}
	return 0;
}

标签:结点,ch,P2414,2444,阿狸,int,st,maxn,fail
来源: https://www.cnblogs.com/zltzlt-blog/p/16399109.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有