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

//线段树的染色类型题目//这题通过逆序修改线段树可以简单的实现,这是一种比较巧妙地做法//也可以通过记录左颜色和右颜色来维护线段树实现//下面的代码是通过逆序覆盖来实现的，逆序覆盖，即将最后一次贴上去的海报优先处理//这样的话可以省去很多按顺序插入处理步骤，但是无法实现动态询问//在离散化的时候我一开始使用MAP来实现，结果800MS，差点爆掉，实践证明MAP的查询速度是很慢的，还不如写个二分搞定//但下面的代码严格来说是错误的，离散化有问题，不过数据太弱，还是可以AC的#include<iostream>#include<algorithm>#include<vector>#define MAX 10005using namespace std;struct seg{int l,r,color,cover;}segTree[8 * MAX];//4 是线段树特征，2是10000次边最坏有20000个不同边界，所以要*8struct 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;}