倍增求lca 和 树上任意两点之间的最值

代码:

void process()
{
     for(int i=0;i<n;i++){
        anc[i][0]=fa[i]; maxcost[i][0]=cost[i];
        for(int j=1;(1<<j)<n;j++){
            anc[i][j]=-1;
        }
     }

     for(int j=1;(1<<j)<n;j++){
        for(int i=0;i<n;i++){
            if(anc[i][j-1]!=-1){
                int a=anc[i][j-1];
                anc[i][j]=anc[a][j-1];
                maxcost[i][j]=max(maxcost[i][j-1],maxcost[a][j-1]);
            }
        }
     }
}

int query(int p,int q)
{
    //printf("%d %d\n",p,q);
    int tmp,log,i;
    if(L[p]<L[q]) swap(p,q);
    for(log=1;(1<<log)<=L[p];log++);

    log--;
    int ans=-inf;

    for(int i=log;i>=0;i--){
        if(L[p]-(1<<i)>=L[q]){
            ans=max(ans,maxcost[p][i]);
            p=anc[p][i];
        }
    }

    if(p==q) return ans;
    for(int i=log;i>=0;i--){
        if(anc[p][i]!=-1&&anc[p][i]!=anc[q][i]){
            ans=max(ans,maxcost[p][i]); p=anc[p][i];
            ans=max(ans,maxcost[q][i]); q=anc[q][i];
        }
    }

    ans=max(ans,cost[p]);
    ans=max(ans,cost[q]);
    return ans;
}

猜你喜欢

转载自blog.csdn.net/yjt9299/article/details/82935250