标签:style class blog ext get string
经典题目:给一个字符串,求字典序第k小的子串是什么。
涉及子串问题,上自动机。
首先我们可以用记忆化搜索的方法,求出到达某一个状态后,能产生多少个新状态。
首先,到达这个状态就不走了,这肯定是一种状态,然后分别考虑后面的26个指针就好了。
不过如果不记忆化肯定是要T的,而且如果用dp好像会有一点问题,因为状态转移不是严格的满足小号点到大号点(nq点啦)。
然后就是赤果果的dfs就可以啦。
对了还有一个有趣的事情,一开始我输出字符的地方T了,后来改变字符串的输出方式,然后就A了。spoj真是奇葩呀。能买个好点服务器吗?
召唤代码君:
#include <iostream> #include <cstdio> #include <cstring> #define maxn 222222 using namespace std; int next[maxn][26],pre[maxn],step[maxn],f[maxn]; char s[maxn]; int N,last,n,k; int p,q,np,nq; void insert(int x,int m) { p=last,np=++N,step[np]=m; while (p!=-1 && next[p][x]==0) next[p][x]=np,p=pre[p]; last=np; if (p==-1) return; q=next[p][x]; if (step[q]==step[p]+1) { pre[np]=q; return; } nq=++N,step[nq]=step[p]+1,pre[nq]=pre[q]; for (int i=0; i<26; i++) next[nq][i]=next[q][i]; pre[np]=pre[q]=nq; for (;p!=-1 && next[p][x]==q; p=pre[p]) next[p][x]=nq; } int get(int x) { if (f[x]!=0) return f[x]; f[x]=1; for (int i=0; i<26; i++) if (next[x][i]) f[x]+=get(next[x][i]); return f[x]; } void output(int pos,int num,int L) { num--; if (num==0) { s[L]=‘\0‘; printf("%s\n",s+1); return; } for (int i=0; i<26; i++) { if (next[pos][i]==0) continue; if (f[next[pos][i]]<num) num-=f[next[pos][i]]; else { s[L]=‘a‘+i; output(next[pos][i],num,L+1); return; } } } int main() { pre[0]=-1; scanf("%s",s); for (int i=0; s[i]; i++) insert(s[i]-‘a‘,i+1); f[0]=get(0); scanf("%d",&n); while (n--) { scanf("%d",&k); output(0,k+1,1); } return 0; }
标签:style class blog ext get string
原文地址:http://www.cnblogs.com/Canon-CSU/p/3796405.html