「美团 CodeM 复赛」城市网络

Posted 日拱一卒 功不唐捐

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了「美团 CodeM 复赛」城市网络相关的知识,希望对你有一定的参考价值。

https://loj.ac/problem/6192

内存限制:64 MiB   时间限制:500 ms标准输入输出

题目描述

有一个树状的城市网络(即 nnn 个城市由 n−1n-1n1 条道路连接的连通图),首都为 111 号城市,每个城市售卖价值为 aia_ia?i?? 的珠宝。

你是一个珠宝商,现在安排有 qqq 次行程,每次行程为从 uuu 号城市前往 vvv 号城市(走最短路径),保证 vvv 在 uuu 前往首都的最短路径上。

在每次行程开始时,你手上有价值为 ccc 的珠宝(每次行程可能不同),并且每经过一个城市时(包括 uuu 和 vvv),假如那个城市中售卖的珠宝比你现在手上的每一种珠宝都要优秀(价值更高,即严格大于),那么你就会选择购入。

现在你想要对每一次行程,求出会进行多少次购买事件。

输入格式

第一行,两个正整数 n,qn , qn,q。

第二行,nnn 个正整数 aia_ia?i?? 描述每个城市售卖的珠宝的价值。

接下来 n−1n-1n1 行,每行描述一条道路 x,yx , yx,y (1≤x,y≤n1 \leq x , y \leq n1x,yn),表示有一条连接 xxx 和 yyy 的道路。

接下来 qqq 行,每行描述一次行程 u,v,cu , v , cu,v,c (1≤u,v≤n1 \leq u , v \leq n1u,vn)。

输出格式

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

样例

样例输入

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

数据范围与提示

对于 100%100 \%100% 的数据,保证 2≤n≤105,1≤q≤1052 \leq n \leq 10^5 , 1 \leq q \leq 10^52n10?5??,1q10?5?? , 1≤ai≤1051 \leq a_i \leq 10^51a?i??10?5?? , 1≤c≤1051 \leq c \leq 10^51c10?5??。

 

将询问看作节点,c为点权,作为起点的孩子

倍增,找到第一个大于询问的点,以这个点为基础继续倍增

#include<cmath>
#include<cstdio>
#define N 200001
using namespace std;
int p,fa[N<<1][20];
int front[N<<1],nxt[N<<1],to[N<<1],tot,val[N<<1];
int dia[N<<1],dep[N<<1],decc[N];
void add(int u,int v)
{
        to[++tot]=v; nxt[tot]=front[u]; front[u]=tot; 
        to[++tot]=u; nxt[tot]=front[v]; front[v]=tot;
}
void dfs(int x,int y)
{int pos=y;
        for(int i=p;i>=0;i--) 
                if(fa[pos][i] && dia[fa[pos][i]]<=dia[x]) pos=fa[pos][i]; 
        if(dia[pos]>dia[x]) fa[x][0]=pos;
        else fa[x][0]=fa[pos][0];
        for(int i=1;fa[fa[x][i-1]][i-1];i++) fa[x][i]=fa[fa[x][i-1]][i-1];
        dep[x]=dep[y]+1;
        for(int i=front[x];i;i=nxt[i])
                if(to[i]!=y) dfs(to[i],x);
}                
int main()
{
        int n,q;
        scanf("%d%d",&n,&q);
        for(int i=1;i<=n;i++) scanf("%d",&dia[i]);
        int u,v;
        for(int i=1;i<n;i++)
        {
                scanf("%d%d",&u,&v);
                add(u,v);
        }
        int c;
        for(int i=1;i<=q;i++)
        {
            scanf("%d%d%d",&u,&v,&c);
            add(n+i,u);
            dia[n+i]=c;
            decc[i]=v;
        }
        p=log(n<<1)/log(2)+1;
        dfs(1,0);
        int ans,pos,t;
        for(int i=n+1;i<=n+q;i++)
        {
                ans=0;
                pos=i;
                t=decc[i-n];
                for(int j=p;j>=0;j--)
                        if(dep[fa[pos][j]]>=dep[t])
                                ans+=1<<j,pos=fa[pos][j];
                printf("%d\n",ans);
        }
}

 

 

以上是关于「美团 CodeM 复赛」城市网络的主要内容,如果未能解决你的问题,请参考以下文章

loj6191「美团 CodeM 复赛」配对游戏 概率期望dp

美团CodeM资格赛第二题

CodeM美团点评编程大赛初赛A轮

美团点评CodeM编程大赛;fastjson 1.2.33 发布;“闪电立方”一次可传100T数据

组委会正在为美团点评CodeM大赛的决赛设计新赛制

CodeM资格赛2