砍树 蓝桥杯 暴力 dfs 树上差分 最近公共祖先 lca(LCA)

本文介绍了两种方法,一种是暴力枚举并动态更新边权,另一种是利用DFS和树上差分优化求解给定无向图中,从起点到终点的路径,使得边权之和等于给定值的问题。
摘要由CSDN通过智能技术生成

暴力:

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N = 1e5 + 10;

typedef pair<int,int> pii;//存图

vector<int>edge[N];

int n, m;

int w[N];//存每一个边的边权。

map<pii, int>id;//存边的编号

//s是路径的起点,v是路径的终点,u表示你当前走到了哪个点。
bool dfs(int s, int u, int father, int v)
{
    if(u == v)
    {
        return true;
    }

    //遍历邻接点(子节点)
    for(int i = 0; i < edge[u].size(); i ++)
    {
        int son = edge[u][i];

        if(son == father)//防止死循环
            continue;

        if(dfs(s, son, u, v))
        {
            int ID = id[{u, son}];
            w[ID] ++;//增加边权
            return true;
        }
    }

    return false;
}

void solve()
{
    cin >> n >> m;
    for(int i = 0; i < n - 1; i ++)
    {
        int x, y; cin >> x >> y;

        edge[x].push_back(y);//无向图存两次
        edge[y].push_back(x);

        id[{x, y}] = id[{y, x}] = i;//编号也存一下
    }

    for(int i = 0; i < m; i ++)
    {
        int x, y; cin >> x >> y;
        dfs(x, x, -1, y);
    }

    int ans = -1;
    for(int i = n - 1; i >= 0; i --)//倒序枚举一定是最大的
    {
        if(w[i] == m)
        {
            ans = i + 1;//我们是从0开始的题目从一开始
            break;
        }
    }

    cout << ans << endl;
}

signed main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    int t = 1;
    // cin >> t;
    while(t--)
        solve();
}

优化:

//砍树:树上差分 + 最近公共祖先
#include<bits/stdc++.h>
using namespace std;
typedef pair<int, int> pii;
const int N = 1e5 + 10;
int n, m;
int w[N];//记录每个点的点权。

map<pii,int>id;
vector<int>edge[N];

int siz[N], dep[N], fa[N], son[N], top[N];

void dfs1(int u, int father)
{
	siz[u] = 1, dep[u] = dep[father] + 1;
	fa[u] = father;
	for(int i = 0; i < edge[u].size(); i ++)
	{
		int s = edge[u][i];
		if(s == fa[u])
			continue;
		dfs1(s, u);
		siz[u] += siz[s];
		if(siz[son[u]] < siz[s])
			son[u] = s;
	}
}

void dfs2(int u, int t)
{
	top[u] = t;
	if(son[u] == 0)
		return;
	dfs2(son[u], t);
	for(int i = 0; i < edge[u].size(); i ++)
	{
		int s = edge[u][i];
		if(s == fa[u] || s == son[u])
			continue;
		dfs2(s, s);
	}
}

int lca(int x, int y)
{
	while(top[x] != top[y])
	{
		if(dep[top[x]] < dep[top[y]])
			swap(x, y);

		x = fa[top[x]];
	}
	return dep[x] < dep[y] ? x : y;
}

void cal_sum(int u, int father)
{
	for(int i = 0; i < edge[u].size(); i ++)
	{
		int son = edge[u][i];
		if(son == father)
			continue;
		cal_sum(son, u);
		w[u] += w[son];
	}
}

void solve()
{
	 cin >> n >> m;
	 for(int i = 1; i <= n - 1; i ++)
	 {
	 	int x, y; cin >> x >> y;
	 	edge[x].push_back(y);
	 	edge[y].push_back(x);
	 	id[{x, y}] = i;
	 	id[{y, x}] = i;
	 }	

	 dfs1(1, 0);
	 dfs2(1, 1);

	 for(int i = 0; i < m; i ++)
	 {
	 	int a, b; cin >> a >> b;

	 	//做树上差分
	 	w[a] ++, w[b] ++;
	 	w[lca(a, b)] -= 2;
	 }


	 cal_sum(1, 0);


	 int ans = -1;
	 for(int i = 1; i <= n; i ++)
	 {
	 	if(w[i] == m)
	 	{
	 		int ID = id[{i, fa[i]}];
	 		ans = max(ans, ID);
	 	}
	 }
	 cout << ans << endl;
}

signed main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	int t = 1;
	// cin >> t;
	while(t--)
	solve();
}

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值