树上差分（砍树）蓝桥杯

#include <bits/stdc++.h>
using namespace std;
int n,m,tot=0,a,b,ans=0;
const int N=1e6+10;
int vis[N]={0},fa[N]={0},F[N]={0};
int ver[N]={0},Next[N]={0},edge[N]={0},head[N]={0};
map<pair<int,int>,int> Pos;
vector<int> query[N];
void add(int x,int y,int z){
    ver[++tot]=y,edge[tot]=z,Next[tot]=head[x],head[x]=tot;
    
}
void add_query(int x,int y){
    query[x].push_back(y);
    query[y].push_back(x);
}
int find(int x){
    if(x==fa[x]) return x;
    return fa[x]=find(fa[x]);
}
void tarjan(int u){
    vis[u]=1;
    for(int i=head[u];i;i=Next[i]){
        int y=ver[i];
        if(!vis[y]){
            tarjan(y);
            fa[y]=u;
        }
    }
    for(int i=0;i<query[u].size();i++){
        int y=query[u][i];
        if(vis[y]==2){
            int lca=find(y);
            F[u]++;
            F[y]++;
            F[lca]-=2;
        }
    }
    vis[u]=2;
}
void dfs(int x,int f){
    for(int i=head[x];i;i=Next[i]){
        int y=ver[i];
        if(y!=f){
            dfs(y,x);
            F[x]+=F[y];
        }
    }
    if(f==0) return;
    int t=Pos[{x,f}];
    if(F[x]==m) ans=max(ans,t);
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++){
        scanf("%d%d",&a,&b);
        add(a,b,1),add(b,a,1);
        Pos[{a,b}]=Pos[{b,a}]=i;
    }
    for(int i=1;i<=m;i++){
        scanf("%d%d",&a,&b);
        add_query(a,b);
    }
    for(int i=1;i<=n;i++) fa[i]=i;
    tarjan(1);
    dfs(1,0);
    if(ans==0) printf("-1");
    else printf("%d",ans);
}