[蓝桥杯] 砍树

树上边差分 + LCA。

假设可以将A，B 分为两个不同的联通块，它们间会有若干条条‘’关键路径‘’连接两个连通块。 对于每组（ai， bi） 都必然经过这些‘’关键路径‘’， 所以‘’关键路径‘’经过m次。我们对每组（ai， bi）路径上的边 + 1，最后经过m次的边就是答案， 通过 树上边差分 和 LCA 处理下即可。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
//#define int long long
#define ios ios::sync_with_stdio(false); cin.tie(0);cout.tie(0)
typedef pair<int, int> pii;
const int N = 1e5+5;
int n, m;
vector<pii > e[N];
int a[N], b[N], w[N], c[N], k[N];
int dep[N], fa[N][22];
void dfs(int u, int father){
    dep[u] = dep[father] + 1;
    fa[u][0] = father;
    //cout << u << " " << father << endl; 
    for(int i = 1; i <= 20; i++){
        fa[u][i] = fa[fa[u][i - 1]][i - 1];
    }
    for(auto v : e[u]){
        if(v.first != father) {
            k[v.first] = v.second;
            dfs(v.first, u);
        }
    }
}
int lca(int u, int v){
    if(dep[u] < dep[v]) swap(u, v);
    for(int i = 20; i >= 0; i--){
        if(dep[fa[u][i]] >= dep[v]) u = fa[u][i];
    }
    if(u == v) return v;
    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 dfs1(int x){
    w[x] = c[x];
    for(auto y : e[x]){
        int v = y.first;
        if(dep[v] == dep[x] + 1){
            dfs1(v);
            w[x] += w[v];
        }
    }
}


void solve(){
    cin >> n >> m;
    int u, v;
    for(int i = 1; i <= n - 1; i++){
        cin >> u >> v;
        e[u].push_back({v, i});
        e[v].push_back({u, i});
    }
    dfs(1, 0);
    for(int i = 1; i <= m; i++){
        cin >> a[i] >> b[i];
        ++c[a[i]], ++c[b[i]], c[lca(a[i], b[i])] -= 2;
    }
    dfs1(1);
    int ans = -1;
    for(int i = 1; i <= n; i++){
        //cout << w[i] << endl;
        if(w[i] == m) ans = max(ans, k[i]); 
    }
    cout << ans << endl;
}

signed main(){
    ios; int _;
    _ = 1; //cin >> _;
    while(_--){
        solve();
    }
    return 0;
}