【BZOJ】2286: [Sdoi2011]消耗战 虚树+DP

【题意】给定n个点的带边权树，每次询问给定ki个特殊点，求隔离点1和特殊点的最小代价。n<=250000，Σki<=500000。

【算法】虚树+DP

【题解】考虑普通树上的dp，设f[x]表示隔离1和子树x内特殊点的最小代价，val[x]表示x到1路径上的最小代价（预处理）。

点x特殊，f[x]=val[x]

否则，f[x]=min{val[x],Σf[y]}，y=son[x]

在询问总数有限制的前提下，可以建虚树进行如上DP。

复杂度O(Σki log n)。

注意：

1.清空时垃圾回收，保证复杂度。

2.询问点数组开两倍，因为要加入两两LCA。

#include<cstdio>
#include<cctype>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
int read(){
    int s=,t=;char c;
    while(!isdigit(c=getchar()))if(c=='-')t=-;
    do{s=s*+c-'';}while(isdigit(c=getchar()));
    return s*t;
}
const int maxn=;
struct edge{int v,w,from;}e[maxn*];
int in[maxn],ou[maxn],deep[maxn],f[maxn][],st[maxn],n,N,tot,first[maxn],a[maxn*];//a[]
bool v[maxn];
ll val[maxn];
 
namespace cyc{
    struct edge{int v,w,from;}e[maxn*];
    int first[maxn],dfsnum=,tot;
    void insert(int u,int v,int w){tot++;e[tot].v=v;e[tot].w=w;e[tot].from=first[u];first[u]=tot;}
    void dfs(int x,int fa){
        in[x]=++dfsnum;
        for(int j=;(<<j)<=deep[x];j++)f[x][j]=f[f[x][j-]][j-];
        for(int i=first[x];i;i=e[i].from)if(e[i].v!=fa){
            deep[e[i].v]=deep[x]+;
            val[e[i].v]=min(val[x],1ll*e[i].w);
            f[e[i].v][]=x;
            dfs(e[i].v,x);
        }
        ou[x]=dfsnum;
    }
    int lca(int x,int y){
        if(deep[x]<deep[y])swap(x,y);
        int d=deep[x]-deep[y];
        for(int i=;i<=;i++)if((<<i)&d)x=f[x][i];
        if(x==y)return x;
        for(int i=;i>=;i--)if((<<i)<=deep[x]&&f[x][i]!=f[y][i]){
            x=f[x][i];y=f[y][i];
        }
        return f[x][];
    }
    void build(){
        n=read();
        for(int i=;i<n;i++){
            int u=read(),v=read(),w=read();
            insert(u,v,w);insert(v,u,w);
        }
        val[]=1ll<<;//
        dfs(,-);
    }
}
void insert(int u,int v){tot++;e[tot].v=v;e[tot].from=first[u];first[u]=tot;}
bool cmp(int a,int b){return in[a]<in[b];}
ll dp(int x){
    if(v[x])return val[x];
    ll sum=;
    for(int i=first[x];i;i=e[i].from)sum+=dp(e[i].v);
    return min(val[x],sum);
}
bool check(int x,int y){return in[y]<=in[x]&&ou[x]<=ou[y];}
void build(){
    int last=read();N=last;
    for(int i=;i<=N;i++)a[i]=read(),v[a[i]]=;//
    sort(a+,a+N+,cmp);
    for(int i=;i<last;i++)a[++N]=cyc::lca(a[i],a[i+]);
    sort(a+,a+N+,cmp);
    N=unique(a+,a+N+)-a-;
    for(int i=;i<=N;i++)first[a[i]]=;tot=;//
    int top=;
    for(int i=;i<=N;i++){
        while(top&&!check(a[i],st[top]))top--;
        if(top)insert(st[top],a[i]);
        st[++top]=a[i];
    }
    printf("%lld\n",dp(a[]));
    for(int i=;i<=N;i++)v[a[i]]=;//
}
int main(){
    cyc::build();
    int m=read();
    while(m--)build();
    return ;
}