首页 > 代码库 > SPOJ7258
SPOJ7258
传送门
这题可以参考平衡树求第k大的过程,需要预处理一下从当前节点往下走能走出多少个子串。
原本准备存个图用反向的topsort,发现极为麻烦,看了别人的代码后发现,他们按step大小用了基排,省了很多麻烦。
仔细看了一下构造过程,貌似确实有这样的性质,后创建且step大的节点在fail树上处于儿子的地位,在原图上处于被转移点的地位。
如果性质成立,那么这里也就说得通了。
#include<iostream> #include<cstdio> #include<cstdlib> #include<string> #include<cstring> #include<cmath> #include<ctime> #include<algorithm> #include<iomanip> #include<set> #include<map> #include<queue> using namespace std; #define mem1(i,j) memset(i,j,sizeof(i)) #define mem2(i,j) memcpy(i,j,sizeof(i)) #define LL long long #define up(i,j,n) for(int i=(j);i<=(n);i++) #define FILE "dealing" #define poi vec #define eps 1e-10 #define db double const int maxn=500010,inf=1000000000,mod=1000000007; int read(){ int x=0,f=1,ch=getchar(); while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();} while(ch<=‘9‘&&ch>=‘0‘){x=(x<<1)+(x<<3)+ch-‘0‘,ch=getchar();} return f*x; } bool cmax(int& a,int b){return a<b?a=b,true:false;} bool cmin(int& a,int b){return a>b?a=b,true:false;} char s[maxn];int f[maxn]; namespace SAM{ int cnt=1,now=1; int pre[maxn],c[maxn][26],len[maxn],ct[maxn],sa[maxn]; void extend(int x){ int np,p,q,nq; p=now;np=++cnt;len[np]=len[p]+1;now=np; while(p&&!c[p][x])c[p][x]=np,p=pre[p]; if(!p)pre[np]=1; else { q=c[p][x]; if(len[q]==len[p]+1)pre[np]=q; else { len[nq=++cnt]=len[p]+1; mem2(c[nq],c[q]); pre[nq]=pre[q]; pre[q]=pre[np]=nq; while(p&&c[p][x]==q)c[p][x]=nq,p=pre[p]; } } } void getsort(){ up(i,1,cnt)ct[len[i]]++; up(i,1,cnt)ct[i]+=ct[i-1]; for(int i=cnt;i>=1;i--)sa[ct[len[i]]--]=i; for(int i=cnt;i>=1;i--){ f[sa[i]]=1; up(j,0,25)f[sa[i]]+=f[c[sa[i]][j]]; } } void build(char* s){ cnt=1,now=1;int n=strlen(s+1); up(i,1,n)extend(s[i]-‘a‘); } }; int main(){ freopen(FILE".in","r",stdin); freopen(FILE".out","w",stdout); scanf("%s",s+1); SAM::build(s); SAM::getsort(); int Q=read(); while(Q--){ int k=read(); int now=1,len=0; while(k)up(i,0,25){ if(SAM::c[now][i]){ if(f[SAM::c[now][i]]>=k){ putchar(‘a‘+i); now=SAM::c[now][i]; k--;//不可省 break; } else k-=f[SAM::c[now][i]]; } } printf("\n"); } return 0; }
SPOJ7258
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。