CF547E Mike and Friends

CF547E Mike and Friends

\(\color{black}{\texttt{S}}\color{red}{\texttt{egmentTree}}\) 一眼就秒了,我调了一下午。

首先离线询问,拆成前缀和的形式进行处理,这样只需要从左到右添加字符串的贡献,扫一遍就做完了。

考虑建出AC自动机(广义SAM写法争议有点大,不是很敢用,也不熟练)和fail树,那么一个字符串子树内的所有状态都包含这个串。

所以只需要维护子树内出现了多少“贡献”。

考虑加入一个字符串产生的贡献,其实就是沿着 \(Trie\) 图跑,跑到的所有节点都加一(即它所有前缀的状态),最后统计子树和就是答案。

这就是一个单点加区间求和的东西,树状数组维护一下就好了。

注意树状数组的上界是总结点数加一,也是最大的dfs序,不要像我一样设成总结点数调不出来。。。

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef double db;
#define x first
#define y second
#define sz(v) (int)v.size()
#define pb(x) push_back(x)
#define mkp(x,y) make_pair(x,y)
inline int read(){
	int x=0,f=1;char c=getchar();
	while(!isdigit(c)){if(c=='-')f=0;c=getchar();}
	while(isdigit(c))x=x*10+c-'0',c=getchar();
	return f?x:-x;
}
#define N 200005
#define Q 500005
int n,q,ans[Q],tot,ch[N][26],fail[N];
int dfn[N],ed[N],tmr,to[N],tr[N];
string str[N];
struct node{
	int op,pos,id;
	node(){op=pos=id=0;}
	node(int o_,int p_,int i_){op=o_,pos=p_,id=i_;}
};
vector<node>v[N];
vector<int>e[N];
void insert(string str,int id){
	int u=0;
	for(int i=0;i<sz(str);++i){
		int c=str[i]-'a';
		if(!ch[u][c])ch[u][c]=++tot;
		u=ch[u][c];
	}
	to[id]=u;
}
void build_fail(){
	queue<int>q;
	for(int i=0;i<26;++i)if(ch[0][i])q.push(ch[0][i]);
	while(!q.empty()){
		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];
	}
	for(int i=1;i<=tot;++i)e[fail[i]].pb(i);
}
void dfs(int u){
	dfn[u]=++tmr;
	for(int v:e[u])dfs(v);
	ed[u]=tmr;
}
void add(int x,int d){for(int i=x;i<=tmr;i+=i&-i)tr[i]+=d;}
int ask(int x){int res=0;for(int i=x;i>0;i-=i&-i)res+=tr[i];return res;}
int ask(int l,int r){return ask(r)-ask(l-1);}
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
	cin>>n>>q;
	for(int i=1;i<=n;++i)
		cin>>str[i],insert(str[i],i);
	build_fail(),dfs(0);
	for(int i=1;i<=q;++i){
		int l,r,k;cin>>l>>r>>k;
		v[l-1].pb(node(-1,k,i)),v[r].pb(node(1,k,i));
	}
	for(int i=1;i<=n;++i){
		for(int u=0,j=0;j<sz(str[i]);++j)
			u=ch[u][str[i][j]-'a'],add(dfn[u],1);
		for(node j:v[i])
			ans[j.id]+=j.op*ask(dfn[to[j.pos]],ed[to[j.pos]]);
	}
	for(int i=1;i<=q;++i)cout<<ans[i]<<'\n';
}
上一篇:CF241B Friends


下一篇:CF1472A Cards for Friends