spoj Query on a tree2(LCA)

本文介绍了一种基于倍增思想的LCA算法实现方法,包括预处理阶段和查询阶段的具体步骤。通过实例演示了如何求解节点间的最近公共祖先问题,并提供了完整的C++代码示例。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

传送门
题解:
第一问:求出倍增LCA然后dis加减一下即可。
第二问:按照倍增思想往上跳,预判一下如果要跳过LCA就反着跳。
水题,TLE了两次因为倍增预处理不小心写进for循环了,对过样例不造成影响。。。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int MAXN=10004;
int n;
int head[MAXN],edge,dis[MAXN],dep[MAXN];
struct EDGE {
    int v,nxt,w;
}e[MAXN<<1];
int f[15][MAXN];
char ss[5];
inline int read() {
    int x=0,f=1;char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
    while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x*f;
}
inline void adde(int u,int v,int w) {
    e[edge].nxt=head[u],e[edge].v=v,e[edge].w=w,head[u]=edge++;
    e[edge].nxt=head[v],e[edge].v=u,e[edge].w=w,head[v]=edge++;
}
void dfs(int p,int fa) {
    for (int i=head[p];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (v^fa) {
            f[0][v]=p;
            dep[v]=dep[p]+1;
            dis[v]=dis[p]+e[i].w;
            dfs(v,p);
        }
    }
}
inline void da() {
    for (int j=1;(1<<j)<=n;++j)
        for (register int i=1;i<=n;++i)
            f[j][i]=f[j-1][f[j-1][i]];
}
inline int LCA(int x,int y) {
    if (dep[x]<dep[y]) x^=y^=x^=y;
    int t=dep[x]-dep[y];
    for (int i=0;i<=14;++i)
        if (t&(1<<i)) x=f[i][x];
    if (x==y) return x;
    for (int i=14;~i;--i)
        if (f[i][x]!=f[i][y]) x=f[i][x],y=f[i][y];
    return f[0][x];
}
inline int find(int k,int pos) {
    for (int i=0;i<=14;++i)
        if (k&(1<<i)) pos=f[i][pos];
    return pos;
}
int main() {
//  freopen("spoj qot2.in","r",stdin);
    int kase=read();
    while (kase--) {
        edge=0;
        memset(head,-1,sizeof(head));
        n=read();
        for (int i=1;i<n;++i) {
            int u=read(),v=read(),w=read();
            adde(u,v,w);
        }
        dis[1]=dep[1]=0,dfs(1,0);
        da();
        while (scanf("%s",ss)&&ss[1]^'O') {
            if (ss[0]=='D') {
                int x=read(),y=read(),lca=LCA(x,y);
                printf("%d\n",dis[x]+dis[y]-(dis[lca]<<1));
            }
            else {
                int x=read(),y=read(),k=read()-1,lca=LCA(x,y);
                if (dep[x]-dep[lca]>=k) printf("%d\n",find(k,x));
                else printf("%d\n",find(dep[x]+dep[y]-(dep[lca]<<1)-k,y));
            }
        }
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值