牛客网

时间限制:C/C++ 2秒,其他语言4秒 空间限制:C/C++ 262144K,其他语言524288K 64bit IO Format:
%lld

题目描述 有一个树状的城市网络(即 n 个城市由 n-1 条道路连接的连通图),首都为 1 号城市,每个城市售卖价值为 a_i 的珠宝。
你是一个珠宝商,现在安排有 q 次行程,每次行程为从 u 号城市前往 v 号城市(走最短路径),保证 v 在 u 前往首都的最短路径上。
在每次行程开始时,你手上有价值为 c 的珠宝(每次行程可能不同),并且每经过一个城市时(包括 u 和 v
),假如那个城市中售卖的珠宝比你现在手上的每一种珠宝都要优秀(价值更高,即严格大于),那么你就会选择购入。
现在你想要对每一次行程,求出会进行多少次购买事件。

输入描述:

第一行,两个正整数 n , q (2 ≤ n ≤ 105 , 1 ≤ q ≤ 105)。 第二行,n 个正整数 a_i (1 ≤
a_i ≤ 10^5) 描述每个城市售卖的珠宝的价值。 接下来 n-1 行,每行描述一条道路 x , y (1 ≤ x,y ≤
n),表示有一条连接 x 和 y 的道路。 接下来 q 行,每行描述一次行程 u , v , c (1 ≤ u,v ≤ n , 1 ≤ c
≤ 105)。

输出描述:

对于每次行程输出一行,为所购买次数。

示例1
输入

5 4
3 5 1 2 4
1 2
1 3
2 4
3 5
4 2 1
4 2 2
4 2 3
5 1 5

输出

2
1
1
0

题解:
毋庸置疑就是用倍增来做
那什么是倍增呢?
(先挖个坑,有空做个倍增的讲解)
递推式fa[i][j] = fa[fa[i][j-1]][j - 1],只要倍增得到fa[i][0]就行,因为有了这个后面都可以推出
fa[i][j] 代表i节点往上走2^j的距离,且比当前大的点
每次查询时,在所有需要问的点加一条新点,连在u的下方,新点的权值就是询问的初始权值,从这个新点往上倍增就可以了

#include<bits/stdc++.h>
#include<vector>
const int maxn=2e5+2;
using namespace std;
int a[maxn];
int dis[maxn];
int fa[maxn][23];
int n,q;
int to[maxn];
vector<int> W[2*maxn];
void dfs(int u,int f)
{
   
	int pos=f;

	for(int i=21;i>=0;i--)
	{
   
		if(fa[pos][i]&&a[fa[pos][i]]<=a[u])
		pos=fa[pos][i];
	}
	if(a[pos]>a[u])fa[u][0]=f;
	else fa[u][0]=fa[pos][0];
	for(int i=1;fa[fa[u][i-1]][i-1];i++)
	{
   
		fa[u][i]=fa[fa[u][i-1]][i-1];
	}
		dis[u]=dis[f]+1;
	for(int v=0;v<W[u].size();v++)
	
	{
   
		if(W[u][v]==f)continue;
		dfs(W[u][v],u);
	}
}
int main()
{
   
	
	cin>>n>>q;
 
    for(int i = 1; i <= n; ++i) 
	{
   
		scanf("%d",&a[i]);
 
	}
    for(int i = 1; i < n; ++i) {
   
        int a;
	  int b;
        scanf("%d%d",&a,&b);
        W[a].push_back(b);
        W[b].push_back(a);
    }
 
    for(int i =1; i <= q ; ++i){
   //增加新点
        int aa,b,c;
        scanf("%d%d%d",&aa,&b,&c);
        W[n+i].push_back(aa);
        W[aa].push_back(i+n);
        a[n+i] = c;
        to[n+i] = b;
    }
    dfs(1,0);
   int sum = 0; 
   int pos;
    for(int i = n+1; i <= n+q; ++i){
   
       pos=i;
      sum=0;
        for(int j = 21; j >= 0; --j){
   
            if(dis[fa[pos][j]] >= dis[to[i]]) {
   
                sum += (1 << j);
				pos= fa[pos][j];
            }
        }
        printf("%d\n",sum);
    }
    return 0;
 
}