[swustoj 764] 校门外的树 Plus Plus

校门外的树 Plus Plus(0764)

问题描述

西南某科技大学的校门外长度为 L 的公路上有一排树，每两棵相邻的树之间的间隔都是 1 米。我们可以把马路看成一个数轴，马路的一端在数轴 1 的位置，另一端在 L 的位置；数轴上的每个整数点，即 1，2，……，L，都种有一棵树。

现在要将这排树的某一段涂成某种颜色，给定 N 组区间[ J,K ]，表示从 J 到 K 的树都要涂上一种颜色（每次涂的颜色不一样,而且每次涂色会覆盖掉原来的颜色），区间之间可能有重叠的部分, L 的长度为 1000,0000。现在你的任务是计算涂色工作结束后，这一排树里总共有多少种颜色的树？（没涂过色的不算）

输入

第一行为一个整数N （1<=N<=10000）。 接下来的N行每行是两个整数数J，K（1<=J<=K<=1000,0000），表示涂色的起止区间。

多组测试数据。

输出

一个整数，表示有多少种颜色的树。

样例输入

2
1 10
5 5
1
1 1

样例输出

2

1

 

线段树可解

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 40005
  
struct tree
{
    int l,r;
}p[N<<2];
  
int tot;
int ans;
int x[N];
int lazy[N<<2];
bool flag[N<<2];
  
void pushdown(int rt)
{
    if(lazy[rt]!=-1)
    {
        lazy[rt<<1]=lazy[rt<<1|1]=lazy[rt];
        lazy[rt]=-1;
    }
}
void build(int l,int r,int rt)
{
    lazy[rt]=-1;
    if(l==r) return;
    int m=(l+r)>>1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
}
void update(int l,int r,int rt,int L,int R,int c)
{
    if(l==L && R==r)
    {
        lazy[rt]=c;
        return;
    }
    pushdown(rt);
    int m=(l+r)>>1;
    if(R<=m) update(l,m,rt<<1,L,R,c);
    else if(L>m) update(m+1,r,rt<<1|1,L,R,c);
    else
    {
        update(l,m,rt<<1,L,m,c);
        update(m+1,r,rt<<1|1,m+1,R,c);
    }
}
void query(int l,int r,int rt)
{
    if(lazy[rt]!=-1)
    {
        if(!flag[lazy[rt]])
        {
            ans++;
            flag[lazy[rt]]=1;
        }
        return;
    }
    if(l>=r) return;
    pushdown(rt);
    int m=(l+r)>>1;
    query(l,m,rt<<1);
    query(m+1,r,rt<<1|1);
}
int main()
{
    int n,i;
    while(scanf("%d",&n)!=EOF)
    {
        ans=0;
        tot=0;
        memset(flag,0,sizeof(flag));
        for(i=0;i<n;i++)
        {
            scanf("%d%d",&p[i].l,&p[i].r);
            x[tot++]=p[i].l;
            x[tot++]=p[i].r;
        }
  
        sort(x,x+tot);
        tot=unique(x,x+tot)-x;
        for(i=tot-1;i>0;i--)
        {
            if(x[i]-x[i-1]>1) x[tot++]=x[i-1]+1;
        }
        sort(x,x+tot);
        build(0,tot-1,1);
        for(i=0;i<n;i++)
        {
            int l=lower_bound(x,x+tot,p[i].l)-x;
            int r=lower_bound(x,x+tot,p[i].r)-x;
            update(0,tot-1,1,l,r,i);
        }
        query(0,tot-1,1);
        cout<<ans<<endl;
    }
    return 0;
}

原来搓代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 20005

struct tree
{
    int l,r;
}p[N<<2];
struct node
{
    int val,id,pos;
}h[N<<2];

int ans;
int color[N<<2];
bool flag[N<<2];
int change[N<<2];

bool cmp(node a,node b)
{
    return a.val<b.val;
}
void pushup(int rt)
{
    color[rt]=0;
}
void pushdown(int rt)
{
    if(change[rt]!=-1)
    {
        change[rt<<1]=change[rt<<1|1]=change[rt];
        color[rt<<1]=color[rt<<1|1]=change[rt];
        change[rt]=-1;
    }
}
void build(int l,int r,int rt)
{
    color[rt]=0;
    change[rt]=-1;//初始化
    if(l==r) return;
    int m=(l+r)>>1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
}
void update(int l,int r,int rt,int L,int R,int c)
{
    if(l==L && R==r)
    {
        color[rt]=c;
        change[rt]=c;
        return;
    }
    pushdown(rt);
    int m=(l+r)>>1;
    if(R<=m)
        update(l,m,rt<<1,L,R,c);
    else if(L>m)
        update(m+1,r,rt<<1|1,L,R,c);
    else
    {
        update(l,m,rt<<1,L,m,c);
        update(m+1,r,rt<<1|1,m+1,R,c);
    }
    pushup(rt);
}
void query(int l,int r,int rt)
{
    int m=(l+r)>>1;
    if(color[rt])
    {
        if(!flag[color[rt]])
        {
            ans++;
            flag[color[rt]]=1;
        }
        return;
    }
    if(l==r) return; //忘了加这一句、一直RE
    query(l,m,rt<<1);
    query(m+1,r,rt<<1|1);
}
int main()
{
    int n,i;
    while(scanf("%d",&n)!=EOF)
    {
        ans=0;
        memset(flag,0,sizeof(flag));
        for(i=0;i<n;i++)
        {
            scanf("%d%d",&p[i].l,&p[i].r);
            h[i*2].val=p[i].l;
            h[i*2].id=i;
            h[i*2].pos=0;
            h[i*2+1].val=p[i].r;
            h[i*2+1].id=i;
            h[i*2+1].pos=1;
        }
        /* 离散化 */
        sort(h,h+n*2,cmp);
        int last=-1,total=0;
        for(i=0;i<n*2;i++)
        {
            if(h[i].val!=last) 
            {
                total++;
                if(i && h[i].val-h[i-1].val>1) total++; //注意这里、= =、手动插点
                last=h[i].val;
            }
            if(h[i].val!=total)
            {
                if(h[i].pos) p[h[i].id].r=total;
                else p[h[i].id].l=total;
            }
        }
        build(1,total,1);
        for(i=0;i<n;i++)
        {
            update(1,total,1,p[i].l,p[i].r,i+1);
        }
        query(1,total,1);
        cout<<ans<<endl;
    }
    return 0;
}