O(n^2)复杂度。
#include
#include
using namespace std;
#define N 1200
string s;
int n, sa[4*N], rank[N], height[N];
int buf[4*N], ct[N], sx[N], sax[N];
inline bool leq(int a, int b, int x, int y);
inline bool leq(int a, int b, int c, int x, int y, int z);
inline int geti(int t, int nx, int sa[]);
//基数排序
static void radix(int a[], int b[], int s[], int n, int k);
/*
倍增算法,构造后缀数组的最坏时间复杂度为O(nlogn)。
参数:
int *r:待排序的字符串放在 r 数组中 , 从 r[0] 到 r[n-1] , 长度为 n , 且最大值小于 m 。
为了函数操作的方便,约定除r[n-1] 外所有的 r[i] 都大于 0, r[n-1]=0 。
int *sa:函数结束后,结果放在 sa 数组中,从 sa[0] 到 sa[n-1] 。
*/
void suffix(int s[], int sa[], int n, int k);
/*
后缀数组 SA 是一个一维数组,它保存 1..n 的某个排列SA[1],SA[2],……,SA[n],
并且保证 Suffix(SA[i]) < Suffix(SA[i+1]) , 1 ≤ i>s && s[0] != '#')
{
n = s.length() + 1;
makesa();
getRank();
lcp();
int ans = 0, minid, maxid;
//枚举字串长度h
for(int i = 1; i <= (n >> 1); i++)
{
minid = 1200, maxid = -1;
//对于每一次的h,利用height数组,找出连续的height大于等于h的里面最左端和最右端得为之l和r。
for(int j = 2; j < n; j++)
if (height[j] >= i)
{
if (sa[j - 1] < minid) minid = sa[j - 1];
if (sa[j - 1] > maxid) maxid = sa[j - 1];
if (sa[j] < minid) minid = sa[j];
if (sa[j] > maxid) maxid = sa[j];
}
else
{
//如果l + h - 1 < r的话,说明没有重叠,答案加1.
if (maxid != -1 && minid + i <= maxid) ans++;
minid = 1200, maxid = -1;
}
//如果l + h - 1 < r的话,说明没有重叠,答案加1.
if (maxid != -1 && minid + i <= maxid) ans++;
}
cout<= 2.
int i, j, e, p, t;
int name = 0, cx = -1, cy = -1, cz = -1;
int nx = (n+2)/3, ny = (n+1)/3, nz = n/3, nxz = nx+nz;
int *syz = s + n + 3, *sayz = sa + n + 3;
for (i=0, j=0; i < n + (nx - ny); i++)
if (i%3 != 0) syz[j++] = i;
radix(syz , sayz, s+2, nxz, k);
radix(sayz, syz , s+1, nxz, k);
radix(syz , sayz, s , nxz, k);
for (i = 0; i < nxz; i++)
{
if (s[ sayz[i] ] != cx || s[ sayz[i] + 1 ] != cy ||s[ sayz[i] + 2 ] != cz)
{
name++; cx = s[ sayz[i] ];
cy = s[ sayz[i] + 1 ]; cz = s[ sayz[i] + 2 ];
}
if (sayz[i] % 3 == 1) syz[ sayz[i] / 3 ] = name;
else syz[ sayz[i]/3 + nx ] = name;
}
if (name < nxz)
{
suffix(syz, sayz, nxz, name);
for (i = 0; i < nxz; i++) syz[sayz[i]] = i + 1;
}
else
{
for (i = 0; i < nxz; i++) sayz[syz[i] - 1] = i;
}
for (i = j = 0; i < nxz; i++)
if (sayz[i] < nx) sx[j++] = 3 * sayz[i];
radix(sx, sax, s, nx, k);
for (p=0, t=nx-ny, e=0; e < n; e++)
{
i = geti(t, nx, sayz); j = sax[p];
if ( sayz[t] < nx ?leq(s[i], syz[sayz[t]+nx], s[j], syz[j/3]) :
leq(s[i], s[i+1], syz[sayz[t]-nx+1],
s[j], s[j+1], syz[j/3+nx]) )
{
sa[e] = i;
if (++t == nxz)
{
for (e++; p < nx; p++, e++)
sa[e] = sax[p];
}
}
else
{
sa[e] = j;
if (++p == nx) for (++e; t < nxz; ++t, ++e)
sa[e] = geti(t, nx, sayz);
}
}
}
/*
后缀数组 SA 是一个一维数组,它保存 1..n 的某个排列SA[1],SA[2],……,SA[n],
并且保证 Suffix(SA[i]) < Suffix(SA[i+1]) , 1 ≤ i= 0 && s[i] != s[ sa[j-1] + k ])
height[j] = (k--), j = rank[ sa[j] + 1 ];
}