【集训DAY4】矩形【线段树】

思路：

我们以x轴为加入时间，y轴为坐标轴建立线段树
在x处加a[i]，然后在x+w处减去a[i]。

$code$

#include<iostream>
#include<cstdio>
#include<algorithm>

using namespace std;

int n, w, h;
int s[100010], tot, cnt;

struct node {
	int x, y1, y2, w;
}q[100010];

struct abc {
	int maxx, l, r, flag;
}tree[400010];

bool cmp(node x, node y) {
	return (x.x < y.x) || (x.x == y.x && y.w < 0);
}

void build(int k, int l, int r) {
	tree[k].l = s[l], tree[k].r = s[r];
	if(r - l == 1) return;
	int mid = l + r >> 1;
	build(k << 1, l, mid);
	build(k << 1 | 1, mid, r);
	tree[k].maxx = max(tree[k << 1].maxx, tree[k << 1 | 1].maxx);
}

void update(int k, int x, int y, int w) {
	if(x <= tree[k].l && tree[k].r <= y) {
		tree[k].flag += w;
		tree[k].maxx += w;
		return ;
	}
	if(x < tree[k << 1].r) update(k << 1, x, min(y, tree[k << 1].r), w);
	if(y > tree[k << 1 | 1].l) update(k << 1 | 1, max(x, tree[k << 1 | 1].l), y, w);
	tree[k].maxx = max(tree[k << 1].maxx, tree[k << 1 | 1].maxx) + tree[k].flag;
}

int main() {
	scanf("%d%d%d", &n, &w, &h);
	for(int i = 1; i <= n; i ++) {
		int x, y, z;
		scanf("%d%d%d", &x, &y, &z);
		q[++ tot].x = x, q[tot].y1 = y, q[tot].y2 = y + h, q[tot].w = z;
		q[++ tot].x = x + w, q[tot].y1 = y, q[tot].y2 = y + h, q[tot].w = -z;
		s[++ cnt] = y, s[++ cnt] = y + h;
	}
	sort(s + 1, s + 1 + cnt);
	cnt = unique(s + 1, s + cnt) - s - 1;
	sort(q + 1, q + 1 + tot, cmp);
	build(1, 1, cnt);
	int ans = 0;
	for(int i = 1; i <= tot; i ++) {
		update(1, q[i].y1, q[i].y2, q[i].w);
		ans = max(ans, tree[1].maxx);
	}
	printf("%d", ans);
	return 0;
}