虚树模版写法

\(\mathbf{} \begin{Bmatrix} \frac{{\Large 虚树模版} }{{\color{Blue}\Large Template} }\mathbf{} {No.2} \end{Bmatrix}\times{}\) NeeDna

#include<bits/stdc++.h>
#define inf LONG_LONG_MAX
#define int long long
using namespace std;
const int N=5e5+10;
struct edge{int v,w,nxt;}e[N<<1],e1[N<<1];
int n,head[N],cnt,tot,m,k,a[N],dp[N],f[N][20],ti,dfn[N],dep[N],st[N],top,mn[N],head1[N];
void add(int u,int v,int w){e[++cnt]={v,w,head[u]};head[u]=cnt;}
void add(int u,int v){if(u==v) return;e1[++tot]={v,0,head1[u]};head1[u]=tot;}
bool cmp(int x,int y){return dfn[x]<dfn[y];}
void gd(int u,int fa){
	dfn[u]=++ti;f[u][0]=fa;dep[u]=dep[fa]+1;
	for(int i=1;(1<<i)<=dep[u];i++) f[u][i]=f[f[u][i-1]][i-1];
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].v,w=e[i].w;if(v==fa) continue;
		mn[v]=min(mn[u],w);gd(v,u);
	}
}
int LCA(int x,int y){
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=19;i>=0;i--){if(dep[f[x][i]]>=dep[y]) x=f[x][i];}
	if(x==y) return x;
	for(int i=19;i>=0;i--){if(f[x][i]!=f[y][i]){x=f[x][i];y=f[y][i];}}
	return f[x][0];
}
void build(int k){
	tot=top=0;
	st[++top]=1;head1[1]=0;
	for(int i=1;i<=k;i++){
		int u=a[i];int l=LCA(u,st[top]);
		while(1){
			if(dep[l]<dep[st[top-1]]){
				add(st[top-1],st[top]);top--;
			}else{
				if(dep[l]<dep[st[top]])add(l,st[top--]);
				break;
			}	
		}
	    if(st[top]!=l) st[++top]=l;
	    if(st[top]!=u) st[++top]=u;
	}
	while(--top) add(st[top],st[top+1]);
}
void dfs(int u){
	int tmp=0;
	for(int i=head1[u];i;i=e1[i].nxt){
		int v=e1[i].v;
		dfs(v);tmp+=dp[v];	
	}
	dp[u]=mn[u];
	if(tmp) dp[u]=min(dp[u],tmp);
	head1[u]=0;
}
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0);cout.tie(0);
	cin>>n;
	for(int i=1,u,v,w;i<n;i++){
		cin>>u>>v>>w;
		add(u,v,w);add(v,u,w);
	}
	mn[1]=inf;gd(1,0);
	cin>>m;
	for(int i=1;i<=m;i++){
		cin>>k;int cnt=1;
		for(int j=1;j<=k;j++){cin>>a[j];}sort(a+1,a+k+1,cmp);
		for(int j=2;j<=k;j++){if(LCA(a[cnt],a[j])!=a[cnt])a[++cnt]=a[j];}	 
		build(cnt);dfs(1);
		cout<<dp[1]<<'\n';
	}
	return 0;
}

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef long double ld;

const int N=3e5+10,inf=2147483647;

vector<int> G[N],nG[N];

int n,q;

int dfn[N],tim=0,dep[N],siz[N],f[N][21];

void dfs(int u,int fa){
    dfn[u]=++tim,dep[u]=dep[fa]+1,siz[u]=1,f[u][0]=fa;
    for(int i=1;i<=20;i++)
        f[u][i]=f[f[u][i-1]][i-1];
    for(auto v:G[u]){
        if(v==fa) continue;
        dfs(v,u);
        siz[u]+=siz[v];
    }
}

int lca(int x,int y){
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=20;~i;i--)
        if(dep[f[x][i]]>=dep[y])
            x=f[x][i];
    if(x==y) return x;
    for(int i=20;~i;i--)
        if(f[x][i]!=f[y][i])
            x=f[x][i],y=f[y][i];
    return f[x][0];
}

int k,que[N],tmp[N],ans[N];

bool vis[N];

int st[N],top=0;

void build(int k){
    st[top=1]=1;
    for(int i=1;i<=k;i++){
        if(que[i]==1) continue;
        int fa=lca(que[i],st[top]);
        while(1){
            if(dep[fa]<dep[st[top-1]]){
                nG[st[top-1]].push_back(st[top]);
                top--;
            }
            else{
                if(dep[fa]<dep[st[top]])
                    nG[fa].push_back(st[top--]);
                break;
            }
        }
        if(fa!=st[top]) st[++top]=fa;
        if(que[i]!=st[top]) st[++top]=que[i];
    }
    while(--top) nG[st[top]].push_back(st[top+1]);
}

int dp[N],g[N];

void dfs1(int u){
    dp[u]=inf;
    for(auto v:nG[u]){
        dfs1(v);
        int dis=dep[v]-dep[u];
        if(dp[v]+dis<dp[u]) dp[u]=dp[v]+dis,g[u]=g[v];
        else if(dp[v]+dis==dp[u]) g[u]=min(g[u],g[v]);
    }
    if(vis[u]) dp[u]=0,g[u]=u;
}

void cal(int u,int v){
    int x=v,y=v;
    for(int i=20;~i;i--)
        if(dep[f[x][i]]>dep[u])
            x=f[x][i];
    for(int i=20;~i;i--){
        if(dep[f[y][i]]<=dep[u]) continue;
        int len1=dep[v]-dep[f[y][i]]+dp[v],len2=dep[f[y][i]]-dep[u]+dp[u];
        if(len1<len2||(len1==len2&&g[u]>g[v])) y=f[y][i];
    }
    ans[g[v]]+=siz[y]-siz[v],ans[g[u]]-=siz[y];
}

void dfs2(int u){
    for(auto v:nG[u]){
        int dis=dep[v]-dep[u];
        if(dp[u]+dis<dp[v]) dp[v]=dp[u]+dis,g[v]=g[u];
        else if(dp[u]+dis==dp[v]) g[v]=min(g[v],g[u]);
        cal(u,v);
        dfs2(v);
    }
    ans[g[u]]+=siz[u];
    vis[u]=false;
    nG[u].clear();
}

int main(){
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    cin>>n;
    for(int i=1,u,v;i<n;i++){
        cin>>u>>v;
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dfs(1,0);
    cin>>q;
    while(q--){
        cin>>k;
        for(int i=1;i<=k;i++){
            cin>>que[i];
            vis[que[i]]=true,ans[que[i]]=0,tmp[i]=que[i];
        }
        sort(que+1,que+1+k,[](int x,int y){
            return dfn[x]<dfn[y];
        });
        build(k);
        dfs1(1);
        dfs2(1);
        for(int i=1;i<=k;i++)
            cout<<ans[tmp[i]]<<' ';
        cout<<"\n";
    }
    return 0;
}//from wyb