POJ 2528 Mayor's posters(离散化+线段树)

//线段树的染色类型题目 //这题通过逆序修改线段树可以简单的实现,这是一种比较巧妙地做法 //也可以通过记录左颜色和右颜色来维护线段树实现 //下面的代码是通过逆序覆盖来实现的，逆序覆盖，即将最后一次贴上去的海报优先处理 //这样的话可以省去很多按顺序插入处理步骤，但是无法实现动态询问 //在离散化的时候我一开始使用MAP来实现，结果800MS，差点爆掉，实践证明MAP的查询速度是很慢的，还不如写个二分搞定 //但下面的代码严格来说是错误的，离散化有问题，不过数据太弱，还是可以AC的 #include<iostream> #include<algorithm> #include<vector> #define MAX 10005 using namespace std; struct seg { int l,r,color,cover; }segTree[8 * MAX];//4 是线段树特征，2是10000次边最坏有20000个不同边界，所以要*8 struct Extent//区间的结构体 { int x,y; }extent[MAX]; bool vis[2 * MAX]; int M[2 * MAX]; int ans,N; void buildTree(int fa,int l,int r)//建树 { segTree[fa].l = l; segTree[fa].r = r; segTree[fa].color = segTree[fa].cover = 0; if(l == r) return; int mid = (segTree[fa].l + segTree[fa].r) >> 1; if(r <= mid) buildTree(2*fa,l,r); else if(l > mid) buildTree(2*fa+1,l,r); else { buildTree(2*fa,l,mid); buildTree(2*fa+1,mid+1,r); } } int getIndex(int data)//通过二分查找返回离散后的数组下标 { int mid,x = 1,y = N+1; while(x < y) { mid = x + (y - x) / 2;//确保分界点总是靠近区间起点 if(M[mid] == data) return mid; else if(M[mid] > data) y = mid; else x = mid + 1; } return -1; } void insert(int fa,int l,int r,int color) { if(segTree[fa].cover > 0) return;//逆序覆盖，所以如果该区间已被覆盖过了，就直接返回即可，因为前面的覆盖对结果没有影响 if(l == segTree[fa].l && r == segTree[fa].r) { if(segTree[fa].cover == 0) { segTree[fa].color = color;//染色 segTree[fa].cover = 1; if(!vis[color])//如果这种颜色没出现过 { ++ans; vis[color] = 1; } } return; } int mid = (segTree[fa].l + segTree[fa].r) >> 1; if(r <= mid) insert(2*fa,l,r,color); else if(l > mid) insert(2*fa+1,l,r,color); else { insert(2*fa,l,mid,color); insert(2*fa+1,mid+1,r,color); } if(segTree[2*fa].cover > 0 && segTree[2*fa+1].cover > 0)//通过回溯跟结点顺便维护区间的覆盖情况 segTree[fa].cover = 1; } int main() { //freopen("in.txt","r",stdin); int t,n,x,y; scanf("%d",&t); while(t--) { N = 0; int color = 0; scanf("%d",&n); vector<int> v; memset(vis,0,sizeof(vis)); ans = 0; for(int i = 0;i < n;++i) { scanf("%d%d",&extent[i].x,&extent[i].y); v.push_back(extent[i].x); v.push_back(extent[i].y); } sort(v.begin(),v.end()); //离散化过程 M[++N] = v[0]; for(int i = 1;i < v.size();++i) if(v[i] != v[i-1]) M[++N] = v[i]; buildTree(1,1,N); for(int i = n - 1;i >= 0;--i) { x = getIndex(extent[i].x); y = getIndex(extent[i].y); insert(1,x,y,++color);//对每条边插入不同的颜色 } printf("%d/n",ans); } return 0; }