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