书P380
① LCA:https://blog.nowcoder.net/n/cbc02f19ffa7450d9298b9474c6bd4b8
② dfs:https://blog.nowcoder.net/n/abf5e27bb5c243ebb12d8aced6e11411
时间复杂度 O(n+m)
#include<bits/stdc++.h>
using namespace std;
typedef pair<int, int> P;
typedef long long ll;
const int MAXN = 1e5 + 5; // 最大节点数
const int LOG_N = 40; //只能是40,不然会TLE
int n, m;
int head[MAXN];
struct Edge
{
int to, next;
}edge[MAXN * 2];
int depth[MAXN]; // 节点深度
int parent[LOG_N][MAXN]; // parent[k][i]表示 i 向上走 2^k 步能到达的节点
int hide[MAXN]; // u到它的父亲节点所连边被覆盖过几次
int cnt;
void add(int u, int v)
{
edge[cnt].to = v;
edge[cnt].next = head[u];
head[u] = cnt++;
}
void dfs(int pre, int u, int d)
{
parent[0][u] = pre;
depth[u] = d;
for(int i = head[u]; ~i; i = edge[i].next)
{
int v = edge[i].to;
if(v != pre) dfs(u, v, d + 1);
}
}
void init()
{
int root = 1;
dfs(-1, root, 0);//pre代表父节点,u是当前点,d是深度
for(int k = 1; k < LOG_N; k++)
{
for(int i = 1; i <= n; i++)
{
if(parent[k - 1][i] < 0) parent[k][i] = -1;
else parent[k][i] = parent[k - 1][parent[k - 1][i]];//树上倍增的经典
}
}
}
int lca(int u, int v)
{
if(depth[u] > depth[v]) swap(u, v);
for(int i = 0; i < LOG_N; i++) // u 和 v 向上走到同一深度
{
if((depth[v] - depth[u]) >> i & 1) // 把 (depth[v] - depth[i]) 化成二进制后可以看到,就是找到所有 1 的位置
{
v = parent[i][v];
}
}
if(v == u) return u;
for(int i = LOG_N - 1; i >= 0; i--) // 找 lca
{
if(parent[i][u] != parent[i][v]) // 如果相同,那么一定是公共祖先或公共祖先之上的节点
{
u = parent[i][u];
v = parent[i][v];
}
}
return parent[0][u];
}
void dfs2(int pre, int u)
{
for(int i = head[u]; ~i; i = edge[i].next)
{
int v = edge[i].to;
if(v != pre)
{
dfs2(u, v);
hide[u] += hide[v];//求以u为根,子树的节点权值总和
}
}
}
int main()
{
while(~scanf("%d%d", &n, &m))
{
memset(head, -1, sizeof head);
for(int i = 1; i < n; i++)
{
int x, y;
scanf("%d%d", &x, &y);
add(x, y);
add(y, x);
}
init();
for(int i = 0; i < m; i++)
{
int x, y;
scanf("%d%d", &x, &y);
int node = lca(x, y);
hide[x]++;
hide[y]++;
hide[node] -= 2;
}
dfs2(-1, 1);
int ans = 0;
for(int i = 2; i <= n; i++)
{
if(hide[i] == 0) ans += m;//没有边覆盖,可以随机一条非树边
else if(hide[i] == 1) ans++;//只有一条非树边覆盖,答案唯一
}
printf("%d\n", ans);
break;//原文没有加进来
}
return 0;
}来源:https://blog.csdn.net/weixin_33946605/article/details/93663457

京公网安备 11010502036488号