HDU3078 Network

LCA问题，

先输入n,和q;表示节点的个数和操作的个数

然后一行n个数，表示这n个节点的权值

然后n-1行，每行两个数，表示节点u与节点v相连

最后q行，表示操作 每行三个数，k,u,v;

当k=0时，将u点的权值改为v

当k>0时，输出从u->v路上第K大的权值

首先求出节点u和节点v的最近公共祖先lc,然后将节点u到lc的点的权值记录到p数组中，再将v到lc的权值记录到p数组中,(祖先节点lc被记录了两次，需要删除一次)，如果数组p的长度小于k则输出"invalid request!"，否则就将p数组从大到小排序，输出p[k-1]

#include<stdio.h>#include<iostream>#include<algorithm>#include<string.h>#include<map>#include<math.h>#include<queue>#include<vector>using namespace std;#define nn 80100#define inf 0x7fffffff#define ll long longvector<int>e[nn];int vis[nn],first[nn],node[nn<<1],dep[nn<<1],fa[nn];int p[nn<<1],dp[nn<<1][20],val[nn];int _pow[25];//node记录DFS序//first记录每个点第一次在DFS序中出现的位置//dep记录DFS序中每个点的深度//fa记录每个几点的父亲节点//val记录每个位置的权重//dp[i][j]表示在i到i+(1<<j)这个区间内dep值最小的下标void dfs(int &index,int u,int d,int par){    index++;    vis[u]=1;    first[u]=index;    node[index]=u;    dep[index]=d;    fa[u]=par;    for(int i=0;i<e[u].size();i++)    {        if(!vis[e[u][i]])        {            dfs(index,e[u][i],d+1,u);            index++;            node[index]=u;            dep[index]=d;        }    }}void rmq_init(int n){    //int m=(int)(log((double)(n*1.0))/long(2.0));    int m=0;    while(_pow[m+1]<=n) m++;    //printf("m==%d\n",m);    for(int i=1;i<=n;i++)        dp[i][0]=i;    for(int j=1;j<=m;j++)    {        for(int i=1;(i+_pow[j]-1)<=n;i++)        {            int a=dp[i][j-1];            int b=dp[i+_pow[j-1]][j-1];            dp[i][j] = dep[a]<dep[b] ? a:b;        }    }}int rmq(int x,int y){    //int m=(int)(log((double)(y-x+1))/long(2.0));    int m=0;    while(_pow[m+1]<=(y-x+1)) m++;    //printf("m===%d\n",m);    int a=dp[x][m];    int b=dp[y-_pow[m]+1][m];    return dep[a]<dep[b]?a:b;}int lca(int u,int v){    int x=first[u];    int y=first[v];    if(x>y) swap(x,y);    int index=rmq(x,y);    //printf("index==%d %d %d\n",x,y,index);    return node[index];}void path(int &index,int s,int t){    while(s!=t)    {        p[index++]=val[s];        s=fa[s];    }    p[index++]=val[t];}bool cmp(int x,int y){    return x>y;}void solve(int k,int u,int v){    int lc=lca(u,v);    //printf("%d %d %d\n",u,v,lc);    int t=0;    path(t,u,lc);    path(t,v,lc);    t--;//公共祖先被记录两次    if(k>t)    {        printf("invalid request!\n");        return;    }    sort(p,p+t,cmp);    printf("%d\n",p[k-1]);}int main(){    for(int i=0;i<20;i++) _pow[i]=1<<i;    int n,q;    scanf("%d%d",&n,&q);    for(int i=1;i<=n;i++) scanf("%d",&val[i]);    for(int i=1;i<n;i++)    {        int u,v;        scanf("%d%d",&u,&v);        e[u].push_back(v);        e[v].push_back(u);    }    int tot=0;    memset(vis,0,sizeof(vis));    dfs(tot,1,1,-1);//分别表示dfs的长度，跑到的节点，深度，该点的父亲节点(-1表示没有父亲节点)//    for(int i=1;i<=tot;i++) printf("%d ",node[i]);//    cout<<endl;//    for(int i=1;i<=tot;i++) printf("%d ",dep[i]);//    cout<<endl;//    for(int i=1;i<=n;i++) printf("%d ",first[i]);//    cout<<endl;//    for(int i=1;i<=n;i++) printf("%d ",fa[i]);//    cout<<endl;//    printf("tot==%d\n",tot);    rmq_init(tot);    while(q--)    {        int op,u,v;        scanf("%d%d%d",&op,&u,&v);        if(op==0) val[u]=v;        else solve(op,u,v);    }    return 0;}