首页 > 其他分享 >【题解】P6071 『MdOI R1』Treequery

【题解】P6071 『MdOI R1』Treequery

时间:2023-01-13 08:44:06浏览次数:56  
标签:sz R1 int 题解 top son dep maxn MdOI

海浪尽头的你啊,到底何时归来?

额滴就木异象啊……

思路

清真树论。

树论地考虑祖先后代关系,分讨一下。

用 ST 表处理一下 \(lca(l, r) = u\):

  1. \(u, p\) 无祖先后代关系,答案为 \(dis(u, p)\)

  2. \(p\) 是 \(u\) 的祖先,答案为 \(dis(u, p)\)

  3. \(u\) 是 \(p\) 的祖先,继续分类讨论:

    • \(p\) 的子树内有 \([l, r]\) 中的点,无解

    • 否则 \([l, r]\) 中的点都在 \(p\) 的子树外,向上倍增找到 \(p\) 最低且子树中包含 \([l, r]\) 中结点的祖先。

最后一步可以主席树维护 \([l, r]\) 内的 dfs 序。

时间复杂度 \(O(n \log^2 n)\)

代码

#include <cstdio>
#include <vector>
#include <iostream>
using namespace std;

const int maxn = 2e5 + 5;
const int lg_sz = 20;
const int t_sz = maxn * 40;

int n, q, cnt;
int fa[maxn], son[maxn], dfn[maxn], top[maxn];
int dep[maxn], sz[maxn], len[maxn], lg[maxn], rt[maxn];
int f[maxn][lg_sz], lc[maxn][lg_sz];
int ls[t_sz], rs[t_sz], sum[t_sz];
vector<int> g[maxn], d[maxn];

void dfs1(int u, int pre)
{
    fa[u] = pre;
    sz[u] = 1;
    dep[u] = dep[pre] + 1;
    for (int i = 0; i < g[u].size(); i++)
    {
        int v = g[u][i], w = d[u][i];
        if (v == pre) continue;
        f[v][0] = u;
        len[v] = len[u] + w;
        dfs1(v, u);
        sz[u] += sz[v];
        if (sz[v] > sz[son[u]]) son[u] = v;
    }
}

void dfs2(int u, int t)
{
    top[u] = t;
    dfn[u] = ++cnt;
    if (son[u]) dfs2(son[u], t);
    for (int v : g[u])
        if ((v != fa[u]) && (v != son[u])) dfs2(v, v);
}

int lca(int u, int v)
{
    while (top[u] != top[v])
    {
        if (dep[top[u]] < dep[top[v]]) swap(u, v);
        u = fa[top[u]];
    }
    return (dep[u] < dep[v] ? u : v);
}

int get_lca(int l, int r)
{
    int k = lg[r - l + 1];
    return lca(lc[l][k], lc[r - (1 << k) + 1][k]);
}

bool in(int u, int v) { return (dfn[u] >= dfn[v]) && (dfn[u] <= dfn[v] + sz[v] - 1); }

int dis(int u, int v) { return len[u] + len[v] - 2 * len[lca(u, v)]; }

int update(int pre, int l, int r, int p)
{
    int k = ++cnt;
    sum[k] = sum[pre] + 1, ls[k] = ls[pre], rs[k] = rs[pre];
    if (l == r) return k;
    int mid = (l + r) >> 1;
    if (p <= mid) ls[k] = update(ls[pre], l, mid, p);
    else rs[k] = update(rs[pre], mid + 1, r, p);
    return k;
}

int query(int k, int pre, int l, int r, int ql, int qr)
{
    if ((l >= ql) && (r <= qr)) return sum[k] - sum[pre];
    int mid = (l + r) >> 1, sum = 0;
    if (ql <= mid) sum += query(ls[k], ls[pre], l, mid, ql, qr);
    if (qr > mid) sum += query(rs[k], rs[pre], mid + 1, r, ql, qr);
    return sum;
}

