码迷,mamicode.com
首页 > 其他好文 > 详细

hdu4533 威威猫系列故事——晒被子

时间:2015-08-04 21:02:41      阅读:119      评论:0      收藏:0      [点我收藏+]

标签:线段树

Problem Description
  因为马拉松初赛中吃鸡腿的题目让不少人抱憾而归,威威猫一直觉得愧对大家,这几天他悄悄搬到直角坐标系里去住了。
  生活还要继续,太阳也照常升起,今天,威威猫在第一象限晒了N条矩形的被子,被子的每条边都和坐标轴平行,不同被子的某些部分可能会叠在一起。这时候,在原点处突然发了场洪水,时间t的时候,洪水会蔓延到( t, t ),即左下角为( 0, 0 ) ,右上角为( t, t )的矩形内都有水。
  悲剧的威威猫想知道,在时间t1, t2, t3 ... tx 的时候,他有多少面积的被子是湿的?
 

Input
输入数据首先包含一个正整数T,表示有T组测试数据;
每组数据的第一行首先是一个整数N,表示有N条被子;
接下来N行,每行包含四个整数x1, y1, x2, y2,代表一条被子的左下角和右上角的坐标;
然后接下来一行输入一个整数x,表示有x次询问;
再接下来x行,输入x个严格单调递增的整数,每行一个,表示威威猫想知道的时间ti。

[Technical Specification]
T <= 5
0 < N <= 20000
1 <= x1 < x2 <= 200000
1 <= y1 < y2 <= 200000
1 <= x <= 20000
1 <= ti <= 200000 (1 <= i <= x )
 

Output
对于每次询问,请计算并输出ti时有多少面积的被子是湿的,每个输出占一行。
 

Sample Input
1 2 1 1 3 3 2 2 4 4 5 1 2 3 4 5
 

Sample Output
0 1 5 8 8

这题里求的面积是各个矩形的面积和,并不是矩形面积的并,即相互不影响,因为对于各个时段,矩形的面积都可以用A*t^2+B*t+C表示,所以我们可以用树状数组或者线段树分别维护A,B,C,每次读入一个矩形,就记录它对不同时间段t的影响,然后把相应的系数加到线段树中。

当一个矩形读入时(左下角坐标(x1,y1),右上角坐标(x2,y2)),有四种影响情况:

1.0~max(x1,y1) 这个时期因为t时间形成的矩形在该矩形下面,对面积没有影响,所以不用考虑。

2.如果存在t时间形成的矩形部分覆盖该矩形,但没有碰到或者超过该矩形的上边或者右边,那么要同时更新A,B,C,表达式为(t-x1)*(t-y1)。这里的判断表达式为如果(max(x1,y1)<min(x2,y2)),则更新max(x1,y1)~min(x2,y2)。

3.t时间内形成的矩形恰好碰到或者越过右边或上边的一条,即这段时间内的矩形面积变化中(x2-x1)或(y2-y1)为常数,那么要更新,注意,这里更新的时候要分类讨论,具体看代码。

4.max(x2,y2)~maxn,这个时期因为t时间形成的矩形已经完全覆盖矩形,所以要更新C,加上面积常数。

写线段树的时候注意,三个变量要一起存,不能开三个线段树,会超内存的。

线段树代码:

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<string>
#include<algorithm>
using namespace std;
#define ll __int64
#define maxn 200050
ll A,B,C;
ll a[10];
struct node{
    ll l,r,cnt1,cnt2,cnt3,flag;
}b[4*maxn];

void build(ll l,ll r,ll i)
{
    ll mid;
    b[i].l=l;b[i].r=r;b[i].cnt1=b[i].cnt2=b[i].cnt3=0;b[i].flag=1;
    if(l==r)return;
    mid=(l+r)/2;
    build(l,mid,i*2);
    build(mid+1,r,i*2+1);
}
void update(ll l,ll r,ll a[],ll i)
{
    ll mid;
    if(b[i].l==l && b[i].r==r){
        b[i].cnt1+=a[1];b[i].cnt2+=a[2];b[i].cnt3+=a[3];return;
    }
    mid=(b[i].l+b[i].r)/2;
    if(r<=mid)update(l,r,a,i*2);
    else if(l>mid)update(l,r,a,i*2+1);
    else{
        update(l,mid,a,i*2);
        update(mid+1,r,a,i*2+1);
    }
}

