LibreOJ #10131 暗的连锁（转换题意+树上差分--解决边覆盖）

题目链接：https://loj.ac/problem/10131

题意转化：
“主要边”构成一棵树，“附加边”则是“非树边”。
把一条附加边(x,y)添加到主要边构成的树中，会与树上x，y之间的路径一起形成一条环。
称为：每条附加边(x,y)都把树上x，y之间的路径上的每条边“覆盖了一次”

转化为的问题模型：
给定一个无向图和一颗生成树，求每条“树边”被“非树边”覆盖了多少次？

引入树上差分算法：
与差分序列的思想类似

具体操作：
对应在树上，我们给每一个结点初始为0的权值，然后对每条非树边(x,y)，令结点x的权值加1，结点y的权值加1，结点LCA(x,y)的权值减2
对整棵树进行一次深度优先遍历
求出F[x]表示以x为根的子树中各个结点的权值之和，F[x]表示x与它的父节点之间的“树边”被覆盖的次数。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=1e6;
int fir[N],to[N],net[N],tot;
int dep[N],f[N][21];
int num[N],ff[N];
int m,n;
void add(int x,int y)
{
    to[++tot]=y;
    net[tot]=fir[x];
    fir[x]=tot;
}
void Deal_first(int u,int father)/*预处理出每个结点的深度以及每个结点走2^0 2^1 2^2...步到达的结点*/
{
    dep[u]=dep[father]+1;
    for(int i=0; i<=19; i++)
        f[u][i+1]=f[f[u][i]][i];
    for(int i=fir[u]; i; i=net[i])
    {
        int v=to[i];
        if(v==father)/*因为无向边，所以会出现v=father的情况*/
            continue;
        f[v][0]=u;
        Deal_first(v,u);
    }
}
int LCA(int x,int y)
{
    if(dep[x]<dep[y])
        swap(x,y);
    for(int i=20; i>=0; i--)
    {
        if(dep[f[x][i]]>=dep[y])
            x=f[x][i];
        if(x==y)
            return x;
    }
    for(int i=20; i>=0; i--)/*同一高度,但不是同一点*/
    {
        if(f[x][i]!=f[y][i])/*刚开始可能跳到结点不存在的位置，f[x][i]=f[y][i]=0,不执行if选择语句*/
        {
            x=f[x][i];
            y=f[y][i];
        }
    }
    return f[x][0];
}
void getf(int u,int father)/*以一个点为根的子树,一直往下遍历,对应路径上的点的权值累加,这个点是子树所有叶节点的LCA,回溯,最后汇聚在这个点,num[LCA(,y)]起初的减2,在这里相抵消*/
{
    for(int i=fir[u]; i; i=net[i])
    {
        int v=to[i];
        if(v==father)
            continue;
        getf(v,u);
        num[u]+=num[v];
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0; i<n-1; i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);/*树的n-1条无向边*/
    }
    Deal_first(1,0);//引入虚拟祖先结点0
    for(int i=1; i<=m; i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        num[x]++;
        num[y]++;
        num[LCA(x,y)]-=2;
    }
    getf(1,0);
    int ans=0;
    for(int i=2; i<=n; i++)
    {
        if(num[i]==0)
            ans+=m;
        if(num[i]==1)
            ans++;
    }
    printf("%d\n",ans);
    return 0;
}