poj3327

来源:http://poj.org/problem?id=3277

更改:加一个cover值,记录区间是否纯合

不能跟h混用

还有一个re(16:tle)

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <iostream>
 
using namespace std;
 
struct node
{
	long long l,r,h,add;
	int cover;
};
 
#define lson 2*u
#define rson 2*u+1
const int maxSize=40000;
int n;
long long a[maxSize*2+5];//sort排序 
long long sum=0;
node N[maxSize*16+5],M[maxSize+5];
 
 
void build(int u,int begin,int end)
{
	int mid;
	
	N[u].l=begin;	N[u].r=end;	N[u].add=0;	N[u].h=0; N[u].cover=-1;
	if (begin==end)
	{
		return ; 
	}
		
	mid=(begin+end)/2;
	build(lson,begin,mid);
	build(rson,mid+1,end);
}
 
void pushdown(int u)
{
	if (N[u].add!=0)
	{
		N[lson].h=N[u].add;
		N[rson].h=N[u].add;
		N[lson].add=N[u].add;
		N[rson].add=N[u].add;
		N[lson].cover=0;
		N[rson].cover=0;
		N[u].add=0;
	}
}

void pushup(int u)
{
	N[u].h=max(N[lson].h,N[rson].h);
	if (N[lson].cover==0 && N[rson].cover==0 && N[lson].h==N[rson].h)
		N[u].cover=0;
	else
		N[u].cover=-1;
}
 
void update(int u,int begin,int end,int h)
{
	int mid;
	
	if (N[u].l==begin && N[u].r==end)
	{
		if (h>N[u].h)
		{
			N[u].h=h;
			N[u].add=h;
			N[u].cover=0;
			return ;
		}
		else if (N[u].cover==0)
			return ;
	}
	
	pushdown(u);
		
	mid=(N[u].l+N[u].r)/2;
	if (end<=mid)
		update(lson,begin,end,h);
	else if (begin>mid)
		update(rson,begin,end,h);
	else
	{
		update(lson,begin,mid,h);
		update(rson,mid+1,end,h);
	}
	pushup(u);
}
 
void query(int u)
{
	int mid;
	
	if (N[u].cover==0)
	{
		sum+=N[u].h*(a[N[u].r+1]-a[N[u].l]);
		return ;
	}
	
	pushdown(u);
	
	mid=(N[u].l+N[u].r)/2;
	query(lson);
	query(rson);
}
 
int main()
{
	int i,size;
	
	freopen("a.txt","r",stdin);
	scanf("%d",&n);
	
	for (i=0;i<n;i++)
	{
		scanf("%lld%lld%lld",&M[i].l,&M[i].r,&M[i].h);
		a[i*2+1]=M[i].l;	a[i*2+2]=M[i].r;	//要改成从1开始!!!!!! 
	}
	sort(a+1,a+n*2+1);			//排序 
	size=unique(a+1,a+1+2*n)-a-1;		//去重 
	for (i=0;i<n;i++)		//离散化 
	{
		M[i].l=lower_bound(a+1,a+1+size,M[i].l)-a;
		M[i].r=lower_bound(a+1,a+1+size,M[i].r)-a;
	}
	
	
	build(1,1,size-1);
	for (i=0;i<n;i++)
		update(1,M[i].l,M[i].r-1,M[i].h);
	query(1);
	printf("%lld\n",sum);
	
	return 0;
}

猜你喜欢

转载自blog.csdn.net/scutbenson/article/details/81450618