Codeforces 547E

不明白这个题为啥会是div1E。。。
套路的,给所有串建个AC自动机,插入的时候给路径打标记,这样统计一个串在 [ l , r ] [l,r] [l,r]中出现了多少次就等价于统计在这个点fail树的子树中有多少个 [ l , r ] [l,r] [l,r]间的标记,直接将询问离线后扫描线,对dfs序用树状数组维护即可。
时间复杂度 O ( ( ∑ i = 1 n ∣ s i ∣ + q ) log ⁡ n ) \mathcal O((\sum_{i=1}^{n}|s_i|+q)\log n) O((i=1nsi+q)logn)

#include <bits/stdc++.h>
#define end end2
#define lowbit(x) (x&-x)
 
using namespace std;
 
int sumv[200005];
 
void add(int x,int n) {
  for(;x<=n;x+=lowbit(x)) sumv[x]++;
}
 
int sum(int x) {
  int s=0;
  for(;x;x-=lowbit(x)) s+=sumv[x];
  return s;
}
 
int sum(int l,int r) {
  return sum(r)-sum(l-1);
}
 
namespace AC {
 
int ch[200005][26],fa[200005],tot=1;
 
int insert(char *s,int n) {
  int x=1;
  for(int i=0;i<n;i++) {
  	int v=s[i]-'a';
  	if (!ch[x][v]) {
  		ch[x][v]=++tot;
  		fa[tot]=x;
	  }
  	x=ch[x][v];
  }
  return x;
}
 
int fail[200005];
queue <int> q;
 
void bfs() {
  for(int i=0;i<26;i++)
    if (ch[1][i]) {
    	fail[ch[1][i]]=1;
    	q.push(ch[1][i]);
	}
	else ch[1][i]=1;
  while (!q.empty()) {
  	int x=q.front();q.pop();
  	for(int i=0;i<26;i++)
  	  if (ch[x][i]) {
  	  	  fail[ch[x][i]]=ch[fail[x]][i];
  	  	  q.push(ch[x][i]);
		}
	  else ch[x][i]=ch[fail[x]][i];
  }
}
 
int dfn[200005],end[200005],dfs_cnt;
vector <int> son[200005];
 
void dfs(int x) {
  dfn[x]=++dfs_cnt;
  for(int i=0;i<son[x].size();i++) 
    dfs(son[x][i]);
  end[x]=dfs_cnt;
}
 
void build() {
  bfs();
  for(int i=2;i<=tot;i++) son[fail[i]].push_back(i);
  dfs(1);
}
 
}
 
int ed[200005];
 
struct Query {
  int x,k,v,id;
  Query() {}
  Query(int a,int b,int c,int d):x(a),k(b),v(c),id(d) {}
  bool operator < (const Query & b) const {return x<b.x;}
};
 
Query q[1000005];
int ans[500005];
 
char str[200005];
 
int main() {
  int n,m;
  scanf("%d%d",&n,&m);
  for(int i=1;i<=n;i++) {
  	scanf("%s",str);
  	int len=strlen(str);
  	ed[i]=AC::insert(str,len);
  }
  AC::build();
  int cnt=0;
  for(int i=1;i<=m;i++) {
  	int x,y,z;
  	scanf("%d%d%d",&x,&y,&z);
  	q[++cnt]=Query(y,z,1,i);
  	if (x>1) q[++cnt]=Query(x-1,z,-1,i);
  }
  sort(q+1,q+cnt+1);
  int r=1;
  for(int i=1;i<=n;i++) {
  	for(int j=ed[i];j;j=AC::fa[j]) add(AC::dfn[j],AC::tot);
  	while (r<=cnt&&q[r].x<=i) {
  		int x=ed[q[r].k];
  		ans[q[r].id]+=sum(AC::dfn[x],AC::end[x])*q[r].v;
  		r++;
	  }
  }
  for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
  return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值