当前位置: 代码迷 >> 综合 >> HDU 1542 Atlantis(线段树扫描线)
  详细解决方案

HDU 1542 Atlantis(线段树扫描线)

热度:8   发布时间:2023-12-17 03:32:35.0

题意:

计算一堆长方形面积

思路:

线段树扫描线模板
以前手动去重,现在用STL写了一次,感觉好看多了

错误及反思:

PE和多组

代码:

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int N = 201;
double segtree[N<<2];
int lazy[N<<2],n,T=0;
struct L{double x1,x2,y;int ta;
} line[201];
vector<double> id;
bool cmp(L w,L e)
{return w.y<e.y;
}
int get_id(double x)
{return lower_bound(id.begin(),id.end(),x)-id.begin();
}
void pushup(int l,int r,int rt)
{if(lazy[rt]) segtree[rt]=id[r+1]-id[l];else if(l==r) segtree[rt]=0;else segtree[rt]=segtree[rt<<1]+segtree[rt<<1|1];
}
void update(int L,int R,int v,int l,int r,int rt)
{if(L<=l&&R>=r){lazy[rt]+=v;pushup(l,r,rt);return ;}int m=(l+r)/2;if(L<=m) update(L,R,v,lson);if(R>m) update(L,R,v,rson);pushup(l,r,rt);return ;
}
int main()
{while(scanf("%d",&n)&&n){printf("Test case #%d\n",++T);for(int i=0;i<n;i++){scanf("%lf%lf%lf%lf",&line[i*2].x1,&line[i*2].y,&line[i*2+1].x2,&line[i*2+1].y);line[i*2].x2=line[i*2+1].x2;line[i*2+1].x1=line[i*2].x1;line[i*2].ta=1;line[i*2+1].ta=-1;id.push_back(line[i*2].x1);id.push_back(line[i*2].x2);}sort(id.begin(),id.end()),id.erase(unique(id.begin(),id.end()),id.end());sort(line,line+n*2,cmp);double ans=0;for(int i=0;i<n*2;i++){int l=get_id(line[i].x1);int r=get_id(line[i].x2);if(i) ans+=1.0*(line[i].y-line[i-1].y)*segtree[1];if(l<=r-1) update(l,r-1,line[i].ta,0,id.size()-2,1);}printf("Total explored area: %.2f\n",ans);id.clear();memset(segtree,0,sizeof(segtree));memset(lazy,0,sizeof(lazy));puts("");}
}