Live2d Test Env

CodeForces 547E:Mike and Friends(AC自动机+DFS序+主席树)

What-The-Fatherland is a strange country! All phone numbers there are strings consisting of lowercase English letters. What is double strange that a phone number can be associated with several bears!

In that country there is a rock band called CF consisting of n bears (including Mike) numbered from 1 to n.

Phone number of i-th member of CF is si. May 17th is a holiday named Phone Calls day. In the last Phone Calls day, everyone called all the numbers that are substrings of his/her number (one may call some number several times). In particular, everyone called himself (that was really strange country).

Denote as call(i, j) the number of times that i-th member of CF called the j-th member of CF.

The geek Mike has q questions that he wants to ask you. In each question he gives you numbers l, r and k and you should tell him the number

Input

The first line of input contains integers n and q (1 ≤ n ≤ 2 × 105 and 1 ≤ q ≤ 5 × 105).

The next n lines contain the phone numbers, i-th line contains a string siconsisting of lowercase English letters ().

The next q lines contain the information about the questions, each of them contains integers l, r and k (1 ≤ l ≤ r ≤ n and 1 ≤ k ≤ n).

Output

Print the answer for each question in a separate line.

Examples

Input
5 5
a
ab
abab
ababab
b
1 5 1
3 5 1
1 5 2
1 5 3
1 4 5
Output
7
5
6
3
6

题意:给定N个串,M次询问,每次询问[L,R]里含多少个X。

思路:(第一次做,结合主席树那里还是不太好想)。对N个串建立AC自动机,建立Fail树,然后得到DFS序。那么,对于每个串S,假设其长度为L,S在AC自动机上面跑,其每个前缀Si在AC自动机上面得到最大深度,对应的Fail树位置,贡献加1,保证这个贡献在Si的后缀的子树里(比如abcdef,那么跑到abcd时,在fail树上面对应的位置贡献加1 ,对于后缀abcd,bcd,cd,d的子树都含这个贡献)。 

        关键是如何出现子树来自于[L,R]的贡献。开始我以为以dfs序为X轴,以来自的串为Y轴建立主席树,查询的时候查询区间[in[u],out[u]],即子树里关键字在[L,R]里的个数。但是发现没有转移关系。所以想不走了。

看了其他人的代码,是以每个串,以来自的串为X轴(从长的前缀到短的前缀),以dfs序为X轴,保证了前缀和的正确性,查询的时候查询区间[L-1,R]的区间在[in[[u],out[u]]的数量。

 

#include<bits/stdc++.h>
using namespace std;
const int maxn=200010;
int ch[maxn][26],fa[maxn],cnt=0; //trie树 
int Laxt[maxn],Next[maxn],To[maxn],tot; //fail树 
int q[maxn],fail[maxn],head,tail; //fail树 
int in[maxn],out[maxn],pos[maxn],times;//dfs序 
int p[maxn],rt[maxn],cur,num; struct in{int l,r,sum;}s[maxn*20];
char c[maxn];
void addedge(int u,int v){ Next[++tot]=Laxt[u]; Laxt[u]=tot; To[tot]=v; }
int insert()
{
    int Now=0; for(int i=1;c[i];i++){
        if(!ch[Now][c[i]-'a']){ 
             ch[Now][c[i]-'a']=++cnt;
             fa[cnt]=Now;
        }
        Now=ch[Now][c[i]-'a'];
    }   return Now;
}
void buildfail()
{
    for(int i=0;i<26;i++){
        if(ch[0][i]) q[++head]=ch[0][i],fail[ch[0][i]]=0;
        else ch[0][i]=0;
    }
    while(tail<head){
        int Now=q[++tail];
        for(int i=0;i<26;i++){
            if(ch[Now][i]) {
                q[++head]=ch[Now][i]; fail[ch[Now][i]]=ch[fail[Now]][i];
            }
            else ch[Now][i]=ch[fail[Now]][i];
        }
    }
    for(int i=1;i<=cnt;i++) addedge(fail[i],i);
}
void dfs(int u)
{
    in[u]=++times;
    for(int i=Laxt[u];i;i=Next[i]) dfs(To[i]);
    out[u]=times;
}
void add(int &Now,int pre,int L,int R,int pos)
{
    Now=++num; s[Now]=s[pre]; s[Now].sum++;
    if(L==R) return ; int Mid=(L+R)>>1;
    if(pos<=Mid) add(s[Now].l,s[pre].l,L,Mid,pos);
    else add(s[Now].r,s[pre].r,Mid+1,R,pos);
}
int query(int pre,int Now,int L,int R,int l,int r)
{
    if(l<=L&&r>=R) return s[Now].sum-s[pre].sum;
    int res=0,Mid=(L+R)>>1;
    if(l<=Mid) res+=query(s[pre].l,s[Now].l,L,Mid,l,r);
    if(r>Mid) res+=query(s[pre].r,s[Now].r,Mid+1,R,l,r);
    return res;
}
int main()
{
    int N,M,Q,L,R,x,i,j;
    scanf("%d%d",&N,&Q);
    for(i=1;i<=N;i++){
        scanf("%s",c+1);
        pos[i]=insert();
    }
    buildfail();
    dfs(0); cur=0;
    for(i=1;i<=N;i++){
        for(j=pos[i];j;j=fa[j]){
            cur++;
            add(rt[cur],rt[cur-1],1,times,in[j]);
        }
        p[i]=rt[cur];
    }
    while(Q--){
        scanf("%d%d%d",&L,&R,&x);
        printf("%d\n",query(p[L-1],p[R],1,times,in[pos[x]],out[pos[x]]));
    }
    return 0;
}

 

posted @ 2018-07-01 14:14  nimphy  阅读(624)  评论(0编辑  收藏  举报