Ural 1471. Tree

最新推荐文章于 2024-10-10 20:14:35 发布

double1994

最新推荐文章于 2024-10-10 20:14:35 发布

阅读量556

点赞数

文章标签：数据结构

本文链接：https://blog.csdn.net/u011439362/article/details/13504829

版权

LCA问题，这篇文章讲得不错：

http://community.topcoder.com/tc?module=Static&d1=tutorials&d2=lowestCommonAncestor

注意事项在代码中

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 110000
#define MAX 55000
int T[MAX];
int P[MAX][50];
int L[MAX];
int W[MAX];
int R[MAX][50];
struct Node
{
    int v;
    int val;
    Node * next;
}*head[N],tree[N];
int ptr;
int n;
int sum;
void add(int x,int y,int w)
{
    tree[ptr].val=w;
    tree[ptr].v=y;
    tree[ptr].next=head[x];
    head[x]=&tree[ptr++];
    tree[ptr].val=w;
    tree[ptr].v=x;
    tree[ptr].next=head[y];
    head[y]=&tree[ptr++];
}
void dfs(int rt,int par,int lv)
{
    Node *p=head[rt];
    while(p!=NULL)
    {
        if(p->v!=par)
        {
            L[p->v]=lv+1;
            T[p->v]=rt;
            W[p->v]=p->val;
            dfs(p->v,rt,lv+1);
        }
        p=p->next;
    }
}
void process()
{
    for(int i=0;i<n;i++)
        for(int j=0;1<<j<n;j++)
        {
            P[i][j]=-1;
            R[i][j]=0;
        }
    for(int i=0;i<n;i++)
    {
        P[i][0]=T[i];
        R[i][0]=W[i];
    }
    for(int j=1;1<<j<n;j++)
        for(int i=0;i<n;i++)
        if(P[i][j-1]!=-1)
        {
            P[i][j]=P[P[i][j-1]][j-1];
            R[i][j]=R[i][j-1]+R[P[i][j-1]][j-1];//是R[P[i][j-1][j-1]不是R[R[i][j-1]][j-1]
        }
}
int query(int x,int y)
{
    int sum=0;
    int tmp,log;
    if(L[x]<L[y])
        tmp=x,x=y,y=tmp;
    for(log=1;1<<log<=L[x];log++);
    log--;
    for(int i=log;i>=0;i--)
        if(L[x]-(1<<i)>=L[y])
        {
            sum+=R[x][i];
            x=P[x][i];
        }
    if(x==y)return sum;
    for(int i=log;i>=0;i--)
    {
        if(P[x][i]!=-1&&P[x][i]!=P[y][i])//不能相等，否则可能不是最近的公共祖先
        {
            sum+=R[x][i]+R[y][i];
            x=P[x][i];
            y=P[y][i];
        }
    }
    sum+=W[x]+W[y];//记得加上
    return sum;
}
int main()
{
    scanf("%d",&n);
    int x,y,w;
    ptr=1;
    memset(head,NULL,sizeof(head));
    for(int i=1;i<n;i++)
    {
        scanf("%d%d%d",&x,&y,&w);
        add(x,y,w);
    }
    dfs(0,0,1);
    L[0]=1;
    process();
    int m;
    scanf("%d",&m);
    int g;
    for(int i=0;i<m;i++)
    {
        scanf("%d%d",&x,&y);
        printf("%d\n",query(x,y));
    }
    return 0;
}