题目
题目链接:http://poj.org/problem?id=2482
题目来源:某人的线段树合集。
简要题意:天空中有些星星,它们有亮度,你拿个框来框星星,求框里星星亮度之和最大多少。
题解
这个题好理解然后做的时候灵光一现,不过想到做法之后很久之后才去写了。
线段树记录的是以某位置为起点的框里有多少亮度和。
将星星转化为对 [x−w+1,x] 内的位置增加/减少一个亮度,也就是区间更新。
求左边到哪里可以用尺取法直接线性做,二分也是可以的。
首先必然是要离散化的,我选择了对 x ,然后
y−h,y 来离散化。因为 y+h 溢出,所以平移下。以 y 轴为基准建立扫描线,
y−h 处加上一条线段, y <script type="math/tex" id="MathJax-Element-7">y</script>处去掉。然后每轮去对线段树的根去求结果就行了。
代码
#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <stack>
#include <queue>
#include <string>
#include <vector>
#include <set>
#include <map>
#define fi first
#define se second
using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
// head
const int N = 1e5+5;
struct Star {
int x, y, b;
};
struct Seg {
int x1, x2, add;
Seg(int x1, int x2, int add) : x1(x1), x2(x2), add(add) {}
};
struct SegmentTree {
#define lson (rt<<1)
#define rson ((rt<<1)|1)
#define MID ((L+R)>>1)
#define lsonPara lson, L, MID
#define rsonPara rson, MID+1, R
const static int TN = N << 2;
int t[TN];
int lazy[TN];
void pushUp(int rt) {
t[rt] = max(t[lson], t[rson]);
}
void pushDown(int rt, int L, int R) {
if (lazy[rt]) {
lazy[lson] += lazy[rt];
lazy[rson] += lazy[rt];
t[lson] += lazy[rt];
t[rson] += lazy[rt];
lazy[rt] = 0;
}
}
void build(int rt, int L, int R) {
lazy[rt] = t[rt] = 0;
if (L == R) return;
build(lsonPara);
build(rsonPara);
pushUp(rt);
}
void modify(int rt, int L, int R, int l, int r, int v) {
if (l > R || r < L) return;
if (l <= L && r >= R) {
lazy[rt] += v;
t[rt] += v;
} else {
pushDown(rt, L, R);
modify(lsonPara, l, r, v);
modify(rsonPara, l, r, v);
pushUp(rt);
}
}
};
Star star[N];
int x[N], y[N*2], l[N];
vector<Seg> a[N*2];
SegmentTree st;
int main() {
int n, w, h;
while (scanf("%d%d%d", &n, &w, &h) == 3) {
for (int i = 0; i < n; i++) {
scanf("%d%d%d", &star[i].x, &star[i].y, &star[i].b);
x[i] = star[i].x;
y[i] = star[i].y - h, y[n+i] = star[i].y;
}
sort(x, x + n);
sort(y, y + 2 * n);
int xn = unique(x, x + n) - x;
int yn = unique(y, y + 2 * n) - y;
int lb = 0;
for (int i = 0; i < xn; i++) {
while (x[i] - x[lb] >= w) lb++;
l[i] = lb;
}
for (int i = 0; i < n; i++) {
int x2 = lower_bound(x, x + xn, star[i].x) - x;
int x1 = l[x2];
int y1 = lower_bound(y, y + yn, star[i].y - h) - y;
int y2 = lower_bound(y, y + yn, star[i].y) - y;
a[y1].push_back(Seg(x1, x2, star[i].b));
a[y2].push_back(Seg(x1, x2, -star[i].b));
}
st.build(1, 0, xn);
int ans = 0;
for (int i = 0; i < yn; i++) {
for (int j = 0; j < a[i].size(); j++) {
Seg &cur = a[i][j];
st.modify(1, 0, xn, cur.x1, cur.x2, cur.add);
}
ans = max(ans, st.t[1]);
a[i].clear();
}
printf("%d\n", ans);
}
return 0;
}