POJ - 3417 Network LCA（倍增）+树上差分（关于边）

POJ - 3417 Network

树上差分讲解

题意；

给你一棵树，原来有 n-1 条边，有 m 条新加的边，求删除一条旧的边和一条新的边将树最少分成两部分 有多少种分法

思路：

思路转自：https://www.cnblogs.com/lsgjcya/p/9247167.html

事实上我们考虑，每多加一条非树边，在不重的情况下，树上都会多出一个环。

考虑断掉某条树边：
1.它可能没进入环，这时很显然再随意断开一条非树边即可。对答案的贡献为m。
2.它进入了一个环，这时我们再断开环中的那条非树边，成为一种方案。
3.它进入了两个环，树上部分成为两部分，但因为这条边处于两个环中，所以还有两条非树边从上半部分连到下半部分，无法使图成为两块。
4.它进入了多条环时与进入了两个环时情况类似，无法使图成为两块。

那这时思路就很显然了，对于每条非树边，我们把环上的每条树边入环个数++，但是我们发现这样做的话时间复杂度比较高，因为这样做需要遍历路径上的每一条边。这时考虑能不能把边转移到点上来记录，巨佬这时很容易想到运用树上差分的思想来解决(我还是太菜了，完全没想到)。对于每条非树边(u,v):

a[u]++;a[v]++;a[lca(u,v)]-=2;

最后对于每个点按照上面的方法处理即可。
注意：不需要考虑根节点

 

#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
const int maxn =1e5+10;
struct node
{
    int to,next;
}e[2*maxn];
int fa[maxn][20]; //父节点   表示节点i往上跳2^j次后的节点  注意j的取值范围
int head[maxn];
int d[maxn];  //当前节点所在的深度
int n,m,cnt;
int a[maxn];
void add(int u,int v)
{
    cnt++;
    e[cnt].to = v;
    e[cnt].next = head[u];
    head[u] = cnt;
}
void dfs(int u,int f)
{
    for(int i=head[u]; i ;i=e[i].next)
    {
        int v = e[i].to;
        if(v==f)
            continue;
        d[v] = d[u]+1;
        fa[v][0] = u;
        dfs(v,u);
    }
} //深搜出各点的深度，存在d中
void bz()
{
    for(int j=1;j<=19;j++)
        for(int i=1;i<=n;i++)
            fa[i][j] = fa[fa[i][j-1]][j-1]; //i的第2^j祖先就是i的第2^(j-1)祖先的第2^(j-1)祖先
                //i的父辈们的fa数组都已经计算完毕，所以可以用来计算i
}
int lca(int u,int v)
{
    if(d[u]<d[v])
        swap(u,v); //保证深度大的点为u，方便操作
    int dc = d[u]-d[v];   //计算深度差
    for(int i=0;i<=19;i++)   //值得注意的是，这里需要从零枚举
    {                      //此循环用于提到深度相同
        if( (1<<i)&dc )
            u = fa[u][i];
    }
    if(u==v)
        return u;
    for(int i=19;i>=0;i--)   //如果从小到大的话就有可能无法提到正确位置
    {
        if(fa[u][i]!=fa[v][i])//跳2^j步，如果不一样，就跳，否则不跳 如果相等有可能都为0 即超过根节点
        {//在同一高度了,他们一起向上跳,跳他们不相同节点，当全都跳完之后fa[u][0]就是lca
            u = fa[u][i];
            v = fa[v][i];
        }
    }
    u = fa[u][0];  //上述过程做完，两点都在LCA下一层，所以走一步即可
    return u;
}
void get(int u,int f)
{
    for(int i=head[u]; i ;i=e[i].next)
    {
        int v = e[i].to;
        if(v==f)
            continue;
        get(v,u);
        a[u] += a[v];
    }
}
int main()
{
    int x,y;
    ll ans = 0;
    cnt = 0;
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    dfs(1,0);
    bz();
    for(int i=0;i<m;i++)
    {
        scanf("%d%d",&x,&y);
        a[x]++,a[y]++;
        a[lca(x,y)] -= 2;
    }
    get(1,0);  //更新 a 数组
    for(int i=2;i<=n;i++)
    {
        if(a[i]==0)
            ans += m;
        else if(a[i]==1)
            ans++;
    }
    printf("%lld",ans);
    return 0;
}