[JLOI 2015]城池攻占

Description

小铭铭最近获得了一副新的桌游,游戏中需要用 m 个骑士攻占 n 个城池。

这 n 个城池用 1 到 n 的整数表示。除 1 号城池外,城池 i 会受到另一座城池 fi 的管辖,
其中 fi <i。也就是说,所有城池构成了一棵有根树。这 m 个骑士用 1 到 m 的整数表示,其
中第 i 个骑士的初始战斗力为 si,第一个攻击的城池为 ci。
每个城池有一个防御值 hi,如果一个骑士的战斗力大于等于城池的生命值,那么骑士就可
以占领这座城池;否则占领失败,骑士将在这座城池牺牲。占领一个城池以后,骑士的战斗力
将发生变化,然后继续攻击管辖这座城池的城池,直到占领 1 号城池,或牺牲为止。
除 1 号城池外,每个城池 i 会给出一个战斗力变化参数 ai;vi。若 ai =0,攻占城池 i 以后骑士战斗力会增加 vi;若 ai =1,攻占城池 i 以后,战斗力会乘以 vi。注意每个骑士是单独计算的。也就是说一个骑士攻击一座城池,不管结果如何,均不会影响其他骑士攻击这座城池的结果。
现在的问题是,对于每个城池,输出有多少个骑士在这里牺牲;对于每个骑士,输出他攻占的城池数量。

Input

第 1 行包含两个正整数 n;m,表示城池的数量和骑士的数量。

第 2 行包含 n 个整数,其中第 i 个数为 hi,表示城池 i 的防御值。
第 3 到 n +1 行,每行包含三个整数。其中第 i +1 行的三个数为 fi;ai;vi,分别表示管辖
这座城池的城池编号和两个战斗力变化参数。
第 n +2 到 n + m +1 行,每行包含两个整数。其中第 n + i 行的两个数为 si;ci,分别表
示初始战斗力和第一个攻击的城池。

Output

 输出 n + m 行,每行包含一个非负整数。其中前 n 行分别表示在城池 1 到 n 牺牲的骑士

数量,后 m 行分别表示骑士 1 到 m 攻占的城池数量。

Sample Input

5 5
50 20 10 10 30
1 1 2
2 0 5
2 0 -10
1 0 10
20 2
10 3
40 4
20 4
35 5

Sample Output

2
2
0
0
0
1
1
3
1
1

HINT

 对于 100% 的数据,1 <= n;m <= 300000; 1 <= fi<i; 1 <= ci <= n; -10^18 <= hi,vi,si <= 10^18;ai等于1或者2;当 ai =1 时,vi > 0;保证任何时候骑士战斗力值的绝对值不超过 10^18。

题解

考虑可并堆。

先将所有 “骑士” 放在 第一个攻占的 “城池” 上。

将不合法的剔除(即战斗力小于防御力的 “骑士”),统计答案。

牺牲的骑士数量直接等于 $pop$ 掉的骑士人数,而骑士攻占的城池数等于起始城池与当前城池间的深度差。

现在考虑修改:可以打上标记, $pushdown$ 时先转移乘法标记,再转移加法标记。转移只需要转移 $merge$ 操作经过的节点。

值得注意的是,在 $pop$ 堆顶元素时需先将堆顶的标记下移。

 1 //It is made by Awson on 2018.1.4
 2 #include <set>
 3 #include <map>
 4 #include <cmath>
 5 #include <ctime>
 6 #include <queue>
 7 #include <stack>
 8 #include <cstdio>
 9 #include <string>
10 #include <vector>
11 #include <cstdlib>
12 #include <cstring>
13 #include <iostream>
14 #include <algorithm>
15 #define LL long long
16 #define LD long double
17 #define Max(a, b) ((a) > (b) ? (a) : (b))
18 #define Min(a, b) ((a) < (b) ? (a) : (b))
19 using namespace std;
20 const int N = 300000;
21 
22 struct mergable_tree {
23     int ch[N+5][2], dist[N+5], root[N+5];
24     LL prod[N+5], sum[N+5], key[N+5];
25     mergable_tree() {
26     for (int i = 1; i <= N; i++) prod[i] = 1;
27     }
28     void pushdown(int o) {
29     #define ls ch[o][0]
30     #define rs ch[o][1]
31     if (prod[o] != 1) {
32         key[ls] *= prod[o], key[rs] *= prod[o]; sum[ls] *= prod[o], sum[rs] *= prod[o]; prod[ls] *= prod[o], prod[rs] *= prod[o];
33         prod[o] = 1;
34     }
35     if (sum[o] != 0) {
36         key[ls] += sum[o], key[rs] += sum[o]; sum[ls] += sum[o], sum[rs] += sum[o];
37         sum[o] = 0;
38     }
39     #undef ls
40     #undef rs
41     }
42     int merge(int a, int b) {
43     if (!a || !b) return a+b;
44     pushdown(a), pushdown(b);
45     if (key[a] > key[b]) swap(a, b);
46     ch[a][1] = merge(ch[a][1], b);
47     if (dist[ch[a][0]] < dist[ch[a][1]]) swap(ch[a][0], ch[a][1]);
48     dist[a] = dist[ch[a][1]]+1;
49     return a;
50     }
51 }T;
52 int n, m, f, a[N+5], c[N+5];
53 LL h[N+5], v[N+5], s;
54 struct tt {
55     int to, next;
56 }edge[N+5];
57 int path[N+5], top;
58 int sum[N+5], ans[N+5], dep[N+5];
59 
60 void add(int u, int v) {
61     edge[++top].to = v;
62     edge[top].next = path[u];
63     path[u] = top;
64 }
65 void dfs(int u, int depth) {
66     dep[u] = depth;
67     for (int i = path[u]; i; i = edge[i].next) {
68     dfs(edge[i].to, depth+1); T.root[u] = T.merge(T.root[u], T.root[edge[i].to]);
69     }
70     while (T.key[T.root[u]] < h[u] && T.root[u] != 0) {
71     ++sum[u];
72     ans[T.root[u]] = dep[c[T.root[u]]]-depth;
73     T.pushdown(T.root[u]);
74     T.root[u] = T.merge(T.ch[T.root[u]][0], T.ch[T.root[u]][1]);
75     }
76     if (a[u] == 0) T.key[T.root[u]] += v[u], T.sum[T.root[u]] += v[u];
77     else T.key[T.root[u]] *= v[u], T.prod[T.root[u]] *= v[u], T.sum[T.root[u]] *= v[u];
78 }
79 void work() {
80     memset(ans, -1, sizeof(ans));
81     scanf("%d%d", &n, &m);
82     for (int i = 1; i <= n; i++) scanf("%lld", &h[i]);
83     for (int i = 2; i <= n; i++) {
84     scanf("%d%d%lld", &f, &a[i], &v[i]);
85     add(f, i);
86     }
87     for (int i = 1; i <= m; i++) {
88     scanf("%lld%d", &s, &c[i]);
89     T.key[i] = s;
90     T.root[c[i]] = T.merge(T.root[c[i]], i);
91     }
92     dfs(1, 1);
93     for (int i = 1; i <= n; i++) printf("%d\n", sum[i]);
94     for (int i = 1; i <= m; i++) printf("%d\n", ans[i] == -1 ? dep[c[i]] : ans[i]);
95 }
96 int main() {
97     work();
98     return 0;
99 }

 

posted @ 2018-01-04 17:23  NaVi_Awson  阅读(170)  评论(0编辑  收藏  举报