hdoj 5008 字尾陣列+RMQ+二分
阿新 • • 發佈:2019-02-01
題意:給一個字串,問該字串第k個子串的最小字典序位置。其中k需要L xor R xor V求得,L R是前一個詢問的子串位置,說白了就是強制線上。
思路:先是字尾陣列處理出子串去重後的編號,然後先求得第k個子串的位置i,因為這個位置只是字尾陣列中第一次出現的位置,而不一定是原字串中最左邊的位置,所以真正的位置可能在i的左邊或右邊。如果遍歷去找肯定會超時,但是我們可以二分。
假設第k個子串的長度為d,起始下標為i,真實下標為j,那麼i到j的公共字首一定大於等於d,即height[i] >= d, height[i + 1] >=d ....height[j]>=d(j也可能在i的左邊),那麼我們只要保證min(height[i],height[i + 1]...height[j])>= d就可以了,這就用到了RMQ,我這裡是用線段樹實現的。
這樣就可以求出含有公共字首的最大範圍L和R(L<=i, R>=i)。
然後只要求出[L,R]中最小的sa就可以了,這裡就用到了另一個RMQ。
所以我們一共需要兩個RMQ,一個用來查詢區間最小公共字首長度(height),一個用來查詢區間最小字尾起始位置(sa)。
另外還有一些細節,字尾陣列得到的答案L,R是以0開始的,xor和輸出答案的時候記得加1(我就這裡wa了好多次。。。TAT),還有k、v是long long。
#include <cstdlib> #include <cstring> #include <cstdio> #include <algorithm> using namespace std; const int MAXN = 100010; int sa[MAXN]; int t1[MAXN],t2[MAXN],c[MAXN]; int Rank[MAXN],height[MAXN]; void build_sa(int s[],int n,int m) { int i,j,p,*x=t1,*y=t2; for(i=0;i<m;i++)c[i]=0; for(i=0;i<n;i++)c[x[i]=s[i]]++; for(i=1;i<m;i++)c[i]+=c[i-1]; for(i=n-1;i>=0;i--)sa[--c[x[i]]]=i; for(j=1;j<=n;j<<=1) { p=0; for(i=n-j;i<n;i++)y[p++]=i; for(i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j; for(i=0;i<m;i++)c[i]=0; for(i=0;i<n;i++)c[x[y[i]]]++; for(i=1;i<m;i++)c[i]+=c[i-1]; for(i=n-1;i>=0;i--)sa[--c[x[y[i]]]]=y[i]; swap(x,y); p=1;x[sa[0]]=0; for(i=1;i<n;i++) x[sa[i]]=y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+j]==y[sa[i]+j]?p-1:p++; if(p>=n)break; m=p; } } void getHeight(int s[],int n) { int i,j,k=0; for(i=0;i<=n;i++)Rank[sa[i]]=i; for(i=0;i<n;i++) { if(k)k--; j=sa[Rank[i]-1]; while(s[i+k]==s[j+k])k++; height[Rank[i]]=k; } } int seq[MAXN]; void suffix_array(char str[]){ int len = strlen(str); for(int i = 0; i <= len; i++) seq[i] = str[i]; seq[len] = 0; build_sa(seq, len + 1, 128); getHeight(seq, len); } char str[MAXN]; //================================================================================== long long sum[MAXN], len; struct Tree{ int l, r, val[2]; }tree[MAXN * 4];//0 height 1 sa int cnt = 0; void buildtree(int rt, int l, int r) { tree[rt].l = l, tree[rt].r = r; if(l == r) { tree[rt].val[0] = height[cnt]; tree[rt].val[1] = sa[cnt++]; return ; } int mid = (l + r) / 2; buildtree(rt * 2, l, mid); buildtree(rt * 2 + 1, mid + 1, r); tree[rt].val[0] = min(tree[rt * 2].val[0], tree[rt * 2 + 1].val[0]); tree[rt].val[1] = min(tree[rt * 2].val[1], tree[rt * 2 + 1].val[1]); } int query(int rt, int l, int r, int id) { if(tree[rt].l == l && tree[rt].r == r) { return tree[rt].val[id]; } int mid = (tree[rt].l + tree[rt].r) / 2; if(l > mid) return query(rt * 2 + 1, l, r, id); else if(r <= mid) return query(rt * 2, l, r, id); else { return min(query(rt * 2, l, mid, id), query(rt * 2 + 1, mid + 1, r, id)); } } int bsl(int x, int LEN) { int l = 1, r = x - 1, mid, ret = x; while(l <= r) { mid = (l + r) >> 1; if(query(1, mid + 1, x, 0) >= LEN) { ret = mid, r = mid - 1; } else l = mid + 1; } return ret; } int bsr(int x, int LEN) { int l = x + 1, r = len, mid, ret = x; while(l <= r) { mid = (l + r) >> 1; if(query(1, x + 1, mid, 0) >= LEN) { ret = mid, l = mid + 1; } else r = mid - 1; } return ret; } main() { int n; while(~scanf("%s", str)) { len = strlen(str); suffix_array(str); memset(tree, 0, sizeof tree); cnt = 1; buildtree(1, 1, len); for(int i = 1; i <= len; i++) { sum[i] = sum[i - 1] + len - sa[i] - height[i]; // printf("sa[%d] = %d height[%d] = %d\n", i, sa[i], i, height[i]); } long long l = 0, r = 0, v, k; scanf("%d", &n); while(n--) { scanf("%I64d", &v); v ^= l ^ r; k = v + 1; // printf("k = %I64d\n", k); if(k > sum[len]) { puts("0 0"); l = r = 0; continue; } int t = lower_bound(sum + 1, sum + 1 + len, k) - sum; l = sa[t], r = len - (sum[t] - k + 1); int d = r - l + 1; int L = bsl(t, d), R = bsr(t, d); // printf("L = %d R = %d\n", L, R); int ans = query(1, L, R, 1); // printf("l = %d\n", ans); l = ans + 1, r = l + d - 1; printf("%I64d %I64d\n", l, r); } } }