hysbz 1036 树的统计Count(树链剖分)

news/2024/7/23 16:20:50

题目链接:hysbz 1036 树的统计Count

题目大意:略。

解题思路:树链剖分+线段树维护。

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int maxn = 30005;
const int INF = 0x3f3f3f3f;

#define lson(x) ((x)<<1)
#define rson(x) (((x)<<1)|1)
int lc[maxn << 2], rc[maxn << 2], s[maxn << 2], m[maxn << 2], val[maxn];

inline void pushup(int u) {
    s[u] = s[lson(u)] + s[rson(u)];
    m[u] = max(m[lson(u)], m[rson(u)]);
}

void build (int u, int l, int r) {
    lc[u] = l;
    rc[u] = r;

    if (l == r) {
        s[u] = m[u] = val[l];
        return;
    }

    int mid = (l + r) / 2;
    build(lson(u), l, mid);
    build(rson(u), mid + 1, r);
    pushup(u);
}

void modify(int u, int x, int w) {
    if (lc[u] == x && rc[u] == x) {
        s[u] = m[u] = w;
        return;
    }

    int mid = (lc[u] + rc[u]) / 2;
    if (x <= mid)
        modify(lson(u), x, w);
    else
        modify(rson(u), x, w);
    pushup(u);
}

int sum(int u, int l, int r) {
    if (l <= lc[u] && rc[u] <= r)
        return s[u];

    int mid = (lc[u] + rc[u]) / 2, ret = 0;
    if (l <= mid)
        ret += sum(lson(u), l, r);
    if (r > mid)
        ret += sum(rson(u), l, r);
    return ret;
}

int query(int u, int l, int r) {
    if (l <= lc[u] && rc[u] <= r)
        return m[u];

    int mid = (lc[u] + rc[u]) / 2, ret = -INF;
    if (l <= mid)
        ret = max(ret, query(lson(u), l, r));
    if (r > mid)
        ret = max(ret, query(rson(u), l, r));
    return ret;
}

int N, Q, ne, first[maxn], jump[maxn * 2], link[maxn * 2];
int id, idx[maxn], far[maxn], son[maxn], cnt[maxn], top[maxn], dep[maxn];

inline void add_Edge (int u, int v) {
    link[ne] = v;
    jump[ne] = first[u];
    first[u] = ne++;
}

void dfs_fir (int u, int pre, int d) {
    far[u] = pre;
    dep[u] = d;
    son[u] = 0;
    cnt[u] = 1;

    for (int i = first[u]; i + 1; i = jump[i]) {
        int v = link[i];
        if (v == pre)
            continue;
        dfs_fir(v, u, d + 1);
        cnt[u] += cnt[v];
        if (cnt[son[u]] < cnt[v])
            son[u] = v;
    }
}

void dfs_sec(int u, int rot) {
    top[u] = rot;
    idx[u] = ++id;

    if (son[u])
        dfs_sec(son[u], rot);
    for (int i = first[u]; i + 1; i = jump[i]) {
        int v = link[i];
        if (v == far[u] || v == son[u])
            continue;
        dfs_sec(v, v);
    }
}

void init () {
    int u, v;
    id = ne = 0;
    memset(first, -1, sizeof(first));
    for (int i = 1; i < N; i++) {
        scanf("%d%d", &u, &v);
        add_Edge(u, v);
        add_Edge(v, u);
    }

    dfs_fir(1, 0, 1);
    dfs_sec(1, 1);
    for (int i = 1; i <= N; i++)
        scanf("%d", &val[idx[i]]);
    build(1, 1, N);
}

int solve_query(int u, int v) {
    int p = top[u], q = top[v], ret = -INF;
    while (p != q) {
        if (dep[p] < dep[q]) {
            swap(p, q);
            swap(u, v);
        }
        ret = max(ret, query(1, idx[p], idx[u]));
        u = far[p];
        p = top[u];
    }
    if (dep[u] > dep[v])
        swap(u, v);
    ret = max(ret, query(1, idx[u], idx[v]));
    return ret;
}

