【树上差分】闇の連鎖

原题链接：https://www.acwing.com/problem/content/description/354/

倍增法求lca：https://blog.csdn.net/qq_44661312/article/details/97947398

传说中的暗之连锁被人们称为 Dark。

Dark 是人类内心的黑暗的产物，古今中外的勇者们都试图打倒它。

经过研究，你发现 Dark 呈现无向图的结构，图中有 N 个节点和两类边，一类边被称为主要边，而另一类被称为附加边。

Dark 有 N – 1 条主要边，并且 Dark 的任意两个节点之间都存在一条只由主要边构成的路径。

另外，Dark 还有 M 条附加边。

你的任务是把 Dark 斩为不连通的两部分。

一开始 Dark 的附加边都处于无敌状态，你只能选择一条主要边切断。

一旦你切断了一条主要边，Dark 就会进入防御模式，主要边会变为无敌的而附加边可以被切断。

但是你的能力只能再切断 Dark 的一条附加边。

现在你想要知道，一共有多少种方案可以击败 Dark。

注意，就算你第一步切断主要边之后就已经把 Dark 斩为两截，你也需要切断一条附加边才算击败了 Dark。

输入格式

第一行包含两个整数 N 和 M。

之后 N – 1 行，每行包括两个整数 A 和 B，表示 A 和 B 之间有一条主要边。

之后 M 行以同样的格式给出附加边。

输出格式

输出一个整数表示答案。

数据范围

N≤100000,M≤200000,数据保证答案不超过2^31−1

输入样例：

4 1 
1 2 
2 3 
1 4 
3 4 

输出样例：

3

思路：加上一条附加边后一定会与主要边形成一个环，那么如果要砍断环上的某一个点，就需要多砍这一条附加边 。每加一条附加边，形成环上的每一点都需要多砍一条边。这样，只需要计算每个点在多少个环中，就可得出方案数。如果点x在零个环中，那么砍断这个点只需砍掉随意一条附加边，方案数为m；如果点x在一个环中，那么砍断这个点需砍掉形成这个环的附加边，方案数为1.如果点x在大于一个环中，那么无法砍断这个点，方案数为0；

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+10,M = 2*N;

int first[N],ne[M],v[M],num_edge;
int st[N][20],p[N],f[N],depth[N];
int n,m,ans;

void add(int x,int y){
    ne[++num_edge]=first[x];
    v[num_edge]=y;
    first[x]=num_edge;
}

void dfs_depth(int x,int y){
    //得到深度和直属父亲
    depth[x]=depth[y]+1;
    f[x]=y;
    int z;
    for(int i = first[x]; i!=-1; i=ne[i]){
        z=v[i];
        if(z==y)continue;
        dfs_depth(z,x);
    }
}

void pre_lca(){
    //跳一步跳到自己的父亲
    for(int i = 1; i <= n; i++)st[i][0]=f[i];
    //跳从i开始跳2^k步
    for(int k=1; k <= 16; k++)
        for(int i = 1; i <= n; i++)
            st[i][k]=st[st[i][k-1]][k-1];
}

int get_lca(int x,int y){//倍增法求lca
    //保证x比y深
    if(depth[x]<depth[y])swap(x,y);
    //将x跳到与y同高
    for(int k = 16; k >= 0; k--)
        if(depth[st[x][k]]>=depth[y])
            x=st[x][k];
    //y即为公共祖先的情况
    if(x==y)return x;
    //xy一起往上跳
    for(int k = 16; k >= 0; k--)
        if(st[x][k]!=st[y][k])
            x=st[x][k],y=st[y][k];
    //xy的父亲即为两者的最近公共祖先
    return f[x];

}

int dfs_ans(int x,int y){
    int res=p[x],u,z;

    for(int i = first[x]; i!=-1; i=ne[i]){
        u=v[i];
        if(u==y)continue;
        z=dfs_ans(u,x);//得到子树的前缀
        //加上砍断z的可能方案数
        if(z==0)ans+=m;
        else if(z==1)ans++;
        res+=z;
    }
    //res即为x的前缀和
    return res;

}

int main(){
    scanf("%d%d",&n,&m);
    memset(first,-1,sizeof(first));
    //建图
    int x,y,z;
    for(int i = 1; i < n; i++){
        scanf("%d%d",&x,&y);
        add(x,y),add(y,x);
    }
    //cout<<"a";
    //得到每个点的深度
    dfs_depth(1,0);
    //cout<<"b";
    //预处理st数组
    pre_lca();
    //处理附加边，即将形成环的每条边权值加一
    for(int i = 1; i <= m; i++){
        scanf("%d%d",&x,&y);
        z=get_lca(x,y);
        //差分约束
        p[x]++,p[y]++,p[z]-=2;
    }
    //求前缀得到每个点的值
    dfs_ans(1,0);
    printf("%d\n",ans);
}