P2495 [SDOI2011]消耗战 虚树入门

一棵树，n个点m个操作。每条边有权值，每个操作给你k个点，问断开若干条边后使k个点与根不相连的最小边权和是多少。

有sigmaK<500000

• 易知一个裸的树dp需要复杂度，m次操作后总复杂度为显然是无法在限定时间内通过的
• 我们注意到一个性质。k的总和才500000，我们能不能从k入手呢
• 每次只把我们需要用到的点提取出来进行dp呢
• 这里就涉及到一个叫虚树的概念
• 虚树就是把一棵联通情况不变，且仅有可能贡献答案的点连起来的一棵树
• 由题目易知，只要维护好根部到当前点的最小边权，就可以得到每次直接砍掉一棵子树的最小花费
• 对于每个非要求点，他的花费为直接割掉整颗子树的花费和分别割去子树里的贡献点的花费的最小值
• 所以，对于这题虚树大小只有要求点以及他们的lca最多有2*k个点

然后我们又遇到一个问题，找到相应点后快速地建一棵树，并保留原有的基本结构

• 我们可以先把原树的欧拉序搞出来
• 然后对相应点根据欧拉序进行排序
• 我们可以用栈模拟一次dfs
  • 易知若栈顶元素的out值小于当前要加入的元素的in值
  • 那么当前元素必然不在栈顶元素的子树内，那么把栈顶元素出栈
  • 否则当前点和栈顶点连边且入栈
• 每次我们只要把虚树维护出来（），进行dp
• 所需时间只要

#include<algorithm>
#include<vector>
#include<iostream>
#include<math.h>
#include<cstring>
#include<string>
#include<stack>
#include<map>
#include<set>
#include<unordered_map>
#include<queue>
#include<assert.h>
#include<iomanip>
#include<bitset>
#include<stdio.h>

#define qcin; ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define pb push_back
#define mp make_pair
#define clr(x) memset(x,0,sizeof x)
#define fmax(x) memset(x,0x3f,sizeof x)
#define finit(x) memset(x,-1,sizeof x)
#define iio(n,m) io(n),io(m)
#define lc(p) (p<<1)
#define rc(p) ((p<<1)|1)

#define dis(l,r) r-l+1
#define gstr(str) scanf("%s",str)
#define glen(str) strlen(str)
using namespace std;

namespace Input{
    const int BUF = 65536;
    char buf[BUF + 1];
    char *head = buf, *tail = buf;
}
inline char inputchar(){
    using namespace Input;
    if(head == tail)
        *(tail = (head = buf) + fread(buf, 1, BUF, stdin)) = 0;
    return *head++;
}
template<class T>
inline void io(T &ret){
    ret = 0;
    char ch = inputchar();
    while((ch < '0' || ch > '9') && ch != '-')
        ch = inputchar();
    bool neg = false;
    if(ch == '-')
        neg = true, ch = inputchar();
    while(ch >= '0' && ch <= '9')
    {
        ret = ret * 10 + ch - '0';
        ch = inputchar();
    }
    if(neg)
        ret = -ret;
}

typedef long long ll;

typedef pair<int,int>pll;
const int maxn = 250010;
const int mod =  1e9+7;
const ll INF = 2147483647;

typedef ll arr[maxn];
typedef char str[maxn];
void file(int x){if(x&&fopen("123.in","r")){freopen("123.in","r",stdin);}}

const long double pi = acos(-1);
const double eps=1e-10;
const double delta=0.993;

arr h,H,in,out,dep,dp,mi;
int fa[25][maxn];
int tot,pos;
struct node{
    int to,nx,cst;
}E[maxn*10];

void add(ll h[],int u,int v,int w=0){
    E[tot].to=v;
    E[tot].nx=h[u];
    E[tot].cst=w;
    h[u]=tot++;
}
void dfs(int u){
    in[u]=++pos;
    for(int i=1;fa[i-1][u];i++){fa[i][u]=fa[i-1][fa[i-1][u]];}
    for(int i=h[u];~i;i=E[i].nx){
        int v=E[i].to,w=E[i].cst;
        if(!in[v]){
            fa[0][v]=u;dep[v]=dep[u]+1;
            mi[v]=min(1LL*w,mi[u]);
            dfs(v);
        }
    }
    out[u]=++pos;
    return;
}
int lca(int u,int v){
    if(dep[u]<dep[v])swap(u,v);
    int d=dep[u]-dep[v];
    for(int i=0;d;d>>=1,i++)if(1&d){
        u=fa[i][u];
        if(u==v)return u;
    }
    for(int i=20;~i;i--)if(fa[i][v]!=fa[i][u]){
        v=fa[i][v];
        u=fa[i][u];
    }
    return fa[0][u];
}
arr key,vis;
int n,u,v,m,k,w;
stack<int>S;
ll res=0;
void dfs_1(int u){
    if(dp[u]!=dp[n+1]){return;}
    dp[u]=0;
    for(int i=H[u];~i;i=E[i].nx){
        int v=E[i].to,w=E[i].cst;
        dfs_1(v);
        dp[u]+=min(dp[v],mi[v]);
    }
    dp[u]=min(dp[u],mi[u]);
}

int cmp(int x,int y){
    return in[x]<in[y];
}
int main(){
    file(1);
    io(n);
    finit(h);finit(H);fmax(mi);fmax(dp);
    for(int i=1;i<n;i++){
        io(u),io(v),io(w);
        add(h,u,v,w);
        add(h,v,u,w);
    }
    dfs(1);
    io(m);
    while(m--){
        while(S.size())S.pop();
        io(k);
        for(int i=1;i<=k;i++){
            io(key[i]);
            vis[key[i]]=1;
            dp[key[i]]=mi[key[i]];
        }
        key[++k]=1;vis[1]=1;
        sort(key+1,key+1+k,cmp);
        for(int i=2;i<=k;i++){
            int anc=lca(key[i],key[i-1]);
            if(!vis[anc]){
                vis[anc]=1;
                key[++k]=anc;
            }
        }
        for(int i=1;i<=k;i++)H[key[i]]=-1;
        tot=0;
        sort(key+1,key+1+k,cmp);
        for(int i=1;i<=k;i++){
            while(S.size()&&out[S.top()]<in[key[i]])S.pop();
            if(S.size())add(H,S.top(),key[i]);
            S.push(key[i]);
        }
        vis[1]=0;
        dfs_1(1);
        printf("%lld\n",dp[1]);
        for(int i=1;i<=k;i++)vis[key[i]]=0,dp[key[i]]=dp[n+1];
    }
}