bzoj 2286 [Sdoi2011]消耗战 虚树

倍增，然后直接上虚树。

#include <bits/stdc++.h>using namespace std;#define N 260000#define ll long long#define inf 1e9int n,Q,m,cnt,top,tot;int head[N],nex[N<<1],to[N<<1],val[N<<1];int pos[N],a[N],st[N],deep[N],bel[N],tp[N];ll f[N];int fa[N][21],mn[N][21];vector<int>vec[N],v[N];void add(int x,int y,int v){    tot++;    nex[tot]=head[x];head[x]=tot;    to[tot]=y;val[tot]=v;}void dfs(int x,int y){    pos[x]=++cnt;    deep[x]=deep[y]+1;    for(int i=1;i<=20;i++)    {        fa[x][i]=fa[fa[x][i-1]][i-1];        mn[x][i]=min(mn[x][i-1],mn[fa[x][i-1]][i-1]);    }    for(int i=head[x];i;i=nex[i])        if(to[i]!=y)        {            fa[to[i]][0]=x;            mn[to[i]][0]=val[i];            dfs(to[i],x);        }}int cmp(int x,int y){return pos[x]<pos[y];}int lca(int x,int y){    if(deep[x]<deep[y])swap(x,y);    for(int i=20;i>=0;i--)        if(deep[fa[x][i]]>=deep[y])            x=fa[x][i];    if(x==y)return x;    for(int i=20;i>=0;i--)        if(fa[x][i]!=fa[y][i])            x=fa[x][i],y=fa[y][i];    return fa[x][0];}int cal(int x,int y){    int ret=inf;    for(int i=20;i>=0;i--)        if(deep[fa[x][i]]>=deep[y])            ret=min(ret,mn[x][i]),x=fa[x][i];    return ret;}void dfs(int x){    f[x]=0;    for(int i=0,t;i<vec[x].size();i++)    {        dfs(t=vec[x][i]);        f[x]+=min(f[t],(ll)v[x][i]);    }    if(tp[x])f[x]=inf;}void cl(int x){    vec[x].clear();    v[x].clear();tp[x]=0;}int main(){    //freopen("tt.in","r",stdin);    scanf("%d",&n);    for(int i=1,x,y,v;i<n;i++)    {        scanf("%d%d%d",&x,&y,&v);        add(x,y,v);add(y,x,v);    }    dfs(1,0);    scanf("%d",&Q);    for(;Q--;)    {        scanf("%d",&m);        for(int i=1;i<=m;i++)            scanf("%d",&a[i]);        sort(a+1,a+1+m,cmp);        st[top=1]=1;bel[1]=(cnt=1);cl(1);        for(int i=1;i<=m;i++)        {            int t;            while((t=lca(a[i],st[top]))!=st[top])            {                int t1=st[top-1],t2=st[top];                top--;                if(deep[t]>deep[st[top]])                {                    st[++top]=t,bel[t]=++cnt;                    t1=t;cl(cnt);                }                vec[bel[t1]].push_back(bel[t2]);                v[bel[t1]].push_back(cal(t2,t1));            }            if(a[i]!=st[top])            {                st[++top]=a[i];                bel[a[i]]=++cnt;cl(cnt);            }        }        for(int i=top;i>=2;i--)        {            int t1=bel[st[i-1]],t2=bel[st[i]];            vec[t1].push_back(t2);            v[t1].push_back(cal(st[i],st[i-1]));        }        for(int i=1;i<=m;i++)tp[bel[a[i]]]=1;        dfs(1);        printf("%lld\n",f[1]);    }    return 0;}