原题传送门
一个正确的暴力思路是遍历树,然后对于每个点,把它和它父亲之间的边断掉。那就看看,边集 E 2 E2 E2中有多少条把两个连通块联通的,记条数为 c n t cnt cnt
- c n t = 0 cnt=0 cnt=0,说明 E 2 E2 E2中每一条边都可以选择,答案累加 m m m
- c n t = 1 cnt=1 cnt=1,那就只能断掉那一条边,答案累加1
- c n t > 1 cnt>1 cnt>1,没有边可以断
如何统计这个 c n t cnt cnt?
可以暴力并查集,但是我们可以树上差分
对于一条 E 2 E2 E2中的边 ( u , v ) (u,v) (u,v)
++delta[u],++delta[v],delta[lca(u,v)]-=2
直接统计
有道是“十年OI两场空,不开ll见祖宗”,可曾想我也有被这个坑到的一天
不开ll直接掉60分
Code:
#include <bits/stdc++.h>
#define maxn 200010
#define int long long
using namespace std;
struct Edge{
int to, next;
}edge[maxn << 1];
int num, head[maxn], delta[maxn], d[maxn], fa[maxn][25], n, m, ans;
inline int read(){
int s = 0, w = 1;
char c = getchar();
for (; !isdigit(c); c = getchar()) if (c == '-') w = -1;
for (; isdigit(c); c = getchar()) s = (s << 1) + (s << 3) + (c ^ 48);
return s * w;
}
void addedge(int x, int y){
edge[++num] = (Edge){
y, head[x]}, head[x] = num; }
void build(int u, int pre){
d[u] = d[pre] + 1, fa[u][0] = pre;
for (int i = 0; fa[u][i]; ++i) fa[u][i + 1] = fa[fa[u][i]][i];
for (int i = head[u]; i; i = edge[i].next){
int v = edge[i].to;
if (v != pre) build(v, u);
}
}
int lca(int u, int v){
if (d[u] < d[v]) swap(u, v);
for (int i = 20; i >= 0; --i) if (d[u] - (1 << i) >= d[v]) u = fa[u][i];
if (u == v) return u;
for (int i = 20; i >= 0; --i)
if (fa[u][i] != fa[v][i]) u = fa[u][i], v = fa[v][i];
return fa[u][0];
}
void dfs(int u, int pre){
for (int i = head[u]; i; i = edge[i].next){
int v = edge[i].to;
if (v == pre) continue;
dfs(v, u);
delta[u] += delta[v];
}
if (u == 1) return;
if (!delta[u]) ans += m;
else if (delta[u] == 1) ++ans;
}
signed main(){
freopen("pa.in", "r", stdin);
freopen("pa.out", "w", stdout);
n = read(), m = read();
for (int i = 1; i < n; ++i){
int x = read(), y = read();
addedge(x, y), addedge(y, x);
}
build(1, 0);
for (int i = 1; i <= m; ++i){
int x = read(), y = read(), Lca = lca(x, y);
++delta[x], ++delta[y], delta[Lca] -= 2;
}
dfs(1, 0);
printf("%lld\n", ans);
return 0;
}