题解:
理论上后缀数组的求解要使用倍增orDC3
算法来求解,但是这显然超出了我的水平范围不符合题目的要求。所以我们就正大光明得使用Hash + Sort来做。
题目比较难处理的点在于,我们对于两个后缀字符串的比较,因为比较的最坏时间复杂度为
O
(
n
)
O(n)
O(n),但是我们快排就已经有了
O
(
n
l
o
g
N
)
O(nlog{N})
O(nlogN)的复杂度,显然对于3e5的数据肯定会超时,那我们应该怎么去处理呢?其实和之前的题目相似,我们使用二分答案的方法做,我们二分得出两个后缀字符串重合的最大长度,然后去找下一位比较就可以得出两个字符串的大小关系了。这里找出最大长度的部分我们就是采用了字符串Hash。
AC代码:
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
#include<stack>
#include<queue>
#include<bitset>
#include<list>
#define ll long long
#define INF 1e12
#define inf -1e12
#define PII pair<int,int>
#define ull unsigned long long
#define ios std :: ios :: sync_with_stdio(false)
using namespace std;
const int maxn = 3e5 + 19;
const int P = 131;
ull p[maxn],h[maxn];
int sa[maxn],len;
string s;
ull gethash(int l,int r)
{
return h[r] - h[l - 1] * p[r - l + 1];
}
int get_max(int a,int b)
{
int l = 0,r = min(len - a + 1,len - b + 1);
while(l < r){
int mid = (l + r + 1) >> 1;
if(gethash(a,a + mid - 1) == gethash(b,b + mid - 1)) l = mid;
else r = mid - 1;
}
return l;
}
bool cmd(int a,int b)
{
int cnt = get_max(a,b);
int a_val,b_val;
a_val = a + cnt > len ? inf : s[a + cnt];
b_val = b + cnt > len ? inf : s[b + cnt];
return a_val < b_val;
}
int main()
{
ios;
cin >> s;
len = s.length();
s = "#" + s;
p[0] = 1;
for(int i = 1;i <= len;i++) {
h[i] = h[i - 1] * P + s[i] - 'a' + 1;
p[i] = p[i - 1] * P;
sa[i] = i;
}
sort(sa + 1,sa + len + 1,cmd);
for(int i = 1;i < len;i++) cout << sa[i] - 1 << " ";
cout << sa[len] - 1 << endl;
for(int i = 1;i <= len;i++){
if(i == 1) cout << 0;
else cout << ' ' << get_max(sa[i],sa[i - 1]);
}
cout << endl;
return 0;
}
PS:关闭同步流后反而比scanf
快也是很迷惑。