CF592D Super M

嘟嘟嘟


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


手玩样例得知，这哥们儿瞬移到的城市(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
0
", 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;
}