hdu 3948 The Number of Palindromes

后缀数组，求不同回文子串的数量

#include <STDIO.H>
#include <STRING.H>

// hdu 3948
const int MAXN = 200010;
int wa[MAXN], wb[MAXN], ws[MAXN], wv[MAXN];
int rank[MAXN], height[MAXN], r[MAXN], sa[MAXN];
inline int min(const int a, const int b){return a<b?a:b;}
inline void swap(int &x, int &y) {int a = x; x = y; y = a;}
bool cmp(int *r, int a, int b, int l)
{
return (r[a] == r[b] && r[a+l] == r[b+l]);
}
// r： 待排序的字符串，长度为n，最大值小于m
// ws：基数排序
// sa：存放排序结果
void da(int *r, int *sa, int n, int m)
{
//用倍增算法求后缀数组
int i, j, p, *x = wa, *y = wb, *t;
// 使用基数排序对长度为1的字符串进行排序
for (i = 0; i< m; ++i) ws[i] = 0; // 初始化ws
for (i = 0; i< n; i++) ws[x[i]=r[i]]++;
for (i = 1; i< m; i++) ws[i]+=ws[i-1];
for (i = n-1; i>= 0; --i) sa[--ws[x[i]]] = i;

// 进行若干次基数排序，基数排序要分两次，第一次是对第二关键字排序
// 第二次是对第一关键字排序。对第二关键字排序的结果可以利用上一次求得的sa直接算出
for (j=1, p=1; p< n; j*=2, m=p)
{
for (p=0, i=n-j; i< n; ++i) y[p++] = i;
//变量j是当前字符串的长度，数组y保存的是对第二关键字排序的结果
for (i=0; i< n; ++i) if (sa[i] >= j) y[p++] = sa[i] - j;
for (i=0; i< n; ++i) wv[i] = x[y[i]];
for (i=0; i< m; ++i) ws[i] = 0;
for (i=0; i< n; ++i) ws[wv[i]]++;
for (i=1; i< m; ++i) ws[i]+= ws[i-1];
for (i=n-1; i>=0; --i) sa[--ws[wv[i]]] = y[i];
// 计算rank 值
for (t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1; i< n; ++i)
x[sa[i]] = cmp(y, sa[i-1], sa[i], j)?p-1:p++;
}
}
/*
height：定义height[i]=suffix(sa[i-1])和suffix(sa[i])的最长公共前缀，
也就是排名相邻的两个后缀的最长公共前缀
*/
void calHeight(int *r, int *sa, int n)
{
int i, j, k = 0;
for (i = 1; i<= n; ++i) rank[sa[i]] = i;
for (i = 0; i< n; height[rank[i++]] = k)
for (k?k--:0, j = sa[rank[i]-1]; r[i+k] == r[j+k]; k++);
}
char str[MAXN];
// 将字符串翻转后用一个特殊字符间隔连接到原字符串后
int mGet() // r = str + '*' + reverse(str)
{
int l;
for (l = 0; str[l]; l++)
r[l] = str[l];
r[l] = '#';
for (int i = l-1; i >= 0; i--)
r[++l] = str[i];
r[++l] = 0;
return l;
}
int LOG[MAXN];
int st[MAXN][20];
/*
st[i][j]表示height[j]到height[j+2^i-1]的最小值
初始化st[i][0]=height[i]，st[i][j]=min(st[i][j-1],st[i+(1<<j-1)][j-1])
*/
void initRMQ(const int &n)
{
//初始化RMQ
int i, j, k, limit ;
for (i=0; i< n; ++i) st[i][0] = height[i];
k = LOG
;
for (j = 1; j<= k; ++j)
{
limit = n - (1<<(j-1));
for (i = 0; i<= limit; ++i)
st[i][j] = min(st[i][j-1], st[i+(1<<(j-1))][j-1]);
}
}

int query(int x, int y)
{
if (x > y) swap(x, y);
++x;
int k= LOG[y-x+1];
return min(st[x][k], st[y-(1<<k)+1][k]);
}

int solve(int n)
{
int ans = 0, st1 = 0, st2 = 0;
int j;
rank
= 0;
for (int i = 2; i<= n; ++i)
{
st1 = min(st1, height[i]);
j = query(i, rank[n-sa[i]-1]);
if (j > st1)
ans += j-st1, st1 = j;
st2 = min(st2, height[i]);
j = query(i, rank[n-sa[i]]);
if ( j > st2)
ans += j-st2, st2 = j;
}
return ans;
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("in.txt", "r", stdin);
#endif
int t, cs = 0;
int i;
// LOG[i] == log2 i ；
for (i = 1, LOG[0] = -1; i< MAXN; ++i) LOG[i] = LOG[i>>1] + 1;
scanf("%d", &t);
while (t--)
{
scanf("%s", str);
int n = mGet();
da(r, sa, n+1, 128);
calHeight(r, sa, n);
initRMQ(n+1);
printf("Case #%d: %d\n", ++cs, solve(n));
}
return 0;
}