HDU 4622 后缀数组（求区间子串数）

题目链接

题意：给一段字符串，多次询问，每次询问求一段区间内子串的数量。（子串不能相同）

思路：后缀数组求字符串数量时，是按照字典序遍历Height数组计算的。那么应用到此题，最简单的想法当然是直接暴力每个询问（提取每个询问的串作为一个字符串求得答案），但是暴力太笨拙（此题暴力多交几次居然可以通过！！！）。再次回忆一下通过Height数组计数的方式，有一个很重要的点——按照字典序遍历。抓住这一点之后，我们在原串中遍历并进行计数即可获得答案。

C++代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2010;

int n;
char s[maxn];
int ch[maxn];
int Sa[maxn];
int Rank[maxn];
int Height[maxn];
int t1[maxn],t2[maxn],c[maxn];

bool cmp( int *r , int a , int b , int l )
{
	return r[a]==r[b]&&r[a+l]==r[b+l];
}

void SA( int m )
{
	n++; int *x=t1,*y=t2;
	for ( int i=0 ; i<m ; i++ ) c[i] = 0;
	for ( int i=0 ; i<n ; i++ ) c[x[i] = ch[i]]++;
	for ( int i=1 ; i<m ; i++ ) c[i] += c[i-1];
	for ( int i=n-1 ; i>=0 ; i-- ) Sa[--c[x[i]]] = i;
	for ( int j=1,p ; j<=n ; j<<=1 )
	{
		p = 0;
		for ( int i=n-j ; i<n ; i++ ) y[p++] = i;
		for ( int i=0 ; i<n ; i++ ) if ( Sa[i]>=j ) y[p++] = Sa[i]-j;
		for ( int i=0 ; i<m ; i++ ) c[i] = 0;
		for ( int i=0 ; i<n ; i++ ) c[x[y[i]]]++;
		for ( int i=1 ; i<m ; i++ ) c[i] += c[i-1];
		for ( int i=n-1 ; i>=0 ; i-- ) Sa[--c[x[y[i]]]] = y[i];
		swap( x , y );
		p = 1;
		x[Sa[0]]  = 0;
		for ( int i=1 ; i<n ; i++ )
			x[Sa[i]] = cmp ( y , Sa[i-1] , Sa[i] , j )?p-1:p++;
		if ( p>=n ) break;
		m = p;	
	}
	for ( int i=0 ; i<n ; i++ ) Rank[Sa[i]] = i;
	n--;
	int l = 0;
	for ( int i=0 ; i<n ; i++ )
	{
		if ( l ) l--;
		int j = Sa[Rank[i]-1];
		while ( ch[i+l]==ch[j+l] ) l++;
		Height[Rank[i]] = l;
	}
}

int rmq[20][maxn];

void rmq_init()
{
	for ( int i=0 ; i<=n ; i++ ) rmq[0][i] = Height[i];
	for ( int i=1 ; (1<<i)<=n+1 ; i++ )
		for ( int j=0 ; j+(1<<i)<=n+1 ; j++ )
			rmq[i][j] = min( rmq[i-1][j] , rmq[i-1][j+(1<<(i-1))] );  
}

int rmq_query( int l , int r )
{
	int k = 0;
	while ( 1<<(k+1)<=r-l+1 ) k++;
	return min( rmq[k][l] , rmq[k][r-(1<<k)+1] );
}

int slove( int l , int r )
{
	int ans = (r-l+1)*(r-l+2)/2;
	int las = -1;
	for ( int i=1 ; i<=n ; i++ )
	{
		if ( Sa[i]+1<l||Sa[i]+1>r ) continue;
		if ( las==-1 )
		{
			las  = i;
			continue;
		}
		int a = las;
		int b = i;
		int lp = rmq_query( a+1 , b );
		int la = r-Sa[a];
		int lb = r-Sa[b];
		ans -= min( lp , min( la , lb ) );
		if ( la<lb||lp<lb ) las = i;
	}
	return ans;
}

int main()
{
    int T; scanf ( "%d" , &T );
    while ( T-- )
    {
    	scanf ( "%s" , s );
		n = strlen(s);	
		for ( int i=0 ; i<n ; i++ )
			ch[i] = s[i]-'a'+1;
		ch[n] = 0;
		SA ( 27 );
		rmq_init();	
		int Q; scanf ( "%d" , &Q );
		while ( Q-- )
		{
			int l,r; scanf ( "%d%d" , &l , &r );
			printf  ( "%d\n" , slove( l , r ) );
		}
	}
    return 0;
}