int solve_sum(int u, int v) {
    int p = top[u], q = top[v], ret = 0;
    while (p != q) {
        if (dep[p] < dep[q]) {
            swap(p, q);
            swap(u, v);
        }
        ret += sum(1, idx[p], idx[u]);
        u = far[p];
        p = top[u];
    }
    if (dep[u] > dep[v])
        swap(u, v);
    ret += sum(1, idx[u], idx[v]);
    return ret;
}

int main () {
    while (scanf("%d", &N) == 1) {
        init();

        int u, v;
        char op[10];
        scanf("%d", &Q);
        while (Q--) {
            scanf("%s%d%d", op, &u, &v);
            if (op[0] == 'C')
                modify(1, idx[u], v);
            else if (op[1] == 'M')
                printf("%d\n", solve_query(u, v));
            else
                printf("%d\n", solve_sum(u, v));
        }
    }
    return 0;
}

http://www.niftyadmin.cn/n/1446805.html

相关文章

python机器学习——聚类分析简介

聚类分析数据聚类理论理论一、聚类定义二、聚类与分类区别三、聚类分析的目的四、聚类主要方法数据聚类理论理论 一、聚类定义 数据聚类 ( Cluster analysis )是指根据数据的内在性质将数据分成一些聚合类&#xff0c;每一聚合类中的元素尽可能具有相同的特性&#xff0c;不同…

什么是3G(转)

3G是英文3rd Generation的缩写&#xff0c;指第三代移动通信技术。相对第一代模拟制式手机(1G)和第二代GSM、TDMA等数字手机(2G)&#xff0c;第三代手机一般地讲&#xff0c;是指将无线通信与国际互联网等多媒体通信结合的新一代移动通信系统。它能够处理图像、音乐、视频流等多…

@RequestBody接收不到前端传递过来的json数据

uniRequest.post(/orderParking,{parkingRecord:this.ParkingRecord})我刚开始只是写RequestBody ParkingRecord parkingRecord 一直获取的都是null, 直到用了Map标签才终于获取到参数了 RequestMapping(value "/orderParking",produces"application/json"…

要懂Greenplum索引,心里得有B树!

了解更多Greenplum技术干货&#xff0c;欢迎访问Greenplum中文社区网站 7月24日&#xff0c;Greenplum原厂内核研发马洪旭和大家直播分享了《深入浅出Greenplum内核》系列直播的第四期《Greenplum内核揭秘之B树索引》。相关视频已上传至Greenplum中文社区B站频道&#xff0c;戳…

python机器学习——Kmeans聚类

Kmeans聚类聚类基本思想Kmeans 介绍python 实现参考聚类基本思想 背景&#xff1a; 由于获取带有标签的数据成本比较高&#xff08;因为需要人工标记&#xff09;&#xff0c;而没有标签的数据却很容易获得。如果我们可以根据样本自身的属性或者说特征&#xff0c;给这写样本进…

秒杀系统架构

一、秒杀业务为什么难做 1&#xff09;im系统&#xff0c;例如qq或者微博&#xff0c;每个人都读自己的数据&#xff08;好友列表、群列表、个人信息&#xff09;&#xff1b; 2&#xff09;微博系统&#xff0c;每个人读你关注的人的数据&#xff0c;一个人读多个人的数据&…

让网站反向连接尽在你的掌握之中(转)

我们都知道&#xff0c;网站的反相链接数量&#xff0c;决定了网站的Link Popularity——当然&#xff0c;反相链接的质量也同样重要——从而最终影响到网站在搜索引擎中的排名。因此&#xff0c;在网站优化过程中&#xff0c;随时掌握有哪些网站、有多少网站建立了指向我们网站…

hysbz 2243 染色(树链剖分)

题目链接&#xff1a;hysbz 2243 染色 题目大意&#xff1a;略。 解题思路&#xff1a;树链剖分线段树的区间合并&#xff0c;但是区间合并比较简单&#xff0c;节点只要记录左右端点的颜色即可。 #include <cstdio> #include <cstring> #include <algorithm&g…