關於SA中基數排序的偷懶寫法

因爲我並沒有學習過標準的SA寫法,以前寫的一直是按照自己瞎yy的n*log^2來做的

然而被卡了好多次

做法就是記錄rk[]和rk2[]作爲第一和第二關鍵字進行sort//這樣做的話第一關鍵字有序  基數排序常數巨大

luogu模板n*log^2   70分

改成基數(巨大常數版)排序  9s AC

於是自己yy出一個不明複雜度的算法。。複雜度大概是n*log*log(n/log)

實測5s  AC

算法記錄sa[i]表示排名第i的位置是哪

先給sa按照rk排個序和普通的基排沒有任何區別

接下來我們考慮 

每次rk[sa[i]]都是有序的,而rk2[sa[i]]則是無序的

於是按照rk分段,每次只對rk相等的一段排序,這樣每次排序的長度可以認爲是n/log的?


#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<ctime>
#include<cstdlib>
#include<algorithm>
#include<queue>
#define For(i,j,k)	for(int i=j;i<=k;++i)
#define Dow(i,j,k)	for(int i=k;i>=j;--i)
#define ll long long
#define inf 1e9
using namespace std;
inline int read()
{
	int t=0,f=1;char c=getchar();
	while(!isdigit(c))	{if(c=='-')	f=-1;c=getchar();}
	while(isdigit(c))	t=t*10+c-'0',c=getchar();
	return t*f;
}
inline void write(int x){if(x>=10)	write(x/10);putchar(x%10+'0');}
inline void writeln(int x){write(x);puts("");}
inline void write_p(int x){write(x);putchar(' ');}
int sa[2000001],n,rk[2000001],rk2[2000001],tmp[2000001];
char s[2000001];
inline bool cmp(int x,int y)
{
	return rk[x]==rk[y]?rk2[x]<rk2[y]:rk[x]<rk[y];
}
inline void resort(int p[])
{
    int las=1;
    For(i,2,n)
        if(rk[p[i]]!=rk[p[i-1]])//處理las~i-1
        {
            sort(p+las,p+i,cmp);
            las=i;
        }
    sort(p+las,p+n+1,cmp);
}
int main()
{
	scanf("%s",s+1);n=strlen(s+1);
	For(i,1,n)	rk[i]=s[i],sa[i]=i;
	sort(sa+1,sa+n+1,cmp);
	for(int len=1;len<=n;len*=2)
	{
		For(i,1,n)	rk2[i]=i+len<=n?rk[i+len]:0;
		resort(sa);
		int tot=1;
		tmp[sa[1]]=1;
		For(i,2,n)	if(rk[sa[i]]==rk[sa[i-1]]&&rk2[sa[i]]==rk2[sa[i-1]])	tmp[sa[i]]=tot;else	tmp[sa[i]]=++tot;
		swap(rk,tmp);
	}
	For(i,1,n)	write_p(sa[i]);
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章