CF547E Mike and Friends (AC 自动机+树状数组)

做法

首先在线算法 是不好搞的
我们先差分 离线询问
我们先用通过ac自动机 求出每个点的fail指针
dfs求出每个子树的大小 和dfn[n]
每个字符串在所有串的出现次数 其实就是他的终止节点 再fail树的子树的贡献
所以我们 需要单点修改 区间查询
用树状数组 维护即可

注意树状数组维护的 字典的 节点个数 不是n 因为这个调了好一会
注意函数别写重名了

#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 = 1e4 + 10;
int n, m;
ll dp[N][N];

struct acam {
	struct node {
		int fail;
		int f = 0;
		int son[26] = {};
	} trie[N];
	int idx = 0;
	map<int, int>mp;
	int fa[N];
	void insert(string &s, int id) {
		int u = 0;
		for(auto c : s) {
			if(!trie[u].son[c - 'a']) {
				fa[trie[u].son[c - 'a'] = ++idx] = u ;
			}
			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;
	
	void change(int id) {
		int  x = mp[id];
		while(x) {
			B.add(dfn[x], 1);
			x = fa[x];
		}
	}
	int query(int id) {
		int k = mp[id];
		return B.query(dfn[k] + siz[k] - 1) - B.query(dfn[k] - 1);
	}
} AC;
int ans[N];
vector<array<int, 3>>  q[N];
void solve() {
	cin >> n >> m;
	for(int i = 1; i <= n; i++) {
		string s;
		cin >> s;
		AC.insert(s, i);
	}
	AC.build();
	AC.B.reset(AC.idx + 1);
//	cout<<AC.B.n<<endl;
	fer(i, 1, m) {
		int l, r, x;
		cin >> l >> r >> x;
		q[r].push_back({x, i, 1});
	  if(l>1)q[l - 1].push_back({x, i, -1});
	}
	for(int i = 1; i <= n; i++) {
		AC.change(i);
		for (auto v : q[i]) {
			ans[v[1]] += ( v[2] * AC.query(v[0]) );
		}
	}
	for(int i = 1; i <= m; i++) {
		cout << ans[i] << endl;
	}

}


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






  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值