题目描述
有一个树状的城市网络(即 n 个城市由 n-1 条道路连接的连通图),首都为 1 号城市,每个城市售卖价值为 a_i 的珠宝。
你是一个珠宝商,现在安排有 q 次行程,每次行程为从 u 号城市前往 v 号城市(走最短路径),保证 v 在 u 前往首都的最短路径上。 在每次行程开始时,你手上有价值为 c 的珠宝(每次行程可能不同),并且每经过一个城市时(包括 u 和 v ),假如那个城市中售卖的珠宝比你现在手上的每一种珠宝都要优秀(价值更高,即严格大于),那么你就会选择购入。
现在你想要对每一次行程,求出会进行多少次购买事件。
你是一个珠宝商,现在安排有 q 次行程,每次行程为从 u 号城市前往 v 号城市(走最短路径),保证 v 在 u 前往首都的最短路径上。 在每次行程开始时,你手上有价值为 c 的珠宝(每次行程可能不同),并且每经过一个城市时(包括 u 和 v ),假如那个城市中售卖的珠宝比你现在手上的每一种珠宝都要优秀(价值更高,即严格大于),那么你就会选择购入。
现在你想要对每一次行程,求出会进行多少次购买事件。
输入描述:
第一行,两个正整数 n , q (2 ≤ n ≤ 10^5 , 1 ≤ q ≤ 10^5)。 第二行,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 ≤ 10^5)。
输出描述:
对于每次行程输出一行,为所购买次数。
示例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
题解:总体的思路就是倍增,同时为了方便做了一些转化,把询问的路径转化一下,新建一个起点接在起点的后面,这个点的价格是询问的C,然后DFS,同时求出倍增的数组,对于每个询问,答案就是这个询问新建的点倍增向上,走到询问的终点的距离
代码:
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#define N 200005
using namespace std;
typedef long long ll;
vector<int>v[N];
int a[N],b[N][20],d[N],to[N];
void dfs(int p,int fa)
{
int x=fa;
for(int i=19;i>=0;i--)
if(a[b[x][i]]<=a[p]&&b[x][i])x=b[x][i];
if(a[x]>a[p])b[p][0]=x;
else b[p][0]=b[x][0];
for(int i=1;i<20;i++)
b[p][i]=b[b[p][i-1]][i-1];
d[p]=d[fa]+1;
int num=v[p].size();
for(int i=0;i<num;i++){
int to=v[p][i];
if(to==fa)continue;
dfs(to,p);
}
}
int main()
{
int n,q,x,y,c;
while(~scanf("%d%d",&n,&q))
{
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=1;i<=n+q;i++)
v[i].clear();
for(int i=1;i<n;i++){
scanf("%d%d",&x,&y);
v[x].push_back(y);
v[y].push_back(x);
}
for(int i=n+1;i<=n+q;i++){
scanf("%d%d%d",&x,&y,&c);
v[i].push_back(x);
v[x].push_back(i);
a[i]=c;
to[i-n]=y;
}
dfs(1,0);
for(int i=1;i<=q;i++)
{
int ans=0;
x=i+n;
for(int j=19;j>=0;j--)
if(d[b[x][j]]>=d[to[i]]){
ans|=(1<<j);
x=b[x][j];
}
printf("%d\n",ans);
}
}
return 0;
}