#F. 啸聚山林

题目

代码

#include<bits/stdc++.h>
using namespace std;
int n,q,a[100001],u,m;
vector<int> b[100001];    //用来存储以x[i]为根的子树中的所有头领
vector<int> adj[100001];
bool cmp(int x,int y)     //排序用的cmp
{
	return x>y;
}
void dfs(int s,int d)  //s是当前的结点，d是s的根
{
	b[s].push_back(a[s]);  //以x[i]为根的子树中也有它自己
	for(int i=0;i<adj[s].size();i++)
	{
		int t=adj[s][i];
		if(adj[s][i]==d)continue; //如果与s相连的结点有它的父节点就不要算它了
		dfs(adj[s][i],s); 
		for(int j=0;j<b[t].size();j++)  //也要将结点s的子树的所有结点的都加上
		{
			b[s].push_back(b[t][j]);
		}
	}
	sort(b[s].begin(),b[s].end(),cmp);  //从大到小排序 
	while(b[s].size()>20)b[s].pop_back();  //不加这个可能会超内存，因为题目说了1<=yi<=20，所以大于20的结点都不要了
}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)cin>>a[i];  //输入
	for(int i=1;i<n;i++)
	{
		int c,d;
		cin>>c>>d;
		adj[c].push_back(d);  //双向建边 ，是无向图
		adj[d].push_back(c);
	}
	dfs(1,-1);
	while(m--)
	{
		int c,d;
		cin>>c>>d;
		cout<<b[c][d-1]<<endl;  //因为vector的起始位置是从0开始的，所以是d-1
	}
	return 0;
}