Display Substring(hdu二分+后缀自动机

二分最后获得的分数，查询复合数量（若数量<k,L=mid+1 else R=mid;```) 查询过程中再次二分自动机每个节点的查询范围为[L,R],利用前缀和计算sum[R]-sum[mid-1],若大于x L=mid+1;else R=mid;以及一些细节处理 需要处理pos[i]等
处理pos[i]的时候才用dfs 注意 你不能直接if(pos[node[i].fa]==0) pos[node[i].fa]=i;如果两个点在主链外并且存在link边 就会出错（在这里我卡了一个下午

#include <bits/stdc++.h>
using namespace std;
#define int long long
typedef long long ll;
const int N=300010;
int tot=1, last=1;char str[N];
struct Node{
    int len, fa;
    int ch[26];
}node[N];
void extend(int c){
    int p=last, np=last=++tot;
    node[np].len=node[p].len+1;
    for(; p && !node[p].ch[c]; p=node[p].fa)
        node[p].ch[c]=np;
    if(!p) node[np].fa=1;
    else {
        int q=node[p].ch[c];
        if(node[q].len==node[p].len+1) 
            node[np].fa=q;
        else {
            int nq=++tot;
            node[nq]=node[q], node[nq].len=node[p].len+1;
            node[q].fa=node[np].fa=nq;
            for( ; p && node[p].ch[c]==q; p=node[p].fa)
                node[p].ch[c]=nq;
        }
    }
}
int T,n,k,pos[N],sum[N],val[N];vector<int> ch[N];
void init(){
    tot=1, last=1;
    for(int i=1;i<=2*n+5;i++){
        ch[i].clear();
        pos[i]=0;node[i].fa=node[i].len=0;
        memset(node[i].ch,0,sizeof node[i].ch);
    }
}
bool check(int x){
    int cnt=0;
    for(int i=2;i<=tot;i++){
        int L=pos[i]-node[i].len+1,R=pos[i]-node[node[i].fa].len;
        while(L<R){
            int mid=(L+R)/2;
            if(sum[pos[i]]-sum[mid-1]<=x) R=mid;
            else L=mid+1;   
        }
        if(sum[pos[i]]-sum[L-1]<=x) cnt+=pos[i]-node[node[i].fa].len-L+1;
    }
    return cnt >= k;
}

void dfs(int u) {
        for (int v : ch[u]) dfs(v);
        if (!pos[u]) pos[u] = pos[ch[u][0]];
}
int32_t main(){
    // freopen("D:\\in.txt","r",stdin);
    // freopen("D:\\out.txt","w",stdout);
    scanf("%lld",&T);
    while(T--){
        scanf("%lld%lld",&n,&k);cin>>str+1;
        init();
        for(int i=0;i<26;i++) scanf("%lld",&val[i]);
        for(int i=1;i<=n;i++) sum[i]=sum[i-1]+val[str[i]-'a'];
        for(int i=1;i<=n;i++){
            extend(str[i]-'a');pos[last]=i;
        }
        pos[1]=1;
        for(int i=2;i<=tot;i++){
            ch[node[i].fa].push_back(i);
        }dfs(1);
        // for(int i=2;i<=tot;i++){
        //     if(pos[node[i].fa]==0) pos[node[i].fa]=i;
        // }
        
        int L=1,R=sum[n];
        while(L<R){
            int mid=(L+R)/2;
            if(check(mid)){R=mid;}else L=mid+1;
        }
        if(check(L))
        printf("%lld\n",L);
        else printf("-1\n");
    }
}