题目描述

【传送门】

题目大意

给一棵树,有两种操作:

  • 求(u,v)路径的距离。
  • 求以u为起点,v为终点的第k的节点.

分析

比较简单的倍增LCA模板题。
首先对于第一问,我们只需要预处理出根节点到各个节点之间的距离,然后倍增LCA求解就可以了。
那么第二问我WA了6发,原来是眼瞎和手残打错了两个字符错掉了。
我们将问题分成3个部分:

  • LCA是第k个
  • 第k个在u到LCA的路径上
  • 第k个在LCA到v的路径上。

首先如果LCA是第k个,那么直接输出。
如果是第二种情况,那么从u开始做倍增,每一次k-(1<<i)就可以了。
小细节:只能将k变成1,模拟可证明。
第三种情况,那么我们的答案就是从v的第(dep[u]+dep[v]-2*dep[lca]+2),模拟可得该式。

ac代码

#include <bits/stdc++.h>
#define ll long long
#define ms(a, b) memset(a, b, sizeof(a))
#define inf 0x3f3f3f3f
#define ll long long
using namespace std;
template <typename T>
inline void read(T &x) {
    x = 0; T fl = 1;
    char ch = 0;
    while (ch < '0' || ch > '9') {
        if (ch == '-') fl = -1;
        ch = getchar();
    }
    while (ch >= '0' && ch <= '9') {
        x = (x << 1) + (x << 3) + (ch ^ 48);
        ch = getchar();
    }
    x *= fl;
}
#define N 20005
struct edge {
    int to, nt;
    ll w;
}E[N << 1];
ll dis[N];
int f[N][31], H[N], dep[N];
int cnt, n;
void add_edge(int u, int v, ll w) {
    E[++ cnt] = (edge){v, H[u], w};
    H[u] = cnt;
}
void dfs(int u, int fa, int dist) {
    f[u][0] = fa;
    dis[u] = dist;
    for (int i = 1; i <= 30; i ++)
        f[u][i] = f[f[u][i - 1]][i - 1];
    for (int e = H[u]; e; e = E[e].nt) {
        int v = E[e].to;
        if (v == fa) continue;
        dep[v] = dep[u] + 1;
        dfs(v, u, dist + E[e].w);
    }
}
int lca(int u, int v) {
    if (dep[u] < dep[v]) swap(u, v);
    for (int i = 30; i >= 0; i --)
        if (dep[v] <= dep[f[u][i]]) u = f[u][i];
    if (u == v) return u;
    for (int i = 30; i >= 0; i --)
        if (f[u][i] != f[v][i]) {
            u = f[u][i];
            v = f[v][i];
        }
    return f[u][0];
}
int solve(int u, int v, int k) {
    int Lca = lca(u, v);
    if (dep[u] - dep[Lca] + 1 == k) return Lca;
    else {
        if (dep[u] - dep[Lca] + 1 > k) {
            for (int i = 30; i >= 0; i --)
                if (k - 1 >= (1 << i)) k -= (1 << i), u = f[u][i];
            return u;
        }
        else {
            k = dep[u] + dep[v]- dep[Lca] * 2 - k + 2;
            for (int i = 30; i >= 0; i --)
                if (k - 1 >= (1 << i)) k -= (1 << i), v = f[v][i];
            return v;
        }
    }
}
int main() {
    int cas;
    read(cas);
    char opt[10];
    while (cas --) {
        cnt = 0;
        ms(H, 0);
        ms(dis, 0);
        ms(dep, 0);
        ms(f, 0);
        read(n);
        for (int i = 1; i < n; i ++) {
            int u, v;  ll w;
            read(u); read(v); read(w);
            add_edge(u, v, w);
            add_edge(v, u, w);
        }
        dep[1] = 1;
        dfs(1, 0, 0);
        while (1) {
            scanf("%s", opt);
            if (opt[1] == 'O') break;
            if (opt[0] == 'D') {
                int u, v;
                read(u); read(v);
                int Lca = lca(u, v), res;
                printf("%lld\n", dis[u] + dis[v] - dis[Lca] * 2);
            }
            else {
                int u, v, k;
                read(u); read(v); read(k);
                printf("%d\n", solve(u, v, k));
            }
        }
    }
    return 0;
}

