E - Common Substrings POJ - 3415

https://cn.vjudge.net/contest/311974#problem/E 

思路：单调栈计算； 计算A对B的贡献再计算B对A的贡献；A对B：遇到B串时，计算其和前面的串的贡献；而它和前面的lcp是的最小值；故维护栈单调递增；计算方法：按前面最大贡献计算；一旦遇到比栈顶小就将栈顶的贡献减掉；

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <queue>
#include <map>
#include <set>
#include <stack>
using namespace std;

#define sfi(x) scanf("%d",&x)
#define sfc(x) scanf("%c",x)
#define sfl(x) scanf("%lld",&x)
#define sfs(x) scanf("%s",x)

#define rint register int
#define pb push_back
#define fl() printf("flag!\n")
#define INF 0x3f3f3f3f
#define ll long long
#define mem(x,y) memset(x,y,sizeof(x))
#define FAST_IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)

const int maxn=1e6+9;
const int mod=1e9+7;

char s[maxn];
int b[maxn];
int t1[maxn],t2[maxn],rk[maxn],sa[maxn],c[maxn],height[maxn];
int n,m;

void get_sa(char *s,int n,int m)
{
    int *x=t1,*y=t2;
    int i,k;
    for(i=0;i<m;i++) c[i]=0;
    for(i=0;i<n;i++) c[x[i]=s[i]]++;
    for(i=1;i<m;i++) c[i]+=c[i-1];
    for(i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
    for(k=1;k<=n;k<<=1)
    {
        int p=0;
        for(i=n-k;i<n;i++) y[p++]=i;
        for(i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        for(i=0;i<m;i++) c[i]=0;
        for(i=0;i<n;i++) c[x[y[i]]]++;
        for(i=1;i<m;i++) c[i]+=c[i-1];
        for(i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        x[sa[0]]=0;
        p=1;
        for(i=1;i<n;i++)
            x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]) ? p-1:p++;
        if(p>=n) break;
        m=p;
    }
}


void get_height(char *s,int n)
{
    int i,j,k=0;
    for(i=1;i<=n;i++) rk[sa[i]]=i;
    for(i=0;i<n;i++)
    {
        if(k) --k;
        else k=0;
        j=sa[rk[i]-1];
        while(s[i+k]==s[j+k]) k++;
        height[rk[i]]=k;
    }
}

void debug(int n)
{
    for(int i=1;i<=n;i++)
    {
        for(int j=sa[i];j<n;j++) cout<<s[j];
        cout<<endl;
    }
}

struct STACK
{
    int h;
    int num;
    STACK()
    {
        h=0;
        num=0;
    }
}st[maxn];
ll top;
ll tot;
ll solve(int n,int k,int index)
{
    ll ans=0;

    tot=0;
    top=0;
    for(int i=1;i<=n;i++)
    {
        if(height[i]<k)
        {
            tot=0;
            top=0;
            continue;
        }
        int cnt=0;
        if(sa[i-1]<index)
        {
            cnt++;
            tot+=height[i]-k+1;
        }
        while(top>0&&height[i]<=st[top-1].h)
        {
            top--;
            tot-=st[top].num*(st[top].h-height[i]);
            cnt+=st[top].num;
        }
        st[top].h=height[i];
        st[top++].num=cnt;
        if(sa[i]>index) ans+=tot;
    }

    tot=0;
    top=0;
    for(int i=1;i<=n;i++)
    {
        if(height[i]<k)
        {
            tot=0;
            top=0;
            continue;
        }
        int cnt=0;
        if(sa[i-1]>index)
        {
            cnt++;
            tot+=height[i]-k+1;

        }
        while(top>0&&height[i]<=st[top-1].h)
        {
            top--;
            tot-=st[top].num*(st[top].h-height[i]);
            cnt+=st[top].num;
        }
        st[top].h=height[i];
        st[top++].num=cnt;
        if(sa[i]<index) ans+=tot;
    }

    return ans;
}

int main()
{
    //FAST_IO;
    //freopen("input.txt","r",stdin);

    int k;
    while(~sfi(k)&&k)
    {
        int index=0;
        sfs(s);
        int len=strlen(s);
        s[len]='#';
        index=len;
        sfs(s+len+1);
        n=strlen(s);
        s[n]=0;
        m=400;

        //cout<<s<<endl;

        get_sa(s,n+1,m);
        get_height(s,n);
        //debug(n);

        ll ans=solve(n,k,index);

        printf("%lld\n",ans);


    }


    return 0;

}