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

题目链接：点击进入

题目

题意

n 次操作，第 i 次给你一个区间 [ l , r ] ，将其染成 i 色，问最后会看见多少种颜色（操作区间交的部分，后来的颜色覆盖前面的颜色 ）

思路

这个题 n 是 1e4 ，而区间达到了 1e7 ，如果常规的线段树的话，是存整个区间里的点也就是存1-1e7，但是考虑到 n 只有 1e4 ,开那么大的空间浪费（也可能会TLE或者MLE），那我们可以考虑一下离散化。我感觉离散化就是根据原序列数值的相对关系，赋予每个值一个新的编号，这个编号对应的相对关系跟原来的是一样的，这样我们就可以直接在新的编号上进行操作，由于新的编号的范围 <= 原序列数值的范围，所以会解决一些数组存不下的情况。
回到这个题，离散化区间坐标（ 排序、去重 ），离散后的序列编号就是原来区间坐标值的相对大小关系，我们可以对这个离散后的序列编号进行线段树建立以及区间覆盖操作。
最后统计答案时，直接遍历树，标记统计颜色即可。

代码

// Problem: Mayor's posters
// Contest: Virtual Judge - POJ
// URL: https://vjudge.net/problem/POJ-2528
// Memory Limit: 65 MB
// Time Limit: 1000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

//#pragma GCC optimize(3)//O3
//#pragma GCC optimize(2)//O2
#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<stack>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
// #define int long long
//#define double long double
//#define rep(i,x,y) for(register int i = x; i <= y;++i)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const int maxn=1e5+10;
const int mod=998244353;
const double eps=1e-9;
const int N=5e3+10;
/*--------------------------------------------*/
inline int read()
{
    int k = 0, f = 1 ;
    char c = getchar() ;
    while(!isdigit(c)){if(c == '-') f = -1 ;c = getchar() ;}
    while(isdigit(c)) k = (k << 1) + (k << 3) + c - 48 ,c = getchar() ;
    return k * f ;
}
/*--------------------------------------------*/

int n,ans,x[maxn];
bool vis[maxn];
struct node
{
	int l;
	int r;
	int data;
	int lazy;
}t[maxn<<2];
struct line
{
	int l;
	int r;
}p[maxn];
void pushup(int p)
{
	t[p].lazy=(t[p<<1].lazy&&t[p<<1|1].lazy&&t[p<<1].data==t[p<<1|1].data)?1:0;
	if(t[p].lazy) t[p].data=t[p<<1].data;
}
void pushdown(int p)
{
	if(t[p].lazy)
	{
		t[p<<1].lazy=t[p].lazy;
		t[p<<1|1].lazy=t[p].lazy;
		t[p<<1].data=t[p].data;
		t[p<<1|1].data=t[p].data;
		t[p].lazy=0;
	}
}
void build(int p,int l,int r)
{
	t[p].l=l,t[p].r=r,t[p].lazy=0;
	if(l==r)
	{
		t[p].lazy=1;
		return ;
	}
	int mid=l+r>>1;
	build(p<<1,l,mid);
	build(p<<1|1,mid+1,r);
	pushup(p);
}
void change(int p,int l,int r,int x)
{
	if(l<=t[p].l&&t[p].r<=r)
	{
		t[p].data=x;
		t[p].lazy=1;
		return ;
	}
	pushdown(p);
	int mid=t[p].l+t[p].r>>1;
	if(l<=mid) change(p<<1,l,r,x);
	if(r>mid) change(p<<1|1,l,r,x);
	pushup(p);
}
void ask(int p,int l,int r)
{
	if(l<=t[p].l&&t[p].r<=r&&t[p].lazy)
	{
		if(t[p].data&&!vis[t[p].data])
			ans++,vis[t[p].data]=1;
		return ;
	}
	pushdown(p);
	int mid=t[p].l+t[p].r>>1;
	if(l<=mid) ask(p<<1,l,r);
	if(r>mid) ask(p<<1|1,l,r);
}

int main() 
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);cout.tie(0);
	int _;
	cin>>_;
	while(_--)
	{
		ans=0;
		cin>>n;
		for(int i=1;i<=n;i++)
		{
			cin>>p[i].l>>p[i].r;
			x[2*i-1]=p[i].l;
			x[2*i]=p[i].r;
		}
		sort(x+1,x+2*n+1);
		int len=unique(x+1,x+2*n+1)-x-1;
		build(1,1,len);
		for(int i=1;i<=n;i++)
		{
			int l=lower_bound(x+1,x+len+1,p[i].l)-x;
			int r=lower_bound(x+1,x+len+1,p[i].r)-x;
			change(1,l,r,i);
		}
		ask(1,1,len);
		cout<<ans<<endl;
		for(int i=0;i<=n;i++) vis[i]=0;
	}
    return 0;
}