BZOJ 2286: [Sdoi2011]消耗战

听说是虚树裸题？ 好像是我第二次打虚树 不过这次更完整一点
DP很弱智啊。。就不说了

记录一下。。。

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N=250005;
char B[1<<14],*S=B,*T=B;
#define gc (S==T&&(T=(S=B)+fread(B,1,1<<14,stdin),S==T)?-1:*S++)
inline int read(){
    int x=0,f=1; char ch=gc;
    while(ch<'0' || ch>'9'){if(ch=='-')f=-1; ch=gc;}
    while(ch>='0' && ch<='9'){x=(x<<1)+(x<<3)+ch-'0'; ch=gc;}
    return x*f;
}
int dfn[N],dep[N],fa[N][18],id,p[N]; LL f[N],d[N];
bool bo[N];
int Cmp(int x,int y){return dfn[x]<dfn[y];}
struct tree{
    int nex[N<<1],fir[N],go[N<<1],v[N<<1],len;
    void ins(int x,int y,int c){nex[++len]=fir[x],fir[x]=len,go[len]=y,v[len]=c;}
    void ins(int x,int y){nex[++len]=fir[x],fir[x]=len,go[len]=y;}
    void dfs1(int x){
        dfn[x]=++id;
        for(int i=1;i<18;++i)
            if(fa[x][i-1])fa[x][i]=fa[fa[x][i-1]][i-1];
            else break;
        for(int k=fir[x],y;k;k=nex[k]) if( (y=go[k])!=fa[x][0])
            fa[y][0]=x,d[y]=min(d[x],(LL)v[k]),dep[y]=dep[x]+1,dfs1(y);
    }
    void dfs2(int x){
        f[x]=d[x]; LL s=0;
        for(int k=fir[x];k;k=nex[k]) dfs2(go[k]),s+=f[go[k]];
        if(s && !bo[x])f[x]=min(f[x],s);
        fir[x]=0;
    }
}g1,g2;
int lca(int x,int y){
    if(dep[x]<dep[y])x^=y^=x^=y;
    for(int i=17;~i;--i)if(dep[fa[x][i]]>=dep[y])x=fa[x][i];
    if(x==y)return x;
    for(int i=17;~i;--i)if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int st[N],u;
void work(){
    int tp=0;
    for(int i=1;i<=u;++i){
        if(!tp){st[++tp]=p[i]; continue;}
        int q=lca(st[tp],p[i]);
        while(dfn[q]<dfn[st[tp]]){
            if(dfn[q]>=dfn[st[tp-1]]){
                g2.ins(q,st[tp]);
                if(q!=st[--tp])st[++tp]=q;
                break;
            }
            g2.ins(st[tp-1],st[tp]),--tp;
        }
        st[++tp]=p[i];
    }
    while(tp>1) g2.ins(st[tp-1],st[tp]),--tp;
    g2.dfs2(st[1]); printf("%lld\n",f[st[1]]);
    for(int i=1;i<=u;++i) bo[p[i]]=0; g2.len=0;
}
int main(){
    int n=read(),i; d[1]=(1ll<<60);
    for(i=1;i<n;++i){
        int x=read(),y=read(),c=read();
        g1.ins(x,y,c),g1.ins(y,x,c);
    }
    g1.dfs1(1); int m=read();
    while(m--){
        u=read();
        for(i=1;i<=u;++i)p[i]=read(),bo[p[i]]=1;
        sort(p+1,p+1+u,Cmp); work();
    }
    return 0;
}