题目链接:XJOI - NOI2015-13 - C

题目分析

使用神奇的线段树合并在 O(nlogn) 的时间复杂度内解决这道题目。

对树上的每个点都建立一棵线段树,key是时间(即第几次操作),动态开点。

线段树的节点维护两个值,一个是这段时间内的 1 操作个数,另一个是这段时间内变化的黑色节点权值和。

在处理所有操作的时候,每棵线段树都是仅代表树上的一个点,因此线段树的每个节点维护的就是这段时间内以这个点为 a 的 1 操作个数和这段时间内这个点的黑色节点权值和(这个点 x 由黑变白就 -x, 由白变黑就 +x)。

在处理完所有操作后,我们进行一次 DFS,自底向上将线段树进行合并。

目前 DFS(x),先递归处理完 x 的每棵子树,然后枚举 x 的每棵子树,依次将它们的线段树合并到 x 的线段树上。

现在已经将 x 的前 j-1 棵子树的线段树合并到了 x 的线段树上,现在将第 j 棵子树的线段树合并到 x 的线段树上。

对于处于 j 子树内的 a 和处于 x 点或前 j-1 棵子树内的黑点,它们的 LCA 就是 x 点,因此他们对 x 的权值有贡献。

同理,处于 j 子树内的黑点和处于 x 点或前 j-1 棵子树内的 a ,他们的 LCA 也是 x 点,也要计算他们对 x 的权值的贡献。

一个黑点权值修改会对时间 key 比它大的 1 操作产生影响。

合并时,记合并的两棵线段子树为 (x, y),那么答案就要加上 Son[x][0] 的黑点权值修改 * Son[y][1] 的 1 操作个数。

同理,答案也要加上 Son[y][0] 的黑点权值修改 * Son[x][1] 的 1 操作个数。

然后递归下去合并 (Son[x][0], Son[y][0]) ,合并 (Son[x][1], Son[y][1]),继续计算两边子树内部的答案。

同时注意,这样计算的答案不包括 a 点本身就是一个黑点时贡献的权值,所以要单独加上这个情况的权值。

代码

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm> using namespace std; typedef long long LL; inline void Read(int &Num)
{
char c = getchar();
bool Neg = false;
while (c < '0' || c > '9')
{
if (c == '-') Neg = true;
c = getchar();
}
Num = c - '0'; c = getchar();
while (c >= '0' && c <= '9')
{
Num = Num * 10 + c - '0';
c = getchar();
}
if (Neg) Num *= -1;
} const int MaxN = 200000 + 5, MaxNode = 7200000 + 5; int n, m, Index;
int A[MaxN], Root[MaxN], Son[MaxNode][2], T[MaxNode]; LL Cnt;
LL Ans[MaxN], Sum[MaxNode]; struct Edge
{
int v;
Edge *Next;
} E[MaxN * 2], *P = E, *Point[MaxN]; inline void AddEdge(int x, int y)
{
++P; P -> v = y;
P -> Next = Point[x]; Point[x] = P;
} inline void Update(int x)
{
T[x] = T[Son[x][0]] + T[Son[x][1]];
Sum[x] = Sum[Son[x][0]] + Sum[Son[x][1]];
} void Add(int &x, int s, int t, int Pos, int Ds, int Dt)
{
if (x == 0) x = ++Index;
if (s == t)
{
Sum[x] += (LL)Ds;
T[x] += Dt;
return;
}
int m = (s + t) >> 1;
if (Pos <= m) Add(Son[x][0], s, m, Pos, Ds, Dt);
else Add(Son[x][1], m + 1, t, Pos, Ds, Dt);
Update(x);
} int Merge(int x, int y, int s, int t)
{
if (!x) return y;
if (!y) return x;
if (s == t)
{
T[x] += T[y];
Sum[x] += Sum[y];
return x;
}
Cnt += (LL)T[Son[x][1]] * Sum[Son[y][0]];
Cnt += (LL)T[Son[y][1]] * Sum[Son[x][0]];
int m = (s + t) >> 1;
Son[x][0] = Merge(Son[x][0], Son[y][0], s, m);
Son[x][1] = Merge(Son[x][1], Son[y][1], m + 1, t);
Update(x);
return x;
} void Solve(int x, int Fa)
{
for (Edge *j = Point[x]; j; j = j -> Next)
{
if (j -> v == Fa) continue;
Solve(j -> v, x);
}
for (Edge *j = Point[x]; j; j = j -> Next)
{
if (j -> v == Fa) continue;
Cnt = 0;
Root[x] = Merge(Root[x], Root[j -> v], 0, m);
Ans[x] += Cnt;
}
} int main()
{
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; ++i)
{
Read(A[i]);
if (A[i] != 1) A[i] = 0;
}
int a, b;
for (int i = 1; i < n; ++i)
{
Read(a); Read(b);
AddEdge(a, b); AddEdge(b, a);
}
for (int i = 1; i <= n; ++i)
Add(Root[i], 0, m, 0, A[i] * i, 0);
int f, x;
for (int i = 1; i <= m; ++i)
{
Read(f); Read(x);
if (f == 1)
{
Add(Root[x], 0, m, i, 0, 1);
if (A[x]) Ans[x] += (LL)x;
}
else
{
A[x] ^= 1;
if (A[x]) Add(Root[x], 0, m, i, x, 0);
else Add(Root[x], 0, m, i, -x, 0);
}
}
Solve(1, 0);
for (int i = 1; i <= n; ++i)
printf("%lld\n", Ans[i]);
return 0;
}

  

