96 lines
2.6 KiB
C++
96 lines
2.6 KiB
C++
|
#include <bits/stdc++.h>
|
|||
|
|
using namespace std;
|
|||
|
|
const int N = 100010, M = 200010;
|
|||
|
|
int depth[N], f[N][25];
|
|||
|
|
int n, m;
|
|||
|
|
int d[N]; // 差分数组
|
|||
|
|
int ans; // 存答案
|
|||
|
|
const int T = 17;
|
|||
|
|
// 邻接表
|
|||
|
|
int e[M], h[N], idx, ne[M];
|
|||
|
|
void add(int a, int b) {
|
|||
|
|
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
|
|||
|
|
}
|
|||
|
|
// 树上倍增
|
|||
|
|
void bfs() {
|
|||
|
|
queue<int> q;
|
|||
|
|
q.push(1);
|
|||
|
|
depth[1] = 1;
|
|||
|
|
while (q.size()) {
|
|||
|
|
int u = q.front();
|
|||
|
|
q.pop();
|
|||
|
|
for (int i = h[u]; ~i; i = ne[i]) {
|
|||
|
|
int v = e[i];
|
|||
|
|
if (!depth[v]) {
|
|||
|
|
depth[v] = depth[u] + 1;
|
|||
|
|
q.push(v);
|
|||
|
|
f[v][0] = u;
|
|||
|
|
for (int k = 1; k <= T; k++) f[v][k] = f[f[v][k - 1]][k - 1];
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// 标准lca
|
|||
|
|
int lca(int a, int b) {
|
|||
|
|
if (depth[a] < depth[b]) swap(a, b);
|
|||
|
|
for (int i = T; i >= 0; i--)
|
|||
|
|
if (depth[f[a][i]] >= depth[b]) a = f[a][i];
|
|||
|
|
if (a == b) return a;
|
|||
|
|
for (int i = T; i >= 0; i--)
|
|||
|
|
if (f[a][i] != f[b][i])
|
|||
|
|
a = f[a][i], b = f[b][i];
|
|||
|
|
return f[a][0];
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 差分数组还原
|
|||
|
|
void dfs(int u, int fa) {
|
|||
|
|
for (int i = h[u]; ~i; i = ne[i]) {
|
|||
|
|
int v = e[i];
|
|||
|
|
if (v == fa) continue;
|
|||
|
|
dfs(v, u);
|
|||
|
|
d[u] += d[v];
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int main() {
|
|||
|
|
int a, b;
|
|||
|
|
scanf("%d %d", &n, &m);
|
|||
|
|
memset(h, -1, sizeof h);
|
|||
|
|
for (int i = 1; i < n; i++) { // n-1条边
|
|||
|
|
scanf("%d %d", &a, &b);
|
|||
|
|
add(a, b), add(b, a);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// lca的准备动作
|
|||
|
|
bfs();
|
|||
|
|
|
|||
|
|
// 读入附加边
|
|||
|
|
for (int i = 0; i < m; i++) {
|
|||
|
|
scanf("%d %d", &a, &b);
|
|||
|
|
// 树上差分
|
|||
|
|
// d[a]的含义：从a->fa这边条，多了一个环
|
|||
|
|
// d[b]的含义：从b->fb这边条，多了一个环
|
|||
|
|
d[a]++, d[b]++;
|
|||
|
|
int p = lca(a, b);
|
|||
|
|
/*
|
|||
|
|
Q:lca(a,b)为什么要减2？
|
|||
|
|
A:边差分，每条边是下放到下面的那个点上，用点来表示这个边的。
|
|||
|
|
其实，每个点表示的是它向上那条边被覆盖的次数，对于lca(a,b)而言，由于dfs统计进行前缀和汇总时，
|
|||
|
|
是左子树+右子树这样的形式进行汇总的，也按同样逻辑处理就会多出2个，需要扣除掉。
|
|||
|
|
*/
|
|||
|
|
d[p] -= 2;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 差分数组求前缀和
|
|||
|
|
dfs(1, 0);
|
|||
|
|
|
|||
|
|
// Q:为什么要从2开始？
|
|||
|
|
// A:因为1是根，1是没有边的，边是向上的，从2开始才有边
|
|||
|
|
for (int i = 2; i <= n; i++) {
|
|||
|
|
if (d[i] == 0) ans += m;
|
|||
|
|
if (d[i] == 1) ans += 1;
|
|||
|
|
}
|
|||
|
|
// 输出
|
|||
|
|
printf("%d\n", ans);
|
|||
|
|
return 0;
|
|||
|
|
}
|