E - Subtree K-th Max

最新推荐文章于 2026-01-08 22:28:34 发布

原创最新推荐文章于 2026-01-08 22:28:34 发布 · 523 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度优先 #算法 #c++

好题精选同时被 2 个专栏收录

11 篇文章

订阅专栏

算法题

11 篇文章

订阅专栏

本文介绍了一段C++代码，展示了如何使用深度优先搜索（DFS）算法在给定图中查找路径。通过递归实现，代码用于解决从顶点0开始，找出其子节点中最优的前20个元素，并处理查询特定节点的第k个路径元素。适合学习图算法和数据结构的学生或开发者参考。

题目描述

题目链接

C++代码

#include<bits/stdc++.h>
#define pb push_back
using namespace std;

const int N = 1e5 + 10;
vector<int> e[N], ans[N]; 
vector<int> X(N);

void dfs(int u, int fa){
	vector<int> order;
	order.pb(X[u]);
	for(auto it : e[u]){
		if(it == fa) continue;
		dfs(it, u);
		for(auto j : ans[it]) order.pb(j);
	}
	
	sort(order.begin(), order.end(), greater<int>());
	int num = min(20, int(order.size()));
	for(int i = 0; i < num; i ++) ans[u].pb(order[i]);
	
	return ;
}

int main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	
	int n, tt;
	cin >> n >> tt;
	for(int i = 0; i < n; i ++) cin >> X[i];
	
	for(int i = 0; i < n - 1; i ++){
		int x, y;
		cin >> x >> y;
		x --;
		y --;
		e[x].pb(y);
		e[y].pb(x);
	}
	
	dfs(0, -1);
	
	while(tt --){
		int z, k;
		cin >> z >> k;
		z --;
		k --;
		cout << ans[z][k] << '\n';
	}
	
	return 0;
}