[洛谷P6292]区间本质不同子串个数

题目

传送门 to luogu

思路

首先看一眼标签啊，后缀自动机？懂了，询问按照右端点排序。

再想想这个题啊，对于固定右端点，左端点在一个区间内很有可能有规律的变换。肯定是用数据结构维护呗。

现在只看右端点的移动带来的影响。无非就是增加了一些后缀。假设长度为 $p$ 的后缀，原本的 $\max endpos$ 为 $x$ ，那么区间加 $[x-p+2,n-p+1]$ 即可，因为左端点属于这个区间内的时候，原来的那个没有被包含，而这个新的被包含了。

所以就是跳后缀树。然而后缀树上的点代表连续区间。也就是说，$p$ 是一个连续区间。手玩可知，加入的是一个梯形——左边是等差，中间是区间加，右边是等差。线段树显然可以维护。

不过，用 $\mathtt{B}\mathtt{I}\mathtt{T}$ 也挺不错，因为只有单点查询，维护差分数组，询问即为区间查询。而此时的等差数列，就是区间修改了。

然后就通过了。虽然复杂度是错误的。正确的复杂度要用 $\mathtt{L}\mathtt{C}\mathtt{T}$ 维护后缀树，一条实链上的点都满足 $\max endpos$ 相同，可以一起改。复杂度就是 $\mathcal{O}(n{\log }^{2}n)$ 的。

代码

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long int_;
inline int readint(){
	int a = 0; char c = getchar(), f = 1;
	for(; c<'0'||c>'9'; c=getchar())
		if(c == '-') f = -f;
	for(; '0'<=c&&c<='9'; c=getchar())
		a = (a<<3)+(a<<1)+(c^48);
	return a*f;
}

const int MaxN = 200005;

int n; // range of BIT
namespace BIT{
	int_ c1[MaxN>>1], c2[MaxN>>1];
	void modify(int l,int r,int_ v){
//		printf("modify %d %d %lld\n",l,r,v);
		for(int i=l; i<=n; i+=(i&-i)){
			c1[i] += v; // diff
			c2[i] += v*l; // with val
		}
		for(int i=r+1; i<=n; i+=(i&-i)){
			c1[i] -= v; // diff
			c2[i] -= v*(r+1);
		}
	}
	int_ query(int l,int r){
		int_ sum = 0, sy = 0;
		for(int i=r; i; i-=(i&-i))
			sum += c1[i], sy += c2[i];
		for(int i=l-1; i; i-=(i&-i))
			sum -= c1[i], sy -= c2[i];
		return sum*(r+1)-sy;
	}
}

const int CharSiz = 26;
namespace SAM{
	int ch[MaxN][CharSiz];
	int fa[MaxN], len[MaxN];
	int cntNode = 1, lst = 1;
	int ep[MaxN]; // endpos
	void add(char c){
//		printf("added %c\n",c+'a');
		int p = lst; lst = ++ cntNode;
		int now = lst;
		len[now] = len[p]+1;
		for(; p&&!ch[p][c]; p=fa[p])
			ch[p][c] = now;
		if(!p) return void(fa[now] = 1);
		int x = ch[p][c];
		if(len[x] == len[p]+1)
			return void(fa[now] = x);
		int nx = ++ cntNode; // split
		memcpy(ch[nx],ch[x],CharSiz<<2);
		ep[nx] = ep[x];
		fa[nx] = fa[x], fa[now] = nx;
		fa[x] = nx, len[nx] = len[p]+1;
		for(; ch[p][c]==x; p=fa[p])
			ch[p][c] = nx; // change
	}
	void adjust(int N){
//		printf("adjust %d\n",N);
		ep[lst] = N; // = len[lst]
		BIT::modify(1,1,N-len[fa[lst]]);
		BIT::modify(2,N-len[fa[lst]]+1,-1);
		for(int i=fa[lst]; i!=1; i=fa[i]){
//			BIT::modify(1,1,-(len[i]-len[fa[i]]));
			BIT::modify(ep[i]-len[i]+2,ep[i]-len[fa[i]]+1,1);
			
//			BIT::modify(1,1,len[i])
			BIT::modify(N-len[i]+2,N-len[fa[i]]+1,-1);
			ep[i] = N; // remember to update
		}
	}
}

struct Query{
	int l, r, id;
	bool operator < (const Query &t) const {
		return r < t.r;
	}
};
Query ask[MaxN];

char xez[MaxN];
int_ ans[MaxN];
int main(){
	scanf("%s",xez+1);
	n = strlen(xez+1);
	int q = readint();
	for(int i=1; i<=q; ++i){
		ask[i].id = i;
		ask[i].l = readint();
		ask[i].r = readint();
	}
	sort(ask+1,ask+q+1);
	for(int i=1,p=1; i<=n; ++i){
		SAM::add(xez[i]-'a');
		SAM::adjust(i); // essential
		while(p <= q && ask[p].r == i){
			ans[ask[p].id] = BIT::query(1,ask[p].l);
			++ p; // don't forget it
		}
//		printf("p = %d\n",p);
	}
	for(int i=1; i<=q; ++i)
		printf("%lld\n",ans[i]);
	return 0;
}