CF163E e-Government （ac自动机+树状数组）

做法

这个题和我的上一篇link很类似 只不过是反着来了

上一个 是 要求 一个字符串串 再一个范围内的串的 出现次数

这个是 给你字符串 让你求一个范围内的字符串在它的出现次数

显然做法 比较套路 我们都需要转换为dfn序列

这个只需要 处理模式串的 时候 把它所在的fail子树都加+1

如果询问是在他的 子树里 显然就有贡献 累加求和即可

#include <bits/stdc++.h>
using namespace  std;
//#define  int long long
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef vector<int> vi;
#define fi first
#define se second
#define pb  push_back
#define inf 1<<62
#define endl "\n"
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define de_bug(x) cerr << #x << "=" << x << endl
#define all(a) a.begin(),a.end()
#define IOS   std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define  fer(i,a,b)  for(int i=a;i<=b;i++)
#define  der(i,a,b)  for(int i=a;i>=b;i--)
const int mod = 1e4 + 7;
const int N = 1e6 + 10;
int n, m;

struct acam {
	struct node {
		int fail;
		int son[26] = {};
	} trie[N];
	int idx = 0;
	map<int, int>mp;
	void insert(string &s, int id) {
		int u = 0;
		for(auto c : s) {
			if(!trie[u].son[c - 'a']) {
				trie[u].son[c - 'a'] = ++idx  ;
			}
			u = trie[u].son[c - 'a'];
		}
		mp[id] = u;
	}
	int cnt = 0;
	int siz[N];
	int dfn[N];
	vi g[N];
	void addedge(int a, int b) {
		g[a].push_back(b);
	}
	void dfs(int x) {
		dfn[x] = ++cnt;
		siz[x] = 1;
		for(auto v : g[x]) {
			dfs(v);
			siz[x] += siz[v];
		}
	}

	void build() {
		queue<int>q;
		for(int i = 0; i <= 25; i++) {
			if(trie[0].son[i]) {
				q.push(trie[0].son[i]);
			}
		}
		while(!q.empty()) {
			int u = q.front();
			q.pop();
			for(int i = 0; i <= 25; i++) {
				int v = trie[u].son[i], fa = trie[trie[u].fail].son[i];
				if(trie[u].son[i]) {
					trie[v].fail = fa;
					q.push(v);
				} else {
					trie[u].son[i] = trie[trie[u].fail].son[i];
				}
			}
		}
		for(int i = 1; i <= idx; i++) addedge(trie[i].fail, i);
		dfs(0);
	}
	struct BIT {
		int n;
		//n = idx + 1;
		int c[N];
		void reset(int len) {
			n = len;
		}
		void  init(int len, int a[]) {
			n = len;
			for(int i = 1; i <= n; i++) {
				c[i] += a[i];
				if(i + (i & (-i)) <= n)c[i + (i & -i)] += c[i];
			}
		}
		void  add(int x, int k) {
			for(; x <= n; x += x & -x)c[x] += k;
		}
		int  query(int x) {
			int res = 0;
			for(; x; x -= x & -x)res += c[x];
			return res;
		}
	} B;
} AC;

int vis[N];
void solve() {
	cin >> n >> m;
	for(int i = 1; i <= m; i++) {
		string s;
		cin >> s;
		AC.insert(s, i);
	}
	AC.build();
	AC.B.reset(AC.idx + 1);
	for(int i = 1; i <= m; i++) {
		int x = AC.mp[i];
		AC.B.add(AC.dfn[x], 1);
		AC.B.add(AC.dfn[x] + AC.siz[x], -1);
		vis[i] = 1;
	}

	for(int i = 1; i <= n; i++) {
		char ch;
		cin >> ch;
		if(ch == '?') {
			string s;
			cin >> s;
			int u = 0;
			int ans = 0;
			for(auto c : s) {
				u = AC.trie[u].son[c - 'a'];
				ans += AC.B.query(AC.dfn[u]);
			}
			cout << ans << endl;
		} else if(ch == '-') {
			int x;
			cin >> x;
			if(!vis[x])continue;
			vis[x] = 0;
			int id = AC.mp[x];
			AC.B.add(AC.dfn[id], -1);
			AC.B.add(AC.dfn[id] + AC.siz[id], 1);
		} else {
			int x;
			cin >> x;
			if(vis[x])continue;
			vis[x] = 1;
			int id = AC.mp[x];
			AC.B.add(AC.dfn[id], 1);
			AC.B.add(AC.dfn[id] + AC.siz[id], -1);
		}
	}

}


int main() {
	IOS;
	int _ = 1;
	//cin>>_;
	while( _-- )
		solve();
}

/*
4 3
bbaaa
abaaab
aa
-2
-2
+1
?abaabaaab*/