hdoj3078 (暴力lca)

题目大意:n结点树 ,两种操作,将结点a的值改为b,或询问a到b的路径间第k大点的值(包括a,b),操作总数为m,

                  n<80000, m<30000

分析:逐个往上记录,排序求第k大,又是暴力过的。。额。惊了

#include<iostream>
#include<stdio.h>
#include<vector>
#include<algorithm>
#include<math.h>
using namespace std;
vector<int>g[80010];
int h[80010],fa[80010],A[80010],B[80010];
void dfs(int u){
     int i,v;
     for(i=0;i<g[u].size();i++){
          v=g[u][i];
          if(v!=fa[u]){
              h[v]=h[u]+1;
              fa[v]=u;
              dfs(v);
          }
     }
}
int cmp(int a,int b){
    return a>b;
}
int main(){
    int k,a,b,i,n,m;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++) scanf("%d",&A[i]);
    for(i=1;i<n;i++){
        scanf("%d%d",&a,&b);
        g[a].push_back(b);
        g[b].push_back(a);
    }
    fa[1]=h[1]=0;
    dfs(1);
    while(m--){
        scanf("%d%d%d",&k,&a,&b);
        if(k==0) A[a]=b;
        else{
            int sum=0;
            if(h[a]<h[b]) swap(a,b);
            while(h[a]!=h[b]){
                B[sum++]=A[a];
                a=fa[a];
            }
            while(a!=b){
                B[sum++]=A[a];
                B[sum++]=A[b];
                a=fa[a];
                b=fa[b];
            }
            B[sum++]=A[a];
            sort(B,B+sum,cmp);
            if(k>sum) printf("invalid request!\n");
            else printf("%d\n",B[k-1]);
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/guogai13/article/details/81256381
lca