CF1029E Tree with Small Distances 题解

题意简述

给定一颗有根树(根节点为 1 1 1)。要求往树中加入一些边使得从根节点到其他节点的距离至多是 2 2 2。求加入边的最小数量。(边全部都是无向的)

解题思路

还是采用贪心的思路。

对于一个到根节点的距离大于 2 2 2 的叶子节点,我们只有两种方法使它到根节点的距离小于等于 2 2 2,要么直接连它和根节点,要么连接它的父亲和根节点。比较显然的是,如果连接它的父亲和根节点,一定不劣于连接它和根节点。

然后接下来这一步和其他题解说的都不太一样。大部分题解在得出贪心的策略后是把所有点按深度排序,然后开始贪心。但是根据我们上面推出的东西,我们只需要保证每次处理的是叶子节点就可以,这启示我们使用拓扑排序。

具体地,先 dfs 一遍求出每个点的父亲,然后进行拓扑排序,将拓扑排序的结果记录下来,进行贪心即可。

由于不需要排序,所以时间复杂度其实是优于其他题解所述的贪心。

代码示例

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
#define int long long
int n,vis[200010],ans=0,du[200010],fa[200010],num[200010],cnt=0;
vector<int> G[200010];
void dfs(int u,int Fa){
	fa[u]=Fa;
	for(int v:G[u]) if(v!=Fa) dfs(v,u);
}
signed main(){
	cin>>n;
	for(int i=1;i<=n-1;i++){
		int u,v;
		cin>>u>>v;
		G[u].push_back(v);
		G[v].push_back(u);
		du[u]++,du[v]++;
	}
	dfs(1,0);//找父亲
	vis[1]=1;
	for(int u:G[1]){
		vis[u]=1;
		for(int v:G[u]) vis[v]=1;
	}
    //把根节点周围先标记了
	queue<int> q;
	for(int i=1;i<=n;i++) if(!vis[i]&&du[i]==1) q.push(i);
	while(!q.empty()){
		int u=q.front();q.pop();
		num[++cnt]=u;
		for(int v:G[u]){
			du[v]--;
			if(du[v]==1) q.push(v);
		}
	}
    //拓扑
	for(int i=1;i<=cnt;i++){
		int u=num[i];
		if(vis[u]) continue;
		vis[fa[u]]=1;
		ans++;
		for(int v:G[fa[u]]) vis[v]=1;
	}
    //贪心
	cout<<ans<<endl;
	return 0;
}
相关推荐
在所不辞兄3 分钟前
常微分方程详解与应用
人工智能·神经网络·算法·机器学习
-dzk-4 分钟前
【动态规划】LC 416.分割等和子集
算法·动态规划·代理模式
夏幻灵24 分钟前
每日一题:LeetCode 1:两数之和——从暴力枚举到 Map 哈希表优化
算法·leetcode·散列表
段一凡-华北理工大学33 分钟前
高炉炼铁机器视觉与智能识别十八讲~系列文章09:AI 算法基础:从传统图像处理到深度学习的视觉“大脑“
图像处理·人工智能·算法·机器视觉·工业智能化·高炉炼铁智能化·高炉智能识别
alphaTao40 分钟前
LeetCode 每日一题 2026/9/28-2026/10/4
算法·leetcode
源码学社43 分钟前
图像去水印实测:OpenCV inpainting vs LaMa,传统算法和 AI 模型差距有多大
人工智能·opencv·算法·去水印·图片水印·ai去水印
一条大祥脚1 小时前
26ICPC网络赛第二场 计数|DFS|MEX|贪心
算法·深度优先·mex·icpc网络赛
aqiu1111111 小时前
【蓝桥杯 / C++】星际争霸 — 字符串逆序对与大整数自定义排序
算法
行者全栈架构师1 小时前
【鸿蒙心迹】ArkUI 列表性能实战——为什么 200 条数据页面掉到 20fps,LazyForEach 怎么救(HarmonyOS 7.x)
前端·算法·架构
一碗甜汤ᐝ1 小时前
高级算法分析与设计 | 常见 NP-Complete 问题总结
算法·np-complete·算法分析与设计·np完备问题