虚树模板

Code

vec记录所有在虚树中的点,方便删除
add1只需要多开一个lst就可以实现O(k)删除
为了方便默认1号点为根

void build() {
    sort(a+1,a+tot+1,cmp);
    st[top=1]=1;vec.push_back(1);
    fo(i,1,tot) {
        int lca=get_lca(a[i],st[top]);
        if (lca==st[top]) {
            if (st[top]!=a[i]) st[++top]=a[i],vec.push_back(a[i]);
            continue;
        }
        while (dep[st[top-1]]>=dep[lca]) 
            add1(st[top-1],st[top],dist(st[top-1],st[top])),top--;
        if (dep[st[top]]>dep[lca]) add1(lca,st[top],dist(lca,st[top])),top--;
        if (lca!=st[top]) st[++top]=lca,vec.push_back(lca);
        st[++top]=a[i];vec.push_back(a[i]);
    }
    while (top>1) add1(st[top-1],st[top],dist(st[top-1],st[top])),top--;
}

猜你喜欢

转载自blog.csdn.net/alan_cty/article/details/80085904