CF592D Super M

时间:2023-03-08 20:38:03

嘟嘟嘟



首先这题虽然不是很难,但是黄题是不是有点过分了……好歹算个蓝题啊。



手玩样例得知,这哥们儿瞬移到的城市\(A\)一定是这些被攻击的城市构成的树的一个叶子,然后他经过的最后一个城市\(B\)和\(A\)构成的链一定是这棵新构成的树的直径(突然想到虚树)。

别激动,这题根本不用虚树。

我们只用求一遍树的直径就行了,只不过这个直径的端点必须满足都是被攻击的城市,则第一问就是端点中的较小值。



考虑第二问。

直径上的城市只会走一遍,而直径外的城市必须走过去再回来。所以我们从直径一段开始遍历整个直径,每经过一个点,就dfs这个点直径之外的子树,并统计子树内走到被攻击的城市的距离和。那么答案就是这些距离+加树的直径长度。

距离和的求法用树形dp就行。我们考虑每一条边的贡献。如果一个点的的儿子的子树内有被攻击的城市,则这条边一定会被走过,答案加2即可。



然后记得特判被攻击的城市只有一个的情况(被hack了……)。

#include<cstdio>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cctype>
#include<vector>
#include<stack>
#include<queue>
using namespace std;
#define enter puts("")
#define space putchar(' ')
#define Mem(a, x) memset(a, x, sizeof(a))
#define In inline
typedef 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;
}