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;}