Bzoj 2286 & Luogu P2495 消耗战(LCA+虚树+欧拉序)

这里是简介

题面

洛谷

Bzoj

题解

很容易想到$O(nk)$的树形$dp$吧,设$f[i]$表示处理完这$i$颗子树的最小花费,同时再设一个$mi[i]$表示$i$到根节点$1$路径上的距离最小值。于是有:

这样就有$40$分了。

考虑优化:这里可以用虚树来优化,先把所有点按照$DFS$序进行排序,然后将相邻两个点的$LCA$以及$1$号点加入进$LCA$,然后虚树就构好了,考虑欧拉序的特殊性质,所以再还原出欧拉序,在上面做$dp$就好了。(xgzc告诉我可以再$dfs$一遍,但我不想写了,欧拉序多好啊)

#include <set>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <algorithm>
using std::min; using std::max;
using std::swap; using std::sort;
typedef long long ll;
using std::set;

template<typename T>
void read(T &x) {
    int flag = 1; x = 0; char ch = getchar();
    while(ch < '0' || ch > '9') { if(ch == '-') flag = -flag; ch = getchar(); }
    while(ch >= '0' && ch <= '9') x = x * 10 + ch - '0', ch = getchar();
    x *= flag;
}

const int N = 2.5e5 + 10, Inf = 1e9 + 7;
int n, m, dfin[N], dfout[N], tim;
int cnt, from[N], to[N << 1], nxt[N << 1];
int siz[N], son[N], dep[N], top[N], fa[N];
int nt[N << 1], vis[N], s[N << 1], tt;
ll mi[N], f[N], dis[N << 1];
inline void addEdge(int u, int v, ll w) {
    to[++cnt] = v, dis[cnt] = w, nxt[cnt] = from[u], from[u] = cnt;
}
inline bool cmp(const int &x, const int &y) {
    int k1 = x > 0 ? dfin[x] : dfout[-x], k2 = y > 0 ? dfin[y] : dfout[-y];
    return k1 < k2;
}

void dfs(int u) {
    siz[u] = 1, dfin[u] = ++tim, dep[u] = dep[fa[u]] + 1;
    for(int i = from[u]; i; i = nxt[i]) {
        int v = to[i]; if(v == fa[u]) continue;
        mi[v] = min(mi[u], dis[i]);
        fa[v] = u, dfs(v), siz[u] += siz[v];
        if(siz[v] > siz[son[u]]) son[u] = v;
    } dfout[u] = ++tim;
}
void dfs(int u, int t) {
    top[u] = t; if(!son[u]) return ;
    dfs(son[u], t);
    for(int i = from[u]; i; i = nxt[i])
        if(to[i] != son[u] && to[i] != fa[u]) dfs(to[i], to[i]);
}
int lca(int x, int y) {
    int fx = top[x], fy = top[y];
    while(fx != fy)
        if(dep[fx] > dep[fy]) x = fa[fx], fx = top[x];
        else y = fa[fy], fy = top[y];
    return dep[x] < dep[y] ? x : y;
}

int main () {
    read(n);
    for(int i = 1; i < n; ++i) {
        int u, v; ll w; read(u), read(v), read(w);
        addEdge(u, v, w), addEdge(v, u, w);
    }
    mi[1] = Inf, dfs(1), dfs(1, 1), read(m);
    for(int i = 1; i <= m; ++i) {
        int tot; read(tot);
        for(int j = 1; j <= tot; ++j)
            read(nt[j]), vis[nt[j]] = true, f[nt[j]] = mi[nt[j]];
        sort(&nt[1], &nt[tot + 1], cmp);
        for(int j = 1; j < tot; ++j) {
            int cf = lca(nt[j], nt[j + 1]);
            if(!vis[cf]) nt[++tot] = cf, vis[cf] = true;
        }
        int tmp = tot;
        for(int j = 1; j <= tmp; ++j)
            nt[++tot] = -nt[j];
        if(!vis[1]) nt[++tot] = 1, nt[++tot] = -1;
        sort(&nt[1], &nt[tot + 1], cmp);
        for(int j = 1; j <= tot; ++j)
            if(nt[j] > 0) s[++tt] = nt[j];
            else {
                int now = s[tt--];
                if(now != 1) { int fat = s[tt]; f[fat] += min(f[now], mi[now]); }
                else printf("%lld\n", f[1]);
                f[now] = vis[now] = 0;
            }
    }
    return 0;
}