[SPOJ913]QTREE2 - Query on a tree II【倍增LCA】的更多相关文章

  1. spoj 913 Query on a tree II (倍增lca)

    Query on a tree II You are given a tree (an undirected acyclic connected graph) with N nodes, and ed ...

  2. 【SPOJ QTREE2】QTREE2 - Query on a tree II(LCA)

    You are given a tree (an undirected acyclic connected graph) with N nodes, and edges numbered 1, 2, ...

  3. Query on a tree II 倍增LCA

    You are given a tree (an undirected acyclic connected graph) with N nodes, and edges numbered 1, 2, ...

  4. LCA SP913 QTREE2 - Query on a tree II

    SP913 QTREE2 - Query on a tree II 给定一棵n个点的树,边具有边权.要求作以下操作: DIST a b 询问点a至点b路径上的边权之和 KTH a b k 询问点a至点 ...

  5. SP913 QTREE2 - Query on a tree II

    思路 第一个可以倍增,第二个讨论在a到lca的路径上还是lca到b的路径上, 倍增即可 代码 #include <cstdio> #include <algorithm> #i ...

  6. SPOJ QTREE2 Query on a tree II

    传送门 倍增水题…… 本来还想用LCT做的……然后发现根本不需要 //minamoto #include<bits/stdc++.h> using namespace std; #defi ...

  7. SPOJ COT2 - Count on a tree II(LCA+离散化+树上莫队)

    COT2 - Count on a tree II #tree You are given a tree with N nodes. The tree nodes are numbered from  ...

  8. QTREE2 spoj 913. Query on a tree II 经典的倍增思想

    QTREE2 经典的倍增思想 题目: 给出一棵树,求: 1.两点之间距离. 2.从节点x到节点y最短路径上第k个节点的编号. 分析: 第一问的话,随便以一个节点为根,求得其他节点到根的距离,然后对于每 ...

  9. SPOJ913 Query on a tree II

    Time Limit: 433MS   Memory Limit: 1572864KB   64bit IO Format: %lld & %llu Description You are g ...

随机推荐

  1. Redis使用和部分源码剖析以及Django缓存和redis的关系

    0.特点: a.持久化 b.单进程.单线程 c.5大数据类型    d.用于操作内存的软件.    e.虽然是缓存数据库但是可以做持久化的工作 MySQL是一个软件,帮助开发者对一台机器的硬盘进行操作 ...

  2. MySQL分页时统计总记录行数并使用limit返回固定数目的记录

    需求很简单:假设有一个user表,表中实际上有10000条数据,但是我不知道有多少条,我要从数据库中每次取20条数据显示,那么怎么完成呢? 方案一: 首先执行一个 select count(*) as ...

  3. docker技术之安装

    由于工作原因需要使用docker完成集群的搭建,特此记录一下研究的docker技术. 首先简单的介绍一下docker: Docker 使用 Google 公司推出的 Go 语言 进行开发实现,基于 L ...

  4. babel (三) babel polly-fill

    Babel includes a polyfill that includes a custom regenerator runtime and core-js. This will emulate ...

  5. EntityFrameworkCore中的实体状态

    Entry表示一个追踪,里面有state属性,是EntityState的枚举类型. 每一个实体都有一个相对应的Entry: var entry = dbContext.ChangeTracker.En ...

  6. Laravel 的十八个最佳实践

    本文翻译改编自 Laravel 的十八个最佳实践 这篇文章并不是什么由 Laravel 改编的 SOLID 原则.模式等. 只是为了让你注意你在现实生活的 Laravel 项目中最常忽略的内容.   ...

  7. vue上传图片到服务器

    https://blog.csdn.net/qq_29712995/article/details/78839093(copy) HTML代码: <input accept="imag ...

  8. python之路--初识函数

    一 . 函数 什么是函数 f(x) = x + 1 y = x + 1 # 函数是对功能或者动作的封装 函数的语法 def 函数名(): 函数体 调用: 函数名() def play(): print ...

  9. qtp 自动货测试桌面程序-笔记(使用参数 parameters)

    dtGlobalSheet:运行整个test时候使用的参数(心得:可以将公共使用的测试数据放于全局表格中,所有action脚本都可以使用同一个数据,如供应商.客户.商品) dtActionSheet: ...

  10. jdbc工具类1.0

    package cn.zhouzhou; import java.sql.Connection; import java.sql.Driver; import java.sql.DriverManag ...