打卡信奥刷题（3544）用C++实现信奥题 P11141 [APC001] F - Extend

最新推荐文章于 2026-09-04 15:58:53 发布

原创最新推荐文章于 2026-09-04 15:58:53 发布 · 316 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

收录于

C++

P11141 [APC001] F - Extend

题目背景

~~本题输入输出量较大，请酌情使用较快的输入输出方式~~。已扩大时限从 $1s→1.5s1s\to 1.5s$ 。

题目描述

对于一棵有 $n$ 个节点，根为 $k$ 的树，最开始有且仅有一个“可扩展的”节点 $z$ ，我们有两种扩展形式：

$I\text{I}$ 类扩展：从一个“可扩展的”节点 $u$ 开始扩展，把 $u$ 的子树中所有节点和所有满足与 $u$ 距离 $≤p\leq p$ 且是 $u$ 祖先的节点标记为“可扩展的”。
$II\text{II}$ 类扩展：从一个“可扩展的”节点 $u$ 开始扩展，把所有与 $u$ 深度相等的节点标记为“可扩展的”。

你需要将所有结点都标记为“可扩展的”，求最小进行扩展的次数。

输入格式

第一行两个整数 $n, k$ 。

接下来 $n - 1$ 行，每行两个整数 $u, v$ ，表示存在一条无向边连接 $u, v$ 。

接下来一行一个整数 $t$ ，表示询问次数。

接下来 $t$ 行，每行两个整数 $z, p$ ，意义如上。

输出格式

$t$ 行，对于每次询问：

若无法将所有节点都标记为“可扩展的”，输出 -1。
否则一行一个整数，表示答案。

输入输出样例 #1

输入 #1

输出 #1

2
2

输入输出样例 #2

输入 #2

输出 #2

1
1
2

输入输出样例 #3

输入 #3

输出 #3

说明/提示

样例 $1$ 解释：两次询问都可以先对 $2$ 节点进行 $II\text{II}$ 类扩展，然后对 $3$ 进行 $I\text{I}$ 类扩展。可以保证对于两次询问，这样操作都是最优的。

对于所有数据， $1≤z,k,u,v≤n≤105,1≤t≤2×106,0≤p≤10181\leq z,k,u,v\leq n\leq 10^5,1\leq t\leq 2\times 10^6,0\leq p\leq 10^{18}$ 。

C++实现

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e5 + 7;
int n, k, cnt, dep[maxn];
vector<int> T[maxn];
void dfs(int u, int fa){ // 统计深度
	dep[u] = dep[fa] + 1;
	for(int v : T[u]){
		if(v == fa) continue;
		dfs(v, u);
	}
}
void dfs2(int u, int fa){ // 统计（说过的特殊情况）从根节点到首次分岔的这条链的长度
	int child = 0, nxt = -1;
	for(int v : T[u]){
		if(v == fa) continue;
		child ++; nxt = v;
	}
	if(child > 1 || nxt == -1) return ; // 到分岔了 或 到叶子了，返回
	cnt ++;
	dfs2(nxt, u);
}
void solve(){
	cin >> n >> k;
	for(int i = 1; i <= n - 1; i ++){
		int u, v; cin >> u >> v;
		T[u].push_back(v), T[v].push_back(u);
	}
	dep[0] = -1; // 根节点层为0，这样计算起来会方便一些
	dfs(k, 0);
	dfs2(k, 0);
	int T; cin >> T;
	while(T --){
		int z, p; cin >> z >> p;
		if(n == 1) cout << "0\n";
		else if(p == 0) cout << "-1\n";
		else if(z == k) cout << "1\n";
		else{
			int t = (dep[z] % p == 0? p : dep[z] % p); // 找倒数第二次时距离根结点的距离，注意特判整除
			if(t > cnt) cout << (dep[z] + p - 1) / p + 1 << '\n'; // ceil(dep[z]/p)+1
			else cout << (dep[z] + p - 1) / p << '\n'; // 否则能省一步
		}
	}
}

signed main(){
	solve();
	return 0;
}