树上二分(树的重心)

题目来源

Problem - G - Codeforces

题目介绍

给定log2n次机会,去猜出树上的一个特殊的点。

解题思路

由给定的机会,很容易看出这是一个在树上二分题目,所以我们可以利用树的重心来保证每次可以去除树上一半的点,最后即可求出答案。

解题代码

cpp 复制代码
void dfs(ll u, ll v) {
	siz[u] = 1;
	maxtree[u] = 0;
	for (auto it : edge[u]) {
		if (it == v || flag[it]) continue;
		dfs(it, u);
		siz[u] += siz[it];
		maxtree[u] = max(siz[it], maxtree[u]);
	}
	maxtree[u] = max(maxtree[u], S - siz[u]);
	if (maxtree[u] < maxtree[root]) root = u;
}

这段代码通过dfs求出siz数组和maxtree数组,siz数组是树上点的子树大小,maxtree数组是去除该点的后剩余子树之中节点个数最多的数量

cpp 复制代码
ll check(ll x, ll y) {
	cout << "? " << x << " " << y << "\n";
	ll pp; cin >> pp;
	return pp;
}

通过交互来获取下一步怎么做

cpp 复制代码
ll dfs1(ll x) {
	dfs(x, 0);
	vector<ll> son;
	for (auto it : edge[x]) {
		if (!flag[it]) son.push_back(it);
	}
	sort(son.begin(), son.end(), [&](auto x, auto y) {return siz[x] > siz[y]; });
	if (son.size() == 0) return x;
	else if (son.size() == 1) {
		ll pp = check(x, son[0]);
		if (pp == 0) return x;
		else return son[0];
	}
	else if (son.size() == 2) {
		ll pp = check(son[0], son[1]);
		if (pp == 1) return x;
		if (pp == 0) {
			S = siz[son[0]];
			root = 0;
			flag[x] = 1;
			dfs(son[0], 0);
			return dfs1(root);
		}
		else {
			S = siz[son[1]];
			root = 0;
			flag[x] = 1;
			dfs(son[1], 0);
			return dfs1(root);
		}
	}
	else {
		ll pp = check(son[0], son[1]);
		if (pp == 1) {
			S = siz[son[2]] + 1;
			root = 0;
			flag[son[0]] = 1;
			flag[son[1]] = 1;
			dfs(x, 0);
			return dfs1(root);
		}
		else if (pp == 0) {
			S = siz[son[0]];
			root = 0;
			flag[x] = 1;
			dfs(son[0], 0);
			return dfs1(root);
		}
		else {
			S = siz[son[1]];
			root = 0;
			flag[x] = 1;
			dfs(son[1], 0);
			return dfs1(root);
		}
	}
}

我们可以对重心x这个点进行下面四个处理

1:x度数为0:直接返回答案x。

2:x度数为1:使用check()询问x和该点即可知道答案,然后返回该值就行。

二分3:x度数为2:使用check()询问x的两个相邻的点,求出之后的重心,再进行dfs1()。

4:x度数为3:使用check()询问x的三个相邻的点中较大的两个。最小的那个连 通块加一,因为要算上x本身,最后根据check()返回值求出之后的重心,再进行dfs1()。

最后求出该题的答案

总结

该题主要利用二分的思想和树的重心去解决这道题目,总体不难,但是其中分类去跑下一次dfs1()的细节比较多。

相关推荐
lqqjuly17 小时前
C++ 完整知识体系—从基础语法到现代 C++23 的系统性总结
c++·c++23
CS创新实验室18 小时前
从顺序表到动态数组:数据结构的永恒基石与现代语言的优雅封装
数据结构·算法
王老师青少年编程18 小时前
信奥赛C++提高组csp-s之FHQ Treap
c++·csp·平衡树·信奥赛·csp-s·提高组·fhq treap
Black蜡笔小新18 小时前
自动化AI算法训练服务器DLTM训推一体化平台助力农业生产管理实现安全智能化
人工智能·算法·自动化
8Qi819 小时前
LeetCode 23. 合并 K 个升序链表 —— 小顶堆(PriorityQueue)
数据结构·算法·leetcode·链表·
QiLinkOS19 小时前
《打破“用爱发电”:一种基于 Gitee 与时间戳的开源权益分配机制探索》
c语言·数据结构·c++·科技·算法·gitee·开源
Irissgwe20 小时前
c++STL--string类
c++·stl·string
松间听晚20 小时前
Agentic RL 环境和代码学习:以HGPO为例
算法
Irissgwe20 小时前
c++类型转换
c++·类型转换·explicit·static_cast·const_cast·dynamic_cast·rtti
智者知已应修善业20 小时前
【51单片机用T0定时器方式1,实现0.5S的时间间隔实现第一次一个灯亮、第二次二个灯亮,直到全部灯亮,然后重复整个过程】2023-12-29
c++·经验分享·笔记·算法·51单片机