[XJOI NOI2015模拟题13] C 白黑树 【线段树合并】的更多相关文章

  1. [XJOI NOI2015模拟题13] B 最小公倍数 【找规律】

    题目链接:XJOI - NOI2015-13 - B 题目分析 通过神奇的观察+打表+猜测,有以下规律和性质: 1) 删除的 n 个数就是 1~n. 2) 当 c = 2 时,如果 n + 1 是偶数 ...

  2. [XJOI NOI2015模拟题13] A 神奇的矩阵 【分块】

    题目链接:XJOI NOI2015-13 A 题目分析 首先,题目定义的这种矩阵有一个神奇的性质,第 4 行与第 2 行相同,于是第 5 行也就与第 3 行相同,后面的也是一样. 因此矩阵可以看做只有 ...

  3. Codeforces Gym 101194G Pandaria (2016 ACM-ICPC EC-Final G题, 并查集 + 线段树合并)

    题目链接  2016 ACM-ICPC EC-Final Problem G 题意  给定一个无向图.每个点有一种颜色. 现在给定$q$个询问,每次询问$x$和$w$,求所有能通过边权值不超过$w$的 ...

  4. [CSP-S模拟测试]:影魔(树状数组+线段树合并)

    题目背景 影魔,奈文摩尔,据说有着一个诗人的灵魂.事实上,他吞噬的诗人灵魂早已成千上万.千百年来,他收集了各式各样的灵魂,包括诗人.牧师.帝王.乞丐.奴隶.罪人,当然,还有英雄.每一个灵魂,都有着自己 ...

  5. 学军NOI训练13 T3 白黑树

    唉,大学军有自己的OJ就是好,无限orz 只有周六的比赛是开放的囧,这场比赛最后因为虚拟机卡住没有及时提交…… 否则就能让大家看到我有多弱了…… 前两题题解写的很详细,可以自己去看,我来随便扯扯T3好 ...

  6. 【2018.06.26NOIP模拟】T1纪念碑square 【线段树】*

    [2018.06.26NOIP模拟]T1纪念碑square 题目描述 2034年,纪念中学决定修建校庆100周年纪念碑,作为杰出校友的你被找了过来,帮校方确定纪念碑的选址. 纪念中学的土地可以看作是一 ...

  7. ZOJ 4100 浙江省第16届大学生程序设计竞赛 A题 Vertices in the Pocket 线段树+并查集

    正赛的时候完全没看这个题,事后winterzz告诉我他想出来的解法. 首先题意是给出n个点,m次操作. 操作有一种是连接两个点,另一种是求此时再为这个图连k条边,最少和最多能有几个联通块. 最少的求法 ...

  8. 【BZOJ】3038: 上帝造题的七分钟2(线段树+暴力)

    http://www.lydsy.com:808/JudgeOnline/problem.php?id=3038 这题我就有得吐槽了,先是线段树更新写错,然后不知哪没pushup导致te,精度问题sq ...

  9. codevs2492上帝造题的七分钟 2(线段树)

    /* 区间修改 区间查询 可以用线段树搞 但是一般的标记下放对这个题好像不合适 只能改叶子 然后更新父亲(虽然跑的有点慢) 小优化:如果某个点是1 就不用再开方了 所以搞一个f[i]标记 i 这个点还 ...

随机推荐

  1. CentOS7.0下载各版本说明 新增Everything版

    来源:http://www.centoscn.com/CentOS/2014/0708/3268.html 下载CentOS-7.0-1406的时候,有很多可选则的版本,对于初学者来说,不知道选择哪个 ...

  2. Divide and conquer method

    分治法是最广泛使用的算法设计方法之一,其基本思想:把大问题分解成一些较小的问题,然后由小问题的解方便地构造出大问题的解. 分治法说穿了就是把问题放小,如果被分的问题还是比较大,那么久继续分下去.为了能 ...

  3. margin,padding之我见

    在网页布局中,margin和padding绝对会占很重要的作用,但是在实际的项目中,很多新手前端程序员们往往不能正确的使用它们,导致会出现一些不必要的麻烦,下面是我在前几天的项目中遇到的一些问题,以及 ...

  4. 数据库字段出现科学计数法e+的情况分析

    问题: 有时候,我们在将excel表格中数据导入数据库中时,对于表格中的数字会默认为float的数据类型,这个时候导入到数据库中的这个表的值是正常显示的: 然而如果你要把导入到数据库中的表,再插入到另 ...

  5. Linq DataTable Group By 分组显示人员明细

    实现功能:       多个字段分组源码样例: 原始数据: 分组后的输出结果: 源代码: public static void PrintPersons() { //准备数据 DataTable dt ...

  6. 深入了解webkit内核第一篇:JavaScript引擎深度解析

    看到HorkeyChen写的文章<[WebKit] JavaScriptCore解析--基础篇(三)从脚本代码到JIT编译的代码实现>,写的很好,深受启发.想补充一些Horkey没有写到的 ...

  7. 在jsp中获取服务器端的时间

    以前只知道在js中这样写:var date = new Date() : 但是这得到的是客户浏览器的时间,并不是服务器的时间. 当然可以通过在Java文件中添加属性,再在js中通过document.g ...

  8. Masonry 控件详解

    1.   Masonry的属性 @property (nonatomic,strong,readonly)MASConstraint *left; //左侧 @property(nonatomic,s ...

  9. Linux内核Radix Tree(三):API介绍

    1.     单值查找radix_tree_lookup 函数radix_tree_lookup执行查找操作,查找方法是:从叶子到树顶,通过数组索引键值值查看数组元素的方法,一层层地查找slot.其列 ...

  10. jQuery—一些常见方法(2)DOM操作【insertBefore(),insertAfter(),appendTo(),prependTo(),before(),after(),append(),prepend(),remove(),on(),off(),scrollTop()】

    一.insertBefore() 如下代码:找到span标签,将span标签剪切到div的前面 <!DOCTYPE html> <html lang="en"&g ...