题目:
相同子串算多个的话,先求好 right ,然后求一个 sm 表示走到这个点之后有几种走法,即把 DAG 上自己能走到的点的 right 都收集起来,可用拓扑序解决。
相同子串算一个的话,给 DAG 上每个节点都赋上一个1,表示走到那个节点的话算一个子串;然后把 DAG 上自己能走到的点的1都收集起来。
然后可按字典序 dfs 这个 DAG ,如果 k 在这个分支里的话就走进去,否则 k -= sm[ ] ,然后看别的分支。
不用管 len[ ] ,因为 len[ ] 表示自己这个点管辖的许多子串,但如果自己是 dfs 地走过来,就只对应了其中一个子串,所以只用管 right 就行了。
之所以相同子串算一个的话要给复制出来的那些 nq 也赋一个1,是因为 nq 只是分了一些 q 管辖的子串;如果自己 dfs 着本应走到 q 点,但因为 nq 分管了一些而走到了 nq 点的话,自己也算走到了一个子串结尾的位置(1个而不是len个),所以应该计数1。
注意用 a[ i ] 而不是 i 。
#include#include #include #define ll long longusing namespace std;const int N=5e5+5,M=1e6+5,K=30;int lst=1,cnt=1,go[M][K],fa[M],l[M],rt[M];ll sm[M];int tx[N],a[M];char ch[N];void add(int w){ int p=lst,np=++cnt;lst=np;l[np]=l[p]+1;rt[np]=1; for(;p&&!go[p][w];p=fa[p])go[p][w]=np; if(!p)fa[np]=1; else { int q=go[p][w]; if(l[q]==l[p]+1)fa[np]=q; else { int nq=++cnt;l[nq]=l[p]+1; fa[nq]=fa[q];fa[q]=nq;fa[np]=nq; memcpy(go[nq],go[q],sizeof go[q]); for(;go[p][w]==q;p=fa[p])go[p][w]=nq; } }}void Rsort(int n,bool T){ for(int i=1;i<=cnt;i++)tx[l[i]]++; for(int i=n-1;i>=0;i--)tx[i]+=tx[i+1]; for(int i=cnt;i;i--)a[tx[l[i]]--]=i; for(int i=1;i<=cnt;i++) { int d=a[i]; if(!T)rt[d]=1; else rt[fa[d]]+=rt[d]; } // if(T)for(int i=1;i<=cnt;i++)rt[fa[a[i]]]+=rt[a[i]];//if()!!!!///a[]!!!!!! for(int i=1;i<=cnt;i++) { int d=a[i];/// sm[d]=rt[d]; for(int j=1;j<=26;j++) if(go[d][j]) { sm[d]+=sm[go[d][j]]; } } rt[1]=0;//}int main(){ scanf("%s",ch);int n=strlen(ch); for(int i=0;i =k)break; k-=rt[cr]; int ycr=cr; for(int i=1,d;i<=26;i++) if(sm[d=go[cr][i]]>=k) {putchar(i+'a'-1);cr=d;break;} else k-=sm[d]; if(cr==ycr){printf("-1");break;} } return puts("");}