ACM-ICPC 2018 青岛赛区网络预赛 ZOJ - 4048 Red Black Tree（LCA）

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=5807

题目大意：给定一棵n个节点的树，节点1为根节点，树上有m个节点是红色的，剩下的节点都是黑色的。定义树上每个节点的花费为当前节点到达最近的为红色的祖先的距离。接下来有q次查询，每次查询给出k个节点，你允许将树上对的任意一个节点染成红色（仅对这次查询有效），现在要求出如何染色，才能使得这k个点中花费是最大的节点的花费最小化，输出这个最大花费。

题目思路：如果每次查询中没有染色的操作的话，我们可以通过dfs预处理出每个节点离它最近的为红色的祖先是哪个，距离是多少。接下来考虑加入染色操作的情况，由于是要使得最大花费最小化，所以我们就肯定是考虑对于一开始花费最大的点的祖先进行染色，这样就可以降低最大值。根据这个思想，我们就可以先将这k个节点按照花费大小由大到小进行排序，在这之后枚举对前p个点的lca进行染色的情况，求出最小的最大值即可，但要记得考虑边界情况。

具体实现看代码：

#include <bits/stdc++.h>
#define fi first
#define se second
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define pb push_back
#define MP make_pair
#define lowbit(x) x&-x
#define clr(a) memset(a,0,sizeof(a))
#define _INF(a) memset(a,0x3f,sizeof(a))
#define FIN freopen("in.txt","r",stdin)
#define IOS ios::sync_with_stdio(false)
#define fuck(x) cout<<"["<<#x<<" "<<(x)<<"]"<<endl
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int MX = 1e5 + 7;

int n, m, q, _;
int red[MX];
struct edge {
    int v, nxt;
    ll w;
} E[MX << 1];
int head[MX], tot, sz;
int dep[MX], ST[MX][20], vis[MX], vec[MX];
ll dis[MX], d[MX];
bool cmp(int a, int b) {
    return d[a] > d[b];
}
void init(int _n) {
    for (int i = 0; i <= _n; i++) {
        head[i] = -1;
        vis[i] = red[i] = dis[i] = d[i] = 0;
    }
    tot = sz = 0;
}
void add_edge(int u, int v, ll w) {
    E[tot].v = v;
    E[tot].w = w;
    E[tot].nxt = head[u];
    head[u] = tot++;
}
void dfs(int u,int dd,int fa,ll nw) {
    if(red[u]) nw = 0;
    d[u] = nw;
    dep[u] = dd;
    ST[u][0] = fa;
    for(int i=head[u]; ~i; i=E[i].nxt) {
        int v=E[i].v,w=E[i].w;
        if(v==fa) continue;
        dis[v] = dis[u]+w;
        dfs(v,dd+1,u,nw + w);
    }
}
void pre_solve() {
    dfs(1,0,1,0);
    for(int i=1; i<20; i++) {
        for(int j=1; j<=n; j++) {
            ST[j][i]=ST[ST[j][i-1]][i-1];
        }
    }
}
int LCA(int u,int v) {
    while(dep[u]!=dep[v]) {
        if(dep[u]<dep[v]) swap(u,v);
        int d=dep[u]-dep[v];
        for(int i=0; i<20; i++)
            if(d>>i&1)u=ST[u][i];
    }
    if(u==v) return u;
    for(int i=19; i>=0; i--) {
        if(ST[u][i]!=ST[v][i]) {
            u=ST[u][i];
            v=ST[v][i];
        }
    }
    return ST[u][0];
}

int main() {
    for (scanf("%d", &_); _; _--) {
        scanf("%d%d%d", &n, &m, &q);
        init(n);
        for (int i = 1, x; i <= m; i++) {
            scanf("%d", &x);
            red[x] = 1;
        }
        for (int i = 1; i < n; i++) {
            int u, v, w;
            scanf("%d%d%d", &u, &v, &w);
            add_edge(u, v, w);
            add_edge(v, u, w);
        }
        pre_solve();
        while (q--) {
            int k;
            scanf("%d", &k);
            for (int i = 1; i <= k; i++) scanf("%d", &vec[i]);
            sort(vec + 1, vec + k + 1, cmp);
            ll ans = d[vec[1]], lastcnt = 0;
            int lastlca = vec[1];
            for (int i = 1; i <= k; i++) {
                int lca = LCA(lastlca, vec[i]);
                //fuck(lca);
                ll res1 = lastcnt + dis[lastlca] - dis[lca];
                if (i > 1 && res1 >= d[vec[i - 1]]) break;
                ll res2 = min(d[vec[i]], dis[vec[i]] - dis[lca]);
                ll cnt = max(res1, res2);
                if (cnt >= ans) break;
                if (i + 1 <= k) ans = min(ans, max(cnt, d[vec[i + 1]]));
                else ans = min(ans, cnt);
                lastlca = lca;
                lastcnt = cnt;
            }
            printf("%lld\n", ans);
        }
    }
    return 0;
}