现在的位置: 首页 > 综合 > 正文

poj 2482 Stars in Your Window(线段树+扫描线)

2014年04月05日 ⁄ 综合 ⁄ 共 2017字 ⁄ 字号 评论关闭

http://poj.org/problem?id=2482

题意:给出n个星星的坐标的亮度,一个矩形的宽和长,分别是w和h。问这个矩形能够框住的星星的最大亮度是多少。

看的解题报告,愧疚中。。。直接说思路吧。


解题:

将每个星星分成两个星星(x,y,val)和(x+w,y,-val)。当不断加入星星的时候相当于一个矩形框从左向右移动,当遇到第一个星星总亮度加上val,说明框住了该星星,当遇到第二个星星再加上-val,相当于没有框住星星。

先对y轴坐标离散化,对每个星星按x坐标从小到大排序,然后从左向右扫描每个线段[y,y+h],在线段树中相应区间加上val。用线段树求出此刻的最大值,插入完所有星星后,最大值就求出来了。

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define lint __int64

using namespace std;

const int maxn = 1000000;
lint y[maxn];
struct item
{
	lint x,y,val;
}point[maxn];

struct line
{
	lint l,r,add,sum;
}tree[maxn];

int cmp(const struct item & a, const struct item & b)
{
	//优先按x排序,若x相等,按val从小到大排序
	if(a.x == b.x)
		return a.val < b.val;
	return a.x < b.x;
}

void build(lint v,lint l, lint r)
{
	tree[v].l = l;
	tree[v].r = r;
	tree[v].add = tree[v].sum = 0;
	if(l == r)
		return;
	lint mid = (l+r)>>1;
	build(v*2,l,mid);
	build(v*2+1,mid+1,r);
}

int binsearch(lint low, lint high,lint key)
{
	while(low <= high)
	{
		int mid = (low+high)>>1;
		if(y[mid] == key)
			return mid;
		if(y[mid] < key)
			low = mid+1;
		else high = mid-1;
	}
	return 0;
}

void update(lint l, lint r, lint val, lint v)
{
	if(tree[v].l == l && tree[v].r == r)
	{
		tree[v].add += val;
		tree[v].sum += val;
		return;
	}

	if(tree[v].add)
	{
		tree[v*2].add += tree[v].add;
		tree[v*2].sum += tree[v].add;
		tree[v*2+1].add += tree[v].add;
		tree[v*2+1].sum += tree[v].add;
		tree[v].add = 0;
	}
	lint mid = (tree[v].l + tree[v].r)>>1;
	if(r <= mid)
		update(l,r,val,v*2);
	else if(l > mid)
		update(l,r,val,v*2+1);
	else
	{
		update(l,mid,val,v*2);
		update(mid+1,r,val,v*2+1);
	}
	tree[v].sum = max(tree[v*2].sum,tree[v*2+1].sum);
}

int main()
{
	lint n,w,h,i,ans,cnt;
	while(~scanf("%I64d %I64d %I64d",&n,&w,&h))
	{
		for(i = 1; i <= n; i++)
		{
			scanf("%I64d %I64d %I64d",&point[i].x,&point[i].y,&point[i].val);
			//分成两个星星。
			y[i] = point[i].y;
			y[n+i] = point[i].y+h;
			point[i+n].x = point[i].x+w;
			point[i+n].y = point[i].y;
			point[i+n].val = -point[i].val;
		}

		sort(y+1,y+1+n*2);
		sort(point+1,point+1+n*2,cmp);
		//对y离散化
		cnt = 1;
		for(i = 2; i <= n*2; i++)
		{
			if(y[i] != y[i-1])
				y[++cnt] = y[i];
		}
		build(1,1,cnt);

		ans = 0;
		for(i = 1; i <= n*2; i++)
		{
			lint l = binsearch(1,cnt,point[i].y);
			lint r = binsearch(1,cnt,point[i].y+h)-1;//对每个星星,求出其纵坐标对应离散后的编号
			if(l > r)
				swap(l,r);
			update(l,r,point[i].val,1);//更新[l,r]区间val值
			ans = max(ans,tree[1].sum);
		}
		printf("%I64d\n",ans);
	}
	return 0;
}

抱歉!评论已关闭.