【bzoj4310】跳蚤【hdu5030】Rabbit's String

Source

bzoj4310
hdu5030

Hint

请先思考后再展开

二分子串(先后缀再前缀)

Solution

请先思考后再展开

这题是我sb了
就是打个标记,表示这里后面第一个分段结尾最晚在哪里
用sa求lcp,时间复杂度为 $O(nlogn)$

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
//Zory-2019
#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<map>
#include<set>
#include<queue>
#include<deque>
#include<stack>
#include<bitset>
#include<vector>
#include<algorithm>
#include<iostream>
#include<deque>
using namespace std;
namespace mine
{
typedef long long ll;
#define double long double
const int INF=0x3f3f3f3f;
int qread()
{
int ans=0;char c=getchar();int f=1;
while(c<'0' or c>'9') {if(c=='-') f=-1;c=getchar();}
while('0'<=c and c<='9') ans=ans*10+c-'0',c=getchar();
return ans*f;
}
void write(int num)
{
if(num<0) {num=-num;putchar('-');}
if(num>9) write(num/10);
putchar('0'+num%10);
}
void writeln(int num){write(num);puts("");}
#define FR first
#define SE second
#define MP make_pair
#define pr pair<int,int>
#define PB push_back
inline void chmax(int &x,int y) {x=x>y?x:y;}
inline void chmin(int &x,int y) {x=x<y?x:y;}
const int MAX_N=1e5+10;
int n;char str[MAX_N];
struct SA
{
int sa[MAX_N],rk[MAX_N],tmp[MAX_N],ct[MAX_N],wr[MAX_N*2];
void getsa()
{
memset(ct,0,sizeof ct);
for(int i=1;i<=n;i++) ct[rk[i]=str[i]]++;
for(int i=1;i<MAX_N;i++) ct[i]+=ct[i-1];
for(int i=n;i>=1;i--) sa[ct[rk[i]]--]=i;
int ln=1;
while(ln<n)
{
int tot=0;for(int i=1;i<=n;i++) if(sa[i]+ln>n) tmp[++tot]=sa[i];
for(int i=1;i<=n;i++) if(sa[i]-ln>=1) tmp[++tot]=sa[i]-ln;
memset(ct,0,sizeof ct);memset(wr,0,sizeof wr);
for(int i=1;i<=n;i++) ct[ wr[tmp[i]]=rk[tmp[i]] ]++;
for(int i=1;i<MAX_N;i++) ct[i]+=ct[i-1];
for(int i=n;i>=1;i--) sa[ct[ wr[tmp[i]] ]--]=tmp[i];
int cnt=1;rk[sa[1]]=1;
for(int i=2;i<=n;i++)
{
if(wr[sa[i]]!=wr[sa[i-1]] or wr[sa[i]+ln]!=wr[sa[i-1]+ln]) cnt++;
rk[sa[i]]=cnt;
}
ln*=2;
}
}
int hei[MAX_N];
void gethei()
{
int lst=0;
for(int i=1;i<=n;i++)
{
if(rk[i]==1) continue;
int j=sa[rk[i]-1];if(lst) lst--;
while(max(i,j)+lst<=n and str[i+lst]==str[j+lst]) lst++;
hei[rk[i]]=lst;
}
}
}sa;
struct STB
{
int bin[30],lg[MAX_N],mm[MAX_N][20];
void pre()
{
bin[0]=1;for(int i=1;i<30;i++) bin[i]=bin[i-1]<<1;
lg[1]=0;for(int i=2;i<MAX_N;i++) lg[i]=lg[i>>1]+1;
for(int i=1;i<20;i++) for(int j=2;j<=n-bin[i]+1;j++) mm[j][i]=min(mm[j][i-1],mm[j+bin[i-1]][i-1]);
}
int ask(int x,int y)
{
int t=lg[y-x+1];
return min(mm[x][t],mm[y-bin[t]+1][t]);
}
}stb;
int K;int mk[MAX_N];//出现一个结尾的下界
bool check(int pos,int len)
{
memset(mk,63,sizeof mk);mk[pos]=pos+len-1;mk[n]=n;
for(int i=1;i<=n;i++) if(sa.rk[i]>sa.rk[pos])
{
int lcp=stb.ask(sa.rk[pos]+1,sa.rk[i]);chmin(lcp,len);
if(lcp==0) return 0;
mk[i]=i+lcp-1;
}
int mi=n+1,cnt=0;
for(int i=1;i<=n;i++)
{
chmin(mi,mk[i]);
if(mi==i) cnt++,mi=n+1;
}
return cnt<=K;
}
void main()
{
while(1)
{
scanf("%d",&K);if(K==0) break;
scanf("%s",str+1);n=strlen(str+1);
sa.getsa();sa.gethei();
for(int i=2;i<=n;i++) stb.mm[i][0]=sa.hei[i];
stb.pre();
int l=1,r=n,pos;
while(l<=r)
{
int mid=(l+r)/2;
if(check(sa.sa[mid],n-sa.sa[mid]+1)) pos=sa.sa[mid],r=mid-1;
else l=mid+1;
}
l=1,r=n-pos+1;int len;
while(l<=r)
{
int mid=(l+r)/2;
if(check(pos,mid)) len=mid,r=mid-1;
else l=mid+1;
}
for(int i=1;i<=len;i++) putchar(str[pos+i-1]);
puts("");
}
}
};
int main()
{
srand(time(0));
mine::main();
}

本文基于 知识共享署名-相同方式共享 4.0 国际许可协议发布
本文地址:http://zory.ink/posts/6af4.html
转载请注明出处,谢谢!