P3181 [HAOI2016]找相同字符（后缀数组+单调栈）

题意：

给定两个字符串，求出在两个字符串中各取出一个子串使得这两个子串相同的方案数。两个方案不同当且仅当这两个子串中有一个位置不同。

题解：

首先我们想到从A，B两个串中选择两个子串相同肯定不好选，但我们可以用容斥，先计算出A和B拼接在一起的总的子串相同的方案数，再减去两个都在A中选，和两个都在B中选的情况，求解方案数时可以用单调栈求出。

单调栈：
处理到 排名为 i的串的时候，如果栈顶的height > height[i], 就弹出，这样在栈顶和当前串之间的串的height 必然是比当前串大的所以这些串的贡献就等于当前串的 height，而对于比栈顶串更靠前的串，因为栈顶串的height 是比当前串的height 小的，所以他们的贡献是在栈顶串之前就决定的，可以直接加上，用一个数组求出区间内的有贡献串的数量就可以了。

---

AC代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int MAXN = 4e5+10;
string s1,s2;
int x[MAXN],y[MAXN],c[MAXN];
int sa[MAXN],rk[MAXN],h[MAXN];
LL a[MAXN];
void init(){
    memset(h,0,sizeof(h));
    memset(y,0,sizeof(y));
    memset(rk,0,sizeof(rk));
    memset(c,0,sizeof(c));
    memset(a,0,sizeof(a));
}
LL solve(string s){
    init();
    int n = s.size(),m = (int)('~');
    s = "~" + s;
    for(int i=1;i<=n;i++) x[i]=s[i],++c[x[i]];
    for(int i=1;i<=m;i++) c[i]+=c[i-1];
    for(int i=1;i<=n;i++) sa[c[x[i]]--] = i;
    for(int k=1;k<=n;k<<=1){
        int num = 0;
        for(int i=n-k+1;i<=n;i++) y[++num] = i;
        for(int i=1;i<=n;i++)
            if(sa[i]>k)
                y[++num] = sa[i]-k;
        for(int i=1;i<=m;i++) c[i]=0;
        for(int i=1;i<=n;i++) ++c[x[i]];
        for(int i=1;i<=m;i++) c[i]+=c[i-1];
        for(int i=n;i>=1;i--)
            sa[c[x[y[i]]]--] = y[i];
        swap(x,y);
        x[sa[1]] = num = 1;
        for(int i=2;i<=n;i++)
            x[sa[i]] = (y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k] ? num : ++num);
        if(num == n) break;
        m = num;
    }

    int k = 0;
    for(int i=1;i<=n;i++) rk[sa[i]] = i;
    for(int i=1;i<=n;i++){
        if(k) --k;
        int j=sa[rk[i]-1];
        while(i+k<=n && j+k<=n && s[i+k]==s[j+k]) ++k;
        h[rk[i]] = k;
    }
    LL res = 0;
    stack<LL> v;
    for(int i=2;i<=n;i++){
        while(!v.empty() && h[v.top()]>=h[i]) v.pop();
        if(!v.empty()) a[i] = a[v.top()]+h[i]*(i-v.top());
        else a[i] = h[i]*(i-1);
        v.push(i);
        res += a[i];
    }
    return res;
}
int main(){
    ios::sync_with_stdio(false);
    cin>>s1>>s2;
    LL res,ans1,ans2;
    res = solve(s1+"~"+s2);
    ans1 = solve(s1);
    ans2 = solve(s2);
    printf("%lld\n",res-ans1-ans2);
}