SPOJ QTREE2 Query on a tree II 倍增lca

题目:

https://vjudge.net/problem/SPOJ-QTREE2

题意:

给定一个树,有边权,有以下操作:

  • DIST a b a b的距离
  • KTH a b k 求从 a b路径上的第 k 个节点,a是第一个

DONE 作为查询结束标志

思路:

第一个操作,任意一个 lca 算法都可以求出来,考虑第二种操作,最简单的方法是一步一步往上找直到第 k 个,可以发现用倍增的思想往上找是log级别的,所以用了倍增 lca

#include <bits/stdc++.h>

using namespace std;

const int N = 10000 + 10, INF = 0x3f3f3f3f;

struct edge
{
    int to, cost, next;
}g[N*2];

int cnt, head[N];
int dep[N], dis[N], fat[N][20];

void init()
{
    cnt = 0;
    memset(head, -1, sizeof head);
}
void add_edge(int v, int u, int cost)
{
    g[cnt].to = u, g[cnt].cost = cost, g[cnt].next = head[v], head[v] = cnt++;
}
void dfs(int v, int fa, int d, int val)
{
    dep[v] = d, fat[v][0] = fa, dis[v] = val;
    for(int i = head[v]; ~i; i = g[i].next)
    {
        int u = g[i].to;
        if(u == fa) continue;
        dfs(u, v, d+1, val + g[i].cost);
    }
}
void lca_init(int n)
{
    for(int j = 1; (1<<j) <= n; j++)
        for(int i = 1; i <= n; i++)
            fat[i][j] = fat[fat[i][j-1]][j-1];
}
int lca(int v, int u)
{
    if(dep[v] < dep[u]) swap(v, u);
    int d = dep[v] - dep[u];
    for(int i = 0; (d>>i) != 0; i++)
        if((d>>i) & 1) v = fat[v][i];
    if(v == u) return v;
    for(int i = 18; i >= 0; i--)
        if(fat[v][i] != fat[u][i]) v = fat[v][i], u = fat[u][i];
    return fat[v][0];
}
int query_kth(int v, int u, int k)
{
    int t = lca(v, u);
    int len = dep[v] - dep[t] + 1;
    if(k <= len)
    {
        --k;
        for(int i = 0; (k>>i) != 0; i++)
            if((k>>i) & 1) v = fat[v][i];
        return v;
    }
    else
    {
        k = dep[u] - dep[t] - (k - len);
        for(int i = 0; (k>>i) != 0; i++)
            if((k>>i) & 1) u = fat[u][i];
        return u;
    }
}
int main()
{
    int t, n;
    scanf("%d", &t);
    while(t--)
    {
        init();
        scanf("%d", &n);
        int a, b, c;
        for(int i = 1; i <= n; i++)
        {
            scanf("%d%d%d", &a, &b, &c);
            add_edge(a, b, c); add_edge(b, a, c);
        }
        dfs(1, 0, 1, 0);
        lca_init(n);
        char opt[20];
        while(scanf("%s", opt), opt[1] != 'O')
        {
            scanf("%d%d", &a, &b);
            if(opt[0] == 'D')
            {
                int ans = lca(a, b);
                printf("%d\n", dis[a] - 2 * dis[ans] + dis[b]);
            }
            else
            {
                scanf("%d", &c);
                printf("%d\n", query_kth(a, b, c));
            }
        }
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值