51nod 1440 迈克打电话 后缀数组+可持久化线段树+二分

题意

有n只熊，从1到n进行编号。
第i只熊的电话号码是si。每只熊会给那些电话号码是他的子串的熊打电话（可能会给自己打）。
call(i, j) 表示第i只熊给第j只熊打电话的次数，也就是第j个串在第i个串中出现的次数。
迈克会有q次询问。每个询问中给出l,r,k,然后请您计算一下$\sum_{i=l}^rcall(i,k)$。
1 ≤ n ≤ 2 × 10^5 , 1 ≤ q ≤ 5 × 10^5，所有串的长度之和≤ 2 × 10^5。

分析

由于已经多年没有碰过后缀数组，所以并没有想到后缀数组的解法。
我们可以把所有串拼在一起，在串之间随意插入一个字符，然后跑出其sa。
对于一个询问(l,r,k)，也就是要求出在原串的一段区间内k这个字符串出现了多少次。
如果没有l,r的限制，用后缀数组+二分的经典做法显然可以做到nlogn。
加上限制以后，我们可以对每个位置求出其所在的字符串，然后用主席树来查询区间中[l,r]出现的次数即可。

在卡常的时候很zz的自己写了一个min函数，结果怎么卡都卡不过。果然还是系统自带的函数比较优越。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=400005;

int n,q,b[N],rank[N*2],c[N],d[N],s[N],sa[N],rmq[N][19],posl[N/2],posr[N/2],bin[25],sz,stack[15],l,r,ul,ur;
struct tree{int l,r,s;}t[N*19];
char ch[N/2];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void pri(int x)
{
    if (!x) {puts("0");return;}
    int top=0;
    while (x) stack[++top]=x%10,x/=10;
    while (top) putchar(stack[top]+'0'),top--;
    puts("");
}

void get_sa(int n,int m)
{
    for (int i=1;i<=n;i++) b[s[i]]++;
    for (int i=1;i<=m;i++) b[i]+=b[i-1];
    for (int i=n;i>=1;i--) c[b[s[i]]--]=i;
    int t=0;
    for (int i=1;i<=n;i++)
    {
        t+=(s[c[i]]!=s[c[i-1]]);
        rank[c[i]]=t;
    }
    int j=1;
    while (j<=n)
    {
        for (int i=1;i<=t;i++) b[i]=0;
        for (int i=1;i<=n;i++) b[rank[i+j]]++;
        for (int i=1;i<=t;i++) b[i]+=b[i-1];
        for (int i=n;i>=1;i--) c[b[rank[i+j]]--]=i;
        for (int i=1;i<=t;i++) b[i]=0;
        for (int i=1;i<=n;i++) b[rank[i]]++;
        for (int i=1;i<=t;i++) b[i]+=b[i-1];
        for (int i=n;i>=1;i--) d[b[rank[c[i]]]--]=c[i];
        t=0;
        for (int i=1;i<=n;i++)
        {
            int x=d[i],y=d[i-1];
            t+=(rank[x]!=rank[y]||rank[x]==rank[y]&&rank[x+j]!=rank[y+j]);
            c[x]=t;
        }
        for (int i=1;i<=n;i++) rank[i]=c[i];
        if (t==n) break;
        j<<=1;
    }
    for (int i=1;i<=n;i++) sa[rank[i]]=i;
}

void get_height(int n)
{
    int k=0;
    for (int i=1;i<=n;i++)
    {
        if (k) k--;
        int j=sa[rank[i]-1];
        while (i+k<=n&&j+k<=n&&s[i+k]==s[j+k]) k++;
        rmq[rank[i]][0]=k;
    }
}

void get_rmq(int n)
{
    bin[0]=1;
    for (int i=1;i<=20;i++) bin[i]=bin[i-1]<<1;
    int w=0;
    c[0]=-1;
    for (int i=1;i<=n;i++)
        if (i==bin[w]) c[i]=c[i-1]+1,w++;
        else c[i]=c[i-1];
    for (int j=1;j<=c[n];j++)
        for (int i=1;i+bin[j]-1<=n;i++)
            rmq[i][j]=min(rmq[i][j-1],rmq[i+bin[j-1]][j-1]);
}

int get_mn(int l,int r)
{
    int w=c[r-l+1];
    return min(rmq[l][w],rmq[r-bin[w]+1][w]);
}

void ins(int &d,int p,int x)
{
    d=++sz;t[d]=t[p];t[d].s++;
    if (l==r) return;
    int mid=l+r>>1;
    if (x<=mid) r=mid,ins(t[d].l,t[p].l,x);
    else l=mid+1,ins(t[d].r,t[p].r,x);
}

int query(int d,int p,int x)
{
    if (l==r) return t[d].s-t[p].s;
    int mid=l+r>>1;
    if (x<=mid) {r=mid;return query(t[d].l,t[p].l,x);}
    else {l=mid+1;return query(t[d].r,t[p].r,x)+t[t[d].l].s-t[t[p].l].s;}
}

void get_pts(int n,int w,int len)
{
    l=2;r=w;
    while (l<=r)
    {
        int mid=l+r>>1;
        if (get_mn(mid,w)>=len) r=mid-1;
        else l=mid+1;
    }
    ul=r+1;
    l=w+1;r=n;
    while (l<=r)
    {
        int mid=l+r>>1;
        if (get_mn(w+1,mid)>=len) l=mid+1;
        else r=mid-1;
    }
    ur=l-1;
}

int main()
{
    n=read();q=read();
    int now=0;
    for (int i=1;i<=n;i++)
    {
        if (i>1) s[++now]=30;
        gets(ch);
        int len=strlen(ch);
        posl[i]=now+1;posr[i]=now+len;
        for (int j=0;j<len;j++) s[++now]=ch[j]-'a'+1;
    }
    get_sa(now,30);
    get_height(now);
    get_rmq(now);
    for (int i=1;i<=now;i++) b[i]=0;
    for (int i=1;i<=n;i++)
        for (int j=posl[i];j<=posr[i];j++)
            b[rank[j]]=i;
    for (int i=1;i<=now;i++)
        if (b[i]) l=1,r=n,ins(d[i],d[i-1],b[i]);
        else d[i]=d[i-1];
    while (q--)
    {
        int x=read(),y=read(),k=read();
        get_pts(now,rank[posl[k]],posr[k]-posl[k]+1);
        l=1,r=n;int ans=query(d[ur],d[ul-2],y);
        if (x>1) l=1,r=n,ans-=query(d[ur],d[ul-2],x-1);
        pri(ans);
    }
    return 0;
}