BZOJ 3998 TJOI2015 弦论 后缀自动机+DAG上的dp

时间:2023-03-08 22:54:27
BZOJ 3998 TJOI2015 弦论 后缀自动机+DAG上的dp

题目链接:http://www.lydsy.com/JudgeOnline/problem.php?id=3998

题意概述:对于一个给定长度为N的字符串,求它的第K小子串是什么,T为0则表示不同位置的相同子串算作一个。T=1则表示不同位置的相同子串算作多个。N<=500000,K<=10^9.

应该是有三种做法的(当然后缀树我还没有看),于是就把这个东西当成后缀自动机的板子了(因为它很裸啊!!!)。

可以注意到当T=0的时候每走动一步的贡献是1,而T=1的时候每走动一步之后的贡献都是走到的这个状态的right集合大小。同时从同一个状态走出去得到的字符串又有很多种,不难想到可以搞个dp表示从这个点出发(包括这个点自己)可以找到的不同子串数(根据T而定)。

有个技巧是根据MAX把所有的状态排序之后就可以欢快地topo图上转移dp啦!(SAM的转移图是一张DAG)

所以还是好好理解一下SAM上的状态和转移吧,,,。。。。

 #include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<queue>
#include<set>
#include<map>
#include<vector>
#include<cctype>
using namespace std;
const int MAXN=; int T,K,n; char S[MAXN]; struct SAM{
static const int maxn=;
static const int sigma_sz=;
int pa[maxn<<],to[maxn<<][sigma_sz],mx[maxn<<],val[maxn<<]; long long cnt[maxn<<];
int sz,last,c[maxn],a[maxn<<];
SAM(){ last=sz=; memset(to[],,sizeof(to[])); }
int newnode(){
memset(to[++sz],,sizeof(to[sz]));
pa[sz]=mx[sz]=val[sz]=cnt[sz]=;
return sz;
}
void extend(int w){
int p=last,np=newnode(); last=np;
pa[np]=mx[np]=cnt[np]=;
mx[np]=mx[p]+,val[np]=;
while(p&&!to[p][w]) to[p][w]=np,p=pa[p];
if(!p) pa[np]=;
else{
int q=to[p][w];
if(mx[p]+==mx[q]) pa[np]=q;
else{
int nq=newnode(); mx[nq]=mx[p]+;
memcpy(to[nq],to[q],sizeof(to[nq]));
pa[nq]=pa[q];
pa[np]=pa[q]=nq;
while(p&&to[p][w]==q) to[p][w]=nq,p=pa[p];
}
}
}
void ready(){
for(int i=;i<=sz;i++) c[mx[i]]++;
for(int i=;i<=n;i++) c[i]+=c[i-];
for(int i=sz;i>;i--) a[c[mx[i]]--]=i;
for(int i=sz;i>;i--){
if(T) val[pa[a[i]]]+=val[a[i]];
else val[a[i]]=;
}
val[]=;
for(int i=sz;i>;i--){
cnt[a[i]]=val[a[i]];
for(int j=;j<;j++) cnt[a[i]]+=cnt[to[a[i]][j]];
}
}
void dfs(int i){
if(val[i]>=K) return;
K-=val[i];
for(int j=;j<;j++) if(to[i][j]){
if(cnt[to[i][j]]>=K){
putchar(j+'a'); dfs(to[i][j]);
return;
}
K-=cnt[to[i][j]];
}
}
}sam; void data_in()
{
gets(S);
scanf("%d%d",&T,&K);
}
void work()
{
n=strlen(S);
for(int i=;i<n;i++) sam.extend(S[i]-'a');
sam.ready();
if(sam.cnt[]<K) printf("%d\n",-);
else sam.dfs();
}
int main()
{
data_in();
work();
return ;
}