int main()
{
    int last_ans = 0;
    scanf("%d%d", &n, &q);
    for (int i = 1, u, v, w; i <= n - 1; i++)
    {
        scanf("%d%d%d", &u, &v, &w);
        g[u].push_back(v), d[u].push_back(w);
        g[v].push_back(u), d[v].push_back(w);
    }
    dfs1(1, 0);
    dfs2(1, 1);
    cnt = 0;
    for (int i = 2; i <= n; i++) lg[i] = lg[i >> 1] + 1;
    for (int j = 1; (1 << j) <= n; j++)
        for (int i = 1; i <= n; i++)
            f[i][j] = f[f[i][j - 1]][j - 1];
    for (int i = 1; i <= n; i++) lc[i][0] = i;
    for (int j = 1; (1 << j) <= n; j++)
        for (int i = 1; i + (1 << j) - 1 <= n; i++)
            lc[i][j] = lca(lc[i][j - 1], lc[i + (1 << (j - 1))][j - 1]);
    for (int i = 1; i <= n; i++) rt[i] = update(rt[i - 1], 1, n, dfn[i]);
    while (q--)
    {
        int p, l, r;
        scanf("%d%d%d", &p, &l, &r);
        p ^= last_ans, l ^= last_ans, r ^= last_ans;
        // printf("get query: %d %d %d\n", p, l, r);
        int u = get_lca(l, r);
        // printf("get lca(%d, %d) = %d\n", 3, 5, get_lca(3, 5));
        if (in(u, p) || (!in(p, u))) last_ans = dis(u, p);
        else if (query(rt[r], rt[l - 1], 1, n, dfn[p], dfn[p] + sz[p] - 1)) last_ans = 0;
        else
        {
            // puts("in");
            int cur = p;
            for (int i = lg[dep[cur]]; i >= 0; i--)
            {
                int v = f[cur][i];
                if (!v) continue;
                if (!query(rt[r], rt[l - 1], 1, n, dfn[v], dfn[v] + sz[v] - 1)) cur = v;
                // puts("query done");
            }
            // printf("cur : %d\n", cur);
            last_ans = dis(p, f[cur][0]);
        }
        printf("%d\n", last_ans);
    }
    return 0;
}

标签:sz,R1,int,题解,top,son,dep,maxn,MdOI
From: https://www.cnblogs.com/lingspace/p/p6071.html

相关文章

  • 洛谷P7792 KRIZA 题解 C++
    洛谷P7792KRIZA题解C++题目概述:题目传送门Sisyphus在一个圆形的房间里,房间内有n扇锁着的门,他有n把钥匙,其中第i把钥匙对应第$v_i$扇门,遇到不匹配的钥匙就放......
  • 【题解】P4899 [IOI2018] werewolf 狼人
    そうやってただ日が暮れるまで語り掛ける本当の言葉题意给定一个有向图和若干询问,每次询问是否存在一条满足条件的路径:端点分别为\(u,v\)前面一段不经过\([1,L......
  • 传递游戏【题解】
    Description毛大神最近在玩一个传递游戏,即有\(N\)个人在做传递物品的游戏,这N个人的编号为\(1,2,3,...,N-1,N\)。游戏规则是这样的:开始时物品可以在任意一人手上,他可把物......
  • 表达式的值【题解】
    [NOIP2011普及组]表达式的值题目描述对于1位二进制变量定义两种运算:运算的优先级是:先计算括号内的,再计算括号外的。“×”运算优先于“⊕”运算,即计算表达式......
  • [NOIP2017 普及组]跳房子 【题解】
    题目背景NOIP2017普及组T4题目描述跳房子,也叫跳飞机,是一种世界性的儿童游戏,也是中国民间传统的体育游戏之一。跳房子的游戏规则如下:在地面上确定一个起点,然后在起点......
  • 洛谷 P8077 [WC2022] 序列变换 题解
    题目链接。WC2023之前补一下WC2022的题,参考了官方题解。首先,把括号序列转化为二叉树,\(\texttt{(A)B}\)转为一个点的左子树是\(A\),右子树是\(B\)。相当于括号序列先......
  • 1.8模拟赛题解
    T1考虑每次反弹后,球的运动轨迹都会偏移\(2\beta\),总偏移量即为\(2k\beta\),而最后需要回到原点,因此\(360|2k\beta\),简单求\(\gcd\)即可。T2设\(ans_k\)表示出现过......
  • 1.11模拟赛题解
    T1对于方阵\(A\),考虑其反方阵\(A'\)。容易发现\(A\)与\(A'\)的权值和相同,而其中必有一个与\(B\)的差不超过\(\lfloor\frac{nm}{2}\rfloor\),因此判断一下哪个满足......
  • 1.9模拟赛题解
    T1从左到右扫描,首先如果\(a_i<b_i\)那么一定无解,否则不断在其右边找最近的\(j\)使得\(a_j\in[b_i,a_i]\),把\(a_i\)和\(a_j\)交换。感性理解这是对的。T2先证操......
  • 1.12模拟赛题解
    T1容易知道答案为原图的最大子二分图大小。枚举每个点在二分图的左边还是右边,计算出答案。时间复杂度\(O(2^n\timesm)\)。T2考虑递推构造方案。假设现在已经有了一组......