蓝桥杯2023年-砍树(dfs,树上差分)

题目描述

给定一棵由 n 个结点组成的树以及 m 个不重复的无序数对 (a1, b1), (a2, b2),

. . . , (am, bm),其中 ai 互不相同,bi 互不相同,ai ≠ bj(1 ≤ i, j ≤ m)。

小明想知道是否能够选择一条树上的边砍断,使得对于每个 (ai , bi) 满足 ai和 bi 不连通,如果可以则输出应该断掉的边的编号(编号按输入顺序从 1 开始),否则输出 -1.

思路

朴素做法:

时间复杂度:O(n²)

由于这是一颗树,所以任意两点之间的路径唯一。如果一条边被砍断后,所有点对都不相连,则意味着被砍的边在所有点对的路径上。

所以朴素做法就是给两点之间的路径边都加1,如果存在边权值为m(所有m个点对都经过该点)。则该边为一个可取边。

代码

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
const int N=100005;
#define int long long
vector<int>son[N];
map<pair<int,int>,int>mp;
int n,m;
int num[N];
bool dfs(int st,int ed,int u,int p){
    if(u==ed)return true;
    for(auto it:son[u]){
        if(it!=p&&dfs(st,ed,it,u)){
            num[mp[{u,it}]]++;
            return true;
        }
    }
    return false;
}
signed main(){
    cin>>n>>m;
    for(int i=0;i<n-1;i++){
        int a,b;cin>>a>>b;
        son[a].push_back(b);
        son[b].push_back(a);
        mp[{a,b}]=i+1;
        mp[{b,a}]=i+1;
    }
    for(int i=0;i<m;i++){
        int a,b;cin>>a>>b;
        dfs(a,b,a,0);
    }
    int ans=-1;
    for(int i=1;i<=n;i++){
        if(num[i]==m){
            ans=i;
        }
    }
    cout<<ans;
}

树上差分优化:

时间复杂度:O(nlogn)

优化的关键点在于如何快速计算路径的权值。转换到一维线性结构中,我们可以联想到借助差分来实现对区间(路径)的快速修改。

在树上,我们相应也可以使用树上差分来实现对路径的权值快速修改。

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
const int N=100005;
#define int long long
vector<int>son[N];
map<pair<int,int>,int>mp;
int n,m;
int num[N];
int f[N][20];
int dep[N];
void dfs(int u,int p){
    dep[u]=dep[p]+1;f[u][0]=p;
    for(int i=1;(1<<i)<=dep[u];i++){
        f[u][i]=f[f[u][i-1]][i-1];
    }
    for(auto it:son[u]){
        if(it!=p)
        dfs(it,u);
    }
}
int lca(int a,int b){
    if(dep[a]<dep[b])swap(a,b);
    for(int i=20;i>=0;i--){
        if(dep[f[a][i]]>dep[b])a=f[a][i];
        if(a==b)return a;
    }
    for(int i=20;i>=0;i--){
        if(f[a][i]!=f[b][i])a=f[a][i],b=f[b][i];
    }
    return f[a][0];
}
int res[N];
int dd(int u,int p){
    for(auto it:son[u]){
        if(it!=p){
            num[u]+=dd(it,u);
        }
    }
    res[mp[{p,u}]]=num[u];
    return num[u];
}
signed main(){
    cin>>n>>m;
    for(int i=0;i<n-1;i++){
        int a,b;cin>>a>>b;
        son[a].push_back(b);
        son[b].push_back(a);
        mp[{a,b}]=i+1;
        mp[{b,a}]=i+1;
    }
    for(int i=0;i<m;i++){
        int a,b;cin>>a>>b;
        num[a]++;num[b]++;
        num[lca(a,b)]-=2;
    }
    dd(1,0);
    int ans=-1;
    for(int i=1;i<=n;i++){
        if(res[i]==m){
            ans=i;
        }
    }
    cout<<ans;
}
相关推荐
一只鱼^_18 小时前
基础算法合集-图论
数据结构·算法·深度优先·图论·广度优先·宽度优先·图搜索算法
稳兽龙11 天前
P3258 [JLOI2014] 松鼠的新家
数据结构·c++·算法·深度优先·lca
minos.cpp11 天前
从厨房到代码台:用做菜思维理解iOS开发 - Swift入门篇①
ios·蓝桥杯·swift
岁忧11 天前
第十六届蓝桥杯C/C++程序设计研究生组国赛 国二
c语言·c++·算法·蓝桥杯
xwztdas11 天前
洛谷P3953 [NOIP 2017 提高组] 逛公园
算法·深度优先·动态规划
嗨信奥编程13 天前
2023年蓝桥杯青少第十四届蓝桥杯Scratch省赛中级组真题——小狗避障
青少年编程·蓝桥杯
小猫咪怎么会有坏心思呢13 天前
华为OD机考-小明减肥-DFS(JAVA 2025B卷)
java·华为od·深度优先
Yingye Zhu(HPXXZYY)13 天前
P12894 [蓝桥杯 2025 国 Java B] 智能交通信号灯
c++·蓝桥杯
可可格子衫15 天前
129. 求根节点到叶节点数字之和 --- DFS +回溯(js)
javascript·算法·深度优先
小猫咪怎么会有坏心思呢15 天前
华为OD机试-返回矩阵中非1的元素、个数/数值同化-BFS(JAVA 2025B卷)
java·华为od·深度优先