标签:
这是一道字符串的题,看到后,第一反应也是把他们全部弄出来后排序,但看了以下数据范围,别说时间了,空间可能都开不下。转而去想数据结构。处理字符串的有trie, 后缀数组, 还有基于哈希值的 LCP, 稍微想了一下, 基于哈希值的 LCP 能进行字符串匹配, 但是比较大小嘛!我就不知道怎么弄了,trie 能作为字典,但要比较大小……(会不会很麻烦,我不大清楚啊!)。但是后缀数组之只看过一次,而且还不大会打,趁这个机会就学一下吧!毕竟很不错。据说,后缀数组有两种写法,我就先只学倍增好了,昨晚花了一些时间看懂了前半部分,但后半部分还不大懂。就把它当成黑盒代码写吧! 要是有大神知道后半段运行的原理,请求指教。
这就是后缀数组倍增的代码了(从lrj白书抄的)
1 #define rep(i,j,k) for(int i = j; i <= k; i++) 2 #define down(i,j,k) for(int i = j; i >= k; i--) 3 4 char s[maxn]; 5 int sa[maxn], t[maxn], t2[maxn], c[maxn], n; 6 7 void build_sa(int m) 8 { 9 int i, *x = t, *y =t2; 10 rep(i,0,m-1) c[i] = 0; 11 rep(i,0,n-1) c[x[i] = s[i]]++; 12 rep(i,1,m-1) c[i] += c[i-1]; 13 down(i,n-1,1) sa[--c[x[i]]] = i; 14 for(int k = 1; k <= n; k <<= 1){ 15 int p = 0; rep(i,n-k,n-1) y[p++] = i; 16 rep(i,0,n-1) if( sa[i] >= k ) y[p++] = sa[i] - k; 17 rep(i,0,m-1) c[i] = 0; 18 rep(i,0,n-1) c[x[y[i]]]++; 19 rep(i,1,m-1) c[i] += c[i-1]; 20 down(i,n-1,0) sa[--c[x[y[i]]]] = y[i]; 21 swap(x,y); 22 p = 1, x[sa[0]] = 0; 23 rep(i,1,n-1) 24 x[sa[i]] = y[sa[i]] == y[sa[i-1]] && y[sa[i-1]+k] == y[sa[i]+k] ? p -1 : p++; 25 if( p >= n ) break; 26 m = p; 27 } 28 }
sa[i]里存的是排名为 i 的后缀的首个字符的位置。
接下来就是此题代码了(知道后缀后这就不难了)
1 #include<cstdio> 2 #include<iostream> 3 #include<cstring> 4 #define rep(i,j,k) for(int i = j; i <= k; i++) 5 #define down(i,j,k) for(int i = j; i >= k; i--) 6 #define maxn 200100 7 using namespace std; 8 9 char s[maxn]; 10 int sa[maxn], t[maxn], t2[maxn], c[maxn], n; 11 12 void build_sa(int m) 13 { 14 int i, *x = t, *y =t2; 15 rep(i,0,m-1) c[i] = 0; 16 rep(i,0,n-1) c[x[i] = s[i]]++; 17 rep(i,1,m-1) c[i] += c[i-1]; 18 down(i,n-1,1) sa[--c[x[i]]] = i; 19 for(int k = 1; k <= n; k <<= 1){ 20 int p = 0; rep(i,n-k,n-1) y[p++] = i; 21 rep(i,0,n-1) if( sa[i] >= k ) y[p++] = sa[i] - k; 22 rep(i,0,m-1) c[i] = 0; 23 rep(i,0,n-1) c[x[y[i]]]++; 24 rep(i,1,m-1) c[i] += c[i-1]; 25 down(i,n-1,0) sa[--c[x[y[i]]]] = y[i]; 26 swap(x,y); 27 p = 1, x[sa[0]] = 0; 28 rep(i,1,n-1) 29 x[sa[i]] = y[sa[i]] == y[sa[i-1]] && y[sa[i-1]+k] == y[sa[i]+k] ? p -1 : p++; 30 if( p >= n ) break; 31 m = p; 32 } 33 } 34 35 int main() 36 { 37 scanf("%s", s); 38 n = strlen(s); 39 rep(i,0,n-1){ 40 s[i+n] = s[i]; 41 } 42 n *= 2; 43 build_sa(128); 44 int size = n / 2; 45 rep(i,0,n-1){ 46 if( sa[i] < size ){ 47 printf("%c", s[sa[i]+size-1] ); 48 } 49 } 50 cout<<endl; 51 return 0; 52 }
BZOJ 1031: [JSOI2007]字符加密Cipher
标签:
原文地址:http://www.cnblogs.com/83131yyl/p/5075673.html