void question(ll id,ll i)
{
    ll mid;
    if(b[i].l==id && b[i].r==id){
        A=b[i].cnt1;B=b[i].cnt2;C=b[i].cnt3;return;
    }
    if(b[i].cnt1){
        b[i*2].cnt1+=b[i].cnt1;
        b[i*2+1].cnt1+=b[i].cnt1;
        b[i].cnt1=0;
    }
    if(b[i].cnt2){
        b[i*2].cnt2+=b[i].cnt2;
        b[i*2+1].cnt2+=b[i].cnt2;
        b[i].cnt2=0;
    }
    if(b[i].cnt3){
        b[i*2].cnt3+=b[i].cnt3;
        b[i*2+1].cnt3+=b[i].cnt3;
        b[i].cnt3=0;
    }
    mid=(b[i].l+b[i].r)/2;
    if(id<=mid)question(id,i*2);
    else question(id,i*2+1);
}

int main()
{
    ll i,j,x1,x2,y1,y2,t,t1,t2;
    int T,n,m;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        build(1,maxn,1);
        for(i=1;i<=n;i++){
            scanf("%I64d%I64d%I64d%I64d",&x1,&y1,&x2,&y2);
            if(max(x1,y1)<min(x2,y2)){
                a[1]=1;a[2]=-(x1+y1);a[3]=x1*y1;
                update(max(x1,y1),min(x2,y2)-1,a,1);
            }
  
            if(x2<y2){
                a[1]=0;a[2]=-x1+x2;a[3]=y1*(x1-x2);
                update(max(x2,y1),y2-1,a,1);  
            }
            if(y2<x2){
                a[1]=0;a[2]=-y1+y2;a[3]=x1*(y1-y2);
                update(max(y2,x1),x2-1,a,1);  
            }
            a[1]=0;a[2]=0;a[3]=(x2-x1)*(y2-y1);
            update(max(x2,y2),maxn,a,1);
            
        }
        scanf("%d",&m);
        for(i=1;i<=m;i++){
            scanf("%I64d",&t);
            A=B=C=0;
            question(t,1);
            printf("%I64d\n",A*t*t+B*t+C);
        }
    }
    return 0;
}


树状数组代码://速度快了一倍= =

#include <iostream>  
#include <cstdio>  
#include <cstring> 
#define maxn 200010  
#define ll __int64 
using namespace std;
ll max(ll a,ll b){
	return a>b?a:b;
}
ll min(ll a,ll b){
	return a<b?a:b;
}


struct segment{
	ll b[maxn],ans;
	void clear(){
		memset(b,0,sizeof(b));
	}
	int lowbit(ll x){
		return x&(-x);
	}
	void update(ll pos,ll num){
		while(pos<=maxn){
			b[pos]+=num;pos+=lowbit(pos);
		}
	}
	ll getsum(ll pos){
		ans=0;
		while(pos>0){
			ans+=b[pos];pos-=lowbit(pos);
		}
		return ans;
	}
}A,B,C;

void fun(ll l,ll r,ll num1,ll num2,ll num3){
	A.update(l,num1);A.update(r+1,-num1);
	B.update(l,num2);B.update(r+1,-num2);
	C.update(l,num3);C.update(r+1,-num3);
}

int main()  
{  
  
    int t;  
    scanf("%d",&t);  
    while(t--)  
    {  
        A.clear(); B.clear(); C.clear();  
       int n,m;  
        scanf("%d",&n);  
        for(int i=0;i<n;i++)  
        {  
            ll x1,y1,x2,y2;  
            scanf("%I64d%I64d%I64d%I64d",&x1,&y1,&x2,&y2);  
  
            if(max(x1,y1)<min(x2,y2))  
                fun(max(x1,y1),min(x2,y2),1,-(x1+y1),x1*y1);  
  
            if(x2<y2) fun(max(x2,y1)+1,y2,0,-x1+x2,y1*(x1-x2));  
            if(y2<x2) fun(max(y2,x1)+1,x2,0,-y1+y2,x1*(y1-y2));  
  
            fun(max(x2,y2)+1,maxn,0,0,(x2-x1)*(y2-y1));  
        }  
  
        scanf("%d",&m);  
        while(m--)  
        {  
            ll t;  
            scanf("%I64d",&t);  
            ll ans=0;  
            ans+=A.getsum(t)*t*t;  
            ans+=B.getsum(t)*t;  
            ans+=C.getsum(t);  
  
            printf("%I64d\n",ans);  
        }  
    }  
    return 0;  
} 



版权声明:本文为博主原创文章,未经博主允许不得转载。

hdu4533 威威猫系列故事——晒被子

标签:线段树

原文地址:http://blog.csdn.net/kirito_acmer/article/details/47281679

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!