UVA1232 - SKYLINE(线段树区间修改)

UVA1232 - SKYLINE(线段树区间修改)

题目链接

题目大意：按照顺序盖楼，如果这个位置（当前要盖的楼覆盖范围内）要新建的楼的高度>=之前就有的最大高度，那么就+1.最后输出这个+1的总数。

解题思路：线段树区间修改值，并且每次修改的时候返回修改的位置总数。因为可能左右子树的高度会有不同，所以这里加入一个sign来表示左右这一段是否高度一致。

代码：

#include <cstdio>
#include <cstring>

const int maxn = 1e5 + 5;

#define lson(x) ((x)<<1)
#define rson(x) (((x)<<1) + 1)

struct Node {

int l, r, h, sign;
void set (int l, int r, int h, int sign) {

this->l = l;
this->r = r;
this->h = h;
this->sign = sign;
}
}node[4 * maxn];

void pushdown(int u) {

node[lson(u)].sign = node[rson(u)].sign = 1;
node[lson(u)].h = node[rson(u)].h = node[u].h;
node[u].sign = 0;
node[u].h = 0;
}

void pushup (int u) {

node[u].l = node[lson(u)].l;
node[u].r = node[rson(u)].r;

if (node[lson(u)].sign && node[rson(u)].sign && node[lson(u)].h == node[rson(u)].h) {

node[u].sign = 1;
node[u].h = node[lson(u)].h;
} else
node[u].sign = node[u].h = 0;
}

void build (int u, int l, int r) {

if (l == r)
node[u].set(l, r, 0, 1);
else {

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

int update (int u, int l, int r, int h) {

if (node[u].sign && node[u].h > h)
return 0;

if (node[u].l >= l && node[u].r <= r && node[u].sign) {

node[u].h = h;
return node[u].r - node[u].l + 1;
}

int ret = 0;
int m = (node[u].l + node[u].r) / 2;

if (node[u].sign)
pushdown(u);

if (l <= m)
ret += update (lson(u), l, r, h);
if (r > m)
ret += update (rson(u), l, r, h);
pushup(u);

return ret;
}

int main () {

int T;
int n, l, r, h, ans;

scanf ("%d", &T);

while (T--) {

build(1, 1, maxn);
scanf ("%d", &n);

ans = 0;
for (int i = 0; i < n; i++) {

scanf ("%d%d%d", &l, &r, &h);
ans += update (1, l + 1, r, h);
}

printf ("%d\n", ans);
}
return 0;
}