SA板子
- 代码
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1000010;
int n;
char s[maxn];
int sa[maxn],rank[maxn],height[maxn];
int p[maxn],tmp[maxn],cnt[maxn];
bool equal(int x,int y,int l){
return rank[x] == rank[y] && rank[x + l] == rank[y + l];
}
void doubling(){
for(int i=1;i<=n;++i){
rank[i]=s[i];
sa[i]=i;
}
for(int l=0,pos=0,sig=255;pos<n;sig=pos){
pos=0;
for(int i=n-l+1;i<=n;++i) p[++pos]=i;
for(int i=1;i<=n;++i)
if(sa[i]>l)p[++pos]=sa[i]-l;
//"p[i]表示第二关键字的后缀标号
memset(cnt,0,sizeof(int)*(sig+1));
for(int i=1;i<=n;++i) ++cnt[rank[i]];
for(int i=1;i<=sig;++i) cnt[i]+=cnt[i-1];
for(int i=n;i;--i) sa[cnt[rank[p[i]]]--]=p[i];
//"更新排名
pos=0;
for(int i=1;i<=n;++i) tmp[sa[i]]=equal(sa[i],sa[i-1],l)?pos:++pos;
for(int i=1;i<=n;++i) rank[i]=tmp[i];
l=!l?1:l<<1;
}
return;
}
int main(){
scanf("%s",s+1);
n=strlen(s+1);
doubling();
for(int i=1;i<=n;++i)
printf("%d ",sa[i]);
puts("");
return 0;
}
- UPD:加入了height和询问rank
#include<bits/stdc++.h>
using namespace std;
const int N=4e5+5;
typedef long long ll;
const int LOG=20;
char ch[N];int n;
ll a[N];
ll height[N];int sa[N],cnt[N],rk[N],tmp[N],p[N];
ll end[N];
ll st[N][20];
void get_height() {
for (int i = 1, j, k = 0; i <= n; i++, (k) ? (k--) : (0)) {
if (rk[i]>1) {
j = sa[rk[i] - 1];
while (i + k <= n && j + k <= n && ch[i + k] == ch[j + k]) k++;
height[rk[i]] = (ll)k;
}
}
}
void get_ST() {
for (int i = 1; i <= n; i++) {
st[i][0] = height[i];
}
for (int j = 1; j <= LOG - 1; j++) {
for (int i = (1 << j); i <= n; i++) {
st[i][j] = min(st[i][j - 1], st[i - (1 << (j - 1))][j - 1]);
}
}
}
inline void buildSA(){
for(int i=1;i<=n;i++){
rk[i]=ch[i],sa[i]=i;
}
for(int l=0,sz=255,pos=0;pos<n;sz=pos){
pos=0;
for(int i=n-l+1;i<=n;++i) p[++pos]=i;
for(int i=1;i<=n;i++) if(sa[i]>l) p[++pos]=sa[i]-l;
memset(cnt,0,sizeof(int)*(sz+1));
for(int i=1;i<=n;++i) cnt[rk[i]]++;
for(int i=1;i<=sz;i++) cnt[i]+=cnt[i-1];
for(int i=n;i;i--) sa[cnt[rk[p[i]]]--]=p[i];
pos=0;
for(int i=1;i<=n;i++)tmp[sa[i]]=(rk[sa[i]]==rk[sa[i-1]] && rk[sa[i]+l]==rk[sa[i-1]+l]) ? pos:++pos;
for(int i=1;i<=n;i++)rk[i]=tmp[i];
l= !l?1:l<<1;
}
get_height();
get_ST();
//for(int i=1;i<=n;i++)cout<<height[i]<<" "<<sa[i]<<endl;
for(int i=n;i;i--){
end[i]=end[i+1]+(ll)(n-sa[i]+1)-height[i];
//cout<<end[i]<<" ";
}
//end[n]=1;
}
ll getrank(int l, int r) {
int x = rk[l], len = r - l + 1;
for (int k = LOG - 1; k >= 0; k--) {
if (st[x][k] >= len) {
x -= 1 << k;
}
}
return end[x + 1] + ll(n - sa[x] + 1 - len + 1);
}
- UPD:子串LCP以及第k大子串(重复不算)
#include<bits/stdc++.h>
using namespace std;
const int N=3e5+5;
char ch[100010];
int n,m,sa[N],p[N],tmp[N],rank[N],cnt[N];
typedef long long ll;
ll height[N],aft[N];
inline void build_height() {
for(int i=1,j,k=0; i<=n; i++,k=max(0,k-1)) {
if(rank[i]>1) {
j=sa[rank[i]-1];
while(i+k<=n&&j+k<=n&&ch[i+k]==ch[j+k])k++;
height[rank[i]]=(ll)k;
}
}
}
int st[N][20];
int LG[N];
inline void buildST() {
for(int i=1; i<=n; i++)st[i][0]=height[i];
for(int i=2; i<=n; i++)LG[i]=LG[i>>1]+1;
for(int j=1; (1<<j)<=n; j++)
for(int i=1; i+(1<<j)-1<=n; i++) {
st[i][j]=min(st[i][j-1],st[i+(1<<(j-1))][j-1]);
}
}
inline int findlcp(int x,int y) {
int L=rank[x],R=rank[y];
if(L>R)swap(L,R);
if(L==R)return n+100;
L++;
int logg=LG[R-L+1];
return min(st[L][logg],st[R-(1<<logg)+1][logg]);
}
inline void build() {
for(int i=1; i<=n; i++) {
sa[i]=i,rank[i]=ch[i];
}
for(int l=0,pos=0,sz=255; pos<n; sz=pos,l=!l?1:l<<1) {
pos=0;
for(int i=n-l+1; i<=n; i++)p[++pos]=i;
for(int i=1; i<=n; i++)if(sa[i]>l)p[++pos]=sa[i]-l;
memset(cnt,0,sizeof(int)*(sz+1));
for(int i=1; i<=n; i++)++cnt[rank[i]];
for(int i=1; i<=sz; i++)cnt[i]+=cnt[i-1];
for(int i=n; i; i--) sa[cnt[rank[p[i]]]--]=p[i];
pos=0;
for(int i=1; i<=n; i++)tmp[sa[i]]=(rank[sa[i]]==rank[sa[i-1]]&&rank[sa[i]+l]==rank[sa[i-1]+l])?pos:++pos;
for(int i=1; i<=n; i++)rank[i]=tmp[i];
}
build_height();
buildST();
for(int i=n; i; i--)
aft[i]=aft[i+1]+n-sa[i]+1-height[i];
for(int i=1; i<n; i++)
aft[i]=aft[i+1];
aft[n]=0;
}
typedef pair<int,int> pii;
inline pii findKth(ll R) {
int l=1,r=n;
while(l<r) {
int mid=(l+r)>>1;
if(aft[mid]>=R) l=mid+1;
else r=mid;
}
R-=aft[l];
return pii(sa[l],n-R+1);
}
bool cmp(int l1,int r1,int l2,int r2) {
int lcp=findlcp(l1,l2);
int len1=r1-l1+1,len2=r2-l2+1;
if(lcp>=len1&&lcp>=len2) return len1<len2;
else if(lcp>=len1) return true;
else if(lcp>=len2) return false;
else return rank[l1]<rank[l2];
}
int main() {
scanf("%d",&m);
scanf("%s",ch+1);
n=strlen(ch+1);
build();
}