首先这题虽然不是很难,但是黄题是不是有点过分了……好歹算个蓝题啊。
手玩样例得知,这哥们儿瞬移到的城市
\(A\)一定是这些被攻击的城市构成的树的一个叶子,然后他经过的最后一个城市
\(B\)和
\(A\)构成的链一定是这棵新构成的树的直径(突然想到虚树)。
别激动,这题根本不用虚树。
我们只用求一遍树的直径就行了,只不过这个直径的端点必须满足都是被攻击的城市,则第一问就是端点中的较小值。
考虑第二问。
直径上的城市只会走一遍,而直径外的城市必须走过去再回来。所以我们从直径一段开始遍历整个直径,每经过一个点,就dfs这个点直径之外的子树,并统计子树内走到被攻击的城市的距离和。那么答案就是这些距离+加树的直径长度。
距离和的求法用树形dp就行。我们考虑每一条边的贡献。如果一个点的的儿子的子树内有被攻击的城市,则这条边一定会被走过,答案加2即可。
然后记得特判被攻击的城市只有一个的情况(被hack了……)。
#include #include #include #include #include #include #include #include #include #include using namespace std;#define enter puts("") #define space putchar(' ')#define Mem(a, x) memset(a, x, sizeof(a))#define In inlinetypedef long long ll;typedef double db;const int INF = 0x3f3f3f3f;const db eps = 1e-8;const int maxn = 2e5 + 5;inline ll read(){ ll ans = 0; char ch = getchar(), last = ' '; while(!isdigit(ch)) last = ch, ch = getchar(); while(isdigit(ch)) ans = (ans << 1) + (ans << 3) + ch - '0', ch = getchar(); if(last == '-') ans = -ans; return ans;}inline void write(ll x){ if(x < 0) x = -x, putchar('-'); if(x >= 10) write(x / 10); putchar(x % 10 + '0');}int n, m;bool vis[maxn];struct Edge{ int nxt, to;}e[maxn << 1];int head[maxn], ecnt = -1;In void addEdge(int x, int y){ e[++ecnt] = (Edge){head[x], y}; head[x] = ecnt;}int dep[maxn], fa[maxn], Max = 0, A, B;In void dfs1(int now, int _f, int dis, int& id){ if(vis[now] && (dis > Max || (dis == Max && now < id))) Max = dis, id = now; dep[now] = dep[_f] + 1; fa[now] = _f; for(int i = head[now], v; ~i; i = e[i].nxt) if((v = e[i].to) ^ _f) dfs1(v, now, dis + 1, id);}bool dia[maxn];int a[maxn], b[maxn], acnt = 0, bcnt = 0;In void solve(int x, int y){ a[++acnt] = x; b[++bcnt] = y; dia[x] = dia[y] = 1; while(x ^ y) { if(dep[x] > dep[y]) a[++acnt] = x = fa[x], dia[x] = 1; else b[++bcnt] = y = fa[y], dia[y] = 1; } --acnt;}int ans = 0;In bool dfs2(int now, int _f, int dis){ int flg = 0; for(int i = head[now], v; ~i; i = e[i].nxt) { if(dia[v = e[i].to] || v == _f) continue; int tp = dfs2(v, now, dis + 1); if(tp) ans += 2; flg |= tp; } return flg || vis[now];}int main(){ Mem(head, -1); n = read(), m = read(); for(int i = 1; i < n; ++i) { int x = read(), y = read(); addEdge(x, y), addEdge(y, x); } for(int i = 1; i <= m; ++i) A = read(), vis[A] = 1; dfs1(A, 0, 0, A), Max = 0, dfs1(A, 0, 0, B); if(!A || !B) {printf("%d\n0\n", A | B); return 0;} solve(A, B); for(int i = 1; i <= acnt; ++i) dfs2(a[i], 0, 0); //这两行是遍历直径 for(int i = 1; i <= bcnt; ++i) dfs2(b[i], 0, 0); write(min(A, B)), enter, write(ans + Max), enter; return 0;}