POJ3415: Common Substring 题解

先把两个串拼起来，中间用不会出现的字符连一下

然后求一下后缀数组和高度数组

然后可以用一个单调栈维护一下lcp数组

对于来自A的后缀，我们要统计它后面的来自B的后缀的答案，同样对于来自B的后缀，我们要统计它后面的来自A的后缀的答案，这两个部分是一样的，这里只讨论第一个

从后往前做，每遇到一个来自B的后缀，就把它压进单调栈，并把栈顶的那些大于它的元素弹掉

注意你每弹掉一个元素，它的答案都会变成现在新加进来的这个位置的lcp，所以还需要一个变量记录当前位置踩掉了多少个位置

用一个变量cnt实时记录当前栈内所有的元素的贡献 (lcp-k+1)*它身上累计的元素个数

每遇到一个来自A的后缀，ans+=cnt

注意后缀数组模板不要敲错

注意ans要开long long

#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <cstdlib>
#include <utility>
#include <vector>
#include <algorithm>
#include <queue>
#include <cmath>
#include <deque>
#include <stack>
#include <cmath>
#include <set>
#include <map>
#define LL long long
#define LB long double
#define Pair pair<int,int>
#define x first
#define y second
#define pb push_back
#define mp make_pair
#define LOWBIT(x) x & (-x)
using namespace std;
 
const int INF=2e9;
const LL LINF=2e16;
const int MOD=1e9+7;
const int magic=348;
const double eps=1e-10;
 
inline int getint()
{
    char ch;int res=0;bool f;
    while (!isdigit(ch=getchar()) && ch!='-') {}
    if (ch=='-') f=false,res=0; else res=ch-'0',f=true;
    while (isdigit(ch=getchar())) res=res*10+ch-'0';
    return f?res:-res;
}

int len,k,K;
char s1[300048],s2[300048],s[300048];
int sa[300048],tmp[300048],rank[300048],lcp[300048];

struct node
{
	LL val;
	LL num;
};
deque<node> q;

bool compare_sa(int x,int y)
{
	if (rank[x]!=rank[y]) return rank[x]<rank[y];
	int tmp1=x+k<=len?rank[x+k]:-1,tmp2=y+k<=len?rank[y+k]:-1;
	return tmp1<tmp2;
}

void construct_sa()
{
	int i,cnt;
	for (i=1;i<=len;i++)
	{
		sa[i]=i;
		rank[i]=s[i];
	}
	for (k=0;k<=len;k==0?k++:k*=2)
	{
		sort(sa+1,sa+len+1,compare_sa);
		tmp[sa[1]]=1;cnt=0;
		for (i=2;i<=len;i++)
		{
			tmp[sa[i]]=tmp[sa[i-1]]+(compare_sa(sa[i-1],sa[i])?1:0);
			cnt+=(tmp[sa[i]]==tmp[sa[i-1]]?0:1);
		}
		for (i=1;i<=len;i++) rank[i]=tmp[i];
		if (cnt>=len-1) break;
	}
}

void construct_lcp()
{
	int i,j,h=0;
	for (i=1;i<=len;i++) rank[sa[i]]=i;
	for (i=1;i<=len;i++)
	{
		j=sa[rank[i]-1];
		if (h) h--;
		while (i+h<=len && j+h<=len && s[i+h]==s[j+h]) h++;
		lcp[rank[i]-1]=h;
	}
}

int main ()
{
	int i,ll;
	LL cnt,ans=0;
	K=getint();
	while (K)
	{
		memset(s,'\0',sizeof(s));
		memset(s1,'\0',sizeof(s1));
		scanf("%s",s+1);
		len=strlen(s+1);s[++len]='&';
		scanf("%s",s1+1);
		ll=strlen(s1+1);
		for (i=len+1;i<=len+ll;i++) s[i]=s1[i-len];
		len=strlen(s+1);
		construct_sa();
		construct_lcp();
		while (!q.empty()) q.pop_front();
		cnt=0;ans=0;node ins;
		for (i=len-1;i>=1;i--)
		{
			ins.val=lcp[i];ins.num=(sa[i+1]<=len/2+1)?0:1;
			while (!q.empty() && q.back().val>ins.val)
			{
				cnt-=(long long)(q.back().val-K+1)*q.back().num;
				ins.num+=q.back().num;
				q.pop_back();
			}
			if (ins.val>=K)
			{
				cnt+=(long long)(ins.val-K+1)*ins.num;
				q.pb(ins);
			}
			if (sa[i]<=len/2) ans+=cnt;
		}
		cnt=0;while (!q.empty()) q.pop_front();
		for (i=len-1;i>=1;i--)
		{
			ins.val=lcp[i];ins.num=(sa[i+1]>len/2)?0:1;
			while (!q.empty() && q.back().val>ins.val)
			{
				cnt-=(long long)(q.back().val-K+1)*q.back().num;
				ins.num+=q.back().num;
				q.pop_back();
			}
			if (ins.val>=K)
			{
				cnt+=(long long)(ins.val-K+1)*ins.num;
				q.pb(ins);
			}
			if (sa[i]>len/2+1) ans+=cnt;
		}
		cout<<ans<<endl;
		K=getint();
	}
	return 0;
}