闇の連鎖 树上差分

在这里插入图片描述
思路
附加边x,y,会把x y形成一个圈,则要把x到y路径上所有的点都加1,表示为覆盖的层数,用d[i]表示覆盖的层数
若d【i】=1,则d[i]=1,要切断这个圈的随便一条主要边(在一个个遍历时,我们就求出了这个圈的每条边),然后必须要切断这条附加边。即ans++;
若d【i】=0,则切一条主要边后,随便切一条附加边即可。即ans+=m;
d【i】=2,则无解

#include <stdio.h>
#include <algorithm>
#include <queue>
#include <math.h>
#define INF 0x3f3f3f3f
#define ll long long
using namespace std;
int n,m,tot,t,ans;
int head[100005],nex[200005],ver[200005];
int d[100005],dep[100005],fa[100005][20];
void add(int u,int v)
{
    nex[++tot]=head[u];
    ver[tot]=v;
    head[u]=tot;
}
void bfs()
{
	queue<int>q;
    q.push(1);
    dep[1]=1;
    while(!q.empty())
    {
        int u=q.front();q.pop();
        for(int i=head[u];i;i=nex[i])
        {
            int v=ver[i];
            if(dep[v])continue;
            dep[v]=dep[u]+1;
            fa[v][0]=u;
            for(int j=1;j<=t;j++)
                fa[v][j]=fa[fa[v][j-1]][j-1];
            q.push(v);
        }
    }
}
int lca(int x,int y)
{
    if(dep[x]>dep[y])swap(x,y);
    for(int i=t;i>=0;i--)
        if(dep[fa[y][i]]>=dep[x])y=fa[y][i];
    if(x==y)return x;
    for(int i=t;i>=0;i--)
        if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
void dfs(int u,int pre)                //来求出d[i]
{
    for(int i=head[u];i;i=nex[i])
    {
        int v=ver[i];
        if(v==pre)continue;
        dfs(v,u);
        d[u]+=d[v];
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    int u,v;
    t=(int)(log(n)/log(2))+1;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        add(u,v),add(v,u);
    }
    bfs();
    for(int i=1;i<=m;i++)                              //差分关键,手画推到
    {
        scanf("%d%d",&u,&v);
        d[u]++,d[v]++,d[lca(u,v)]-=2;
    }
    dfs(1,0);
    for(int i=2;i<=n;i++)
    {
    	if(d[i]==1)ans++;
    	else if(d[i]==0)ans+=m;
	}
    printf("%d",ans);
}
发布了88 篇原创文章 · 获赞 8 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_44879687/article/details/103115301