美团点评2017年CodeM大赛-复赛 A 城市网络（倍增）

题目描述

有一个树状的城市网络（即 n 个城市由 n-1 条道路连接的连通图），首都为 1 号城市，每个城市售卖价值为 a_i 的珠宝。
你是一个珠宝商，现在安排有 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;
}