1. 程式人生 > >Mayor's posters【線段樹+離散化】

Mayor's posters【線段樹+離散化】

題解:

區間修改線段樹+離散化。cnt=unique(arr,arr+cnt)-arr 用於離散化時非常方便。

程式碼:

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<set>
#include<map>
using namespace std;
int se[20005];
int n;
struct node
{
    int l,r;
}p[10005];
int color[20005*4],laz[20005*4];
bool vis[20005];
void pushdown(int v)
{
    laz[v*2]=laz[v];
    laz[v*2+1]=laz[v];
    color[v*2]=laz[v*2];
    color[v*2+1]=laz[v*2+1];
    laz[v]=-1;
}
void updata(int v,int L,int R,int ql,int qr,int c)
{
    if(ql<=L && R<=qr)
    {
        color[v]=c;
        laz[v]=c;
        return;
    }
    if(laz[v]!=-1) pushdown(v);
    int mid=(L+R)/2;
    if(ql<=mid) updata(v*2,L,mid,ql,qr,c);
    if(qr>mid) updata(v*2+1,mid+1,R,ql,qr,c);
}
void query(int v,int L,int R)
{
    if(laz[v]!=-1 || L==R)
    {
        if(color[v]!=-1)
            vis[color[v]]=1;
        return;
    }
    if(laz[v]!=-1) pushdown(v);
    int mid=(L+R)/2;
    query(v*2,L,mid);
    query(v*2+1,mid+1,R);
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        int cnt=0;
        for(int i=0;i<n;i++)
        {
            scanf("%d%d",&p[i].l,&p[i].r);
            se[cnt++]=p[i].l;
            se[cnt++]=p[i].r;
        }
        sort(se,se+cnt);
        cnt=unique(se,se+cnt)-se;
        memset(laz,-1,sizeof laz);
        memset(color,-1,sizeof color);
        memset(vis,0,sizeof vis);
        for(int i=0;i<n;i++)
        {
            updata(1,1,cnt,lower_bound(se,se+cnt,p[i].l)-se+1,lower_bound(se,se+cnt,p[i].r)-se+1,i);
        }
        query(1,1,cnt);
        int sum=0;
        for(int i=0;i<n;i++)
            sum+=vis[i];
        printf("%d\n",sum);
    }
    return 0;
}