2016安徽省赛 - 木条染色 离散化+线段树

水模板题当时脑袋短路连想都没想啊!!!
补题的时候居然因为一个条件打错超时好几发,太弱太傻比了
离散化敲了很长时间还是要加油啊

#include
#include
#include
using namespace std;
#define lson i<<1
#define rson (i<<1)+1
int maxn,t,n,m;
int va[500000];
int ha[1000005],pre,pre2;
struct node{
    int x,y;
}b[100005];
int a[100005];
void build(int i,int l,int r)
{
       va[i] = 1;
       if(l==r)return;
       build(lson,l,(l+r)/2);
       build(rson,(l+r)/2+1,r);
}
inline void pushUp(int i)
{
     if(va[lson]==va[rson])va[i] = va[lson];
     else va[i] = -1;
}
inline void pushDown(int i)
{
    va[lson] = va[i];
    va[rson] = va[i];
}
void update(int i,int l,int r,int L,int R)
{
     if(l==L&&r==R)
     {
         va[i] = 0;
         return ;
     }
     if(va[i]!=-1)pushDown(i);
     if(R<=(l+r)/2)update(lson,l,(l+r)/2,L,R);
     else if(L>=(l+r)/2+1)update(rson,(l+r)/2+1,r,L,R);
     else {
         update(lson,l,(l+r)/2,L,(l+r)/2);
         update(rson,(l+r)/2+1,r,(l+r)/2+1,R);
     }
     pushUp(i);
}
int  query(int i,int l,int r,int L,int R)
{
     if(va[i]!=-1)
     {
         return va[i]; 
     }
     if(R<=(l+r)/2)return query(lson,l,(l+r)/2,L,R);
     else if(L>=(l+r)/2+1)return query(rson,(l+r)/2+1,r,L,R);
     else return query(lson,l,(l+r)/2,L,(l+r)/2) + query(rson,(l+r)/2+1,r,(l+r)/2+1,R);
}
inline void init()
{
        pre = 2;
        sort(a,a+2*n);
        ha[a[0]] = pre;
        for(int i=1;i<2*n;i++)
            if(a[i]>a[i-1])
            {
                pre+=2;
                ha[a[i]] = pre;
            }
        pre2 = 1;
        for(int i=max(a[0]-1,0);i<=a[2*n-1]+2;i++)
           if(ha[i])
           {
                  pre2 = ha[i]+1;
           }
           else ha[i] = pre2;
}
int main() 
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(int i=0;iscanf("%d%d",&a[i*2],&a[i*2+1]);
            b[i].x = a[i*2];
            b[i].y = a[i*2+1];
        }
        init();
        build(1,1,pre2);
        for(int i=0;i1,1,pre2,ha[b[i].x],ha[b[i].y]);
        for(int i=0;iint c,d;
            c = max(c,max(0,a[0]-1));
            d = min(a[2*i-1]+2,d);
            scanf("%d%d",&c,&d);
            printf("%d\n",query(1,1,pre2,ha[c],ha[d]));
        }
        for(int i=max(a[0]-1,0);i<=a[2*n-1]+2;i++)
            ha[i] = 0;
        if(t!=0)printf("\n");
    }
    return 0;
}

你可能感兴趣的:(线段树,离散化,线段树)