BZOJ 1594: [Usaco2008 Jan]猜数游戏 线段树 + 思维 + 二分

Code:

#include<bits/stdc++.h>
#define maxn 3000000
using namespace std;
void setIO(string s)
{
    string in=s+".in"; 
    freopen(in.c_str(),"r",stdin); 
}
int n,Q;   
struct OPT
{
    int l,r,v;    
}opt[maxn],yy[maxn]; 
bool cmp(OPT a,OPT b)
{
    return a.v>b.v; 
}
struct Seg
{
    #define lson (x<<1)
    #define rson ((x<<1)|1)
    int lazy[maxn<<2],sumv[maxn<<2]; 
    void re()
    {
        memset(lazy,0,sizeof(lazy)); 
        memset(sumv,0,sizeof(sumv));    
    }
    void mark(int l,int r,int x)
    {        
        sumv[x]=r-l+1; 
        lazy[x]=1;     
    }
    void pushdown(int l,int r,int x)
    {
        if(lazy[x])
        {
            int mid=(l+r)>>1; 
            if(l<=mid) mark(l,mid,lson); 
            if(r>mid) mark(mid+1,r,rson); 
            lazy[x]=0; 
        }
    }
    void update(int l,int r,int x,int L,int R)
    {
        if(l>=L&&r<=R) 
        {
            mark(l,r,x); 
            return; 
        }
        pushdown(l,r,x); 
        int mid=(l+r)>>1; 
        if(L<=mid) update(l,mid,lson,L,R); 
        if(R>mid) update(mid+1,r,rson,L,R); 
        sumv[x]=sumv[lson]+sumv[rson]; 
    }
    int query(int l,int r,int x,int L,int R)
    {
        if(l>=L&&r<=R) return sumv[x]; 
        pushdown(l,r,x); 
        int mid=(l+r)>>1,tmp=0; 
        if(L<=mid) tmp+=query(l,mid,lson,L,R); 
        if(R>mid)  tmp+=query(mid+1,r,rson,L,R); 
        return tmp;    
    }   
}tr; 
bool check(int mid)
{
    int i,j,k,l1,r1,l2,r2;             
    tr.re(); 
    for(i=1;i<=mid;++i) yy[i]=opt[i]; 
    sort(yy+1,yy+1+mid,cmp); 
    for(i=1;i<=mid;i=j)
    {
        for(j=i;j<=mid&&yy[j].v==yy[i].v;++j); 
        l1=l2=yy[i].l;
        r1=r2=yy[i].r; 
        for(k=i+1;k<j;++k)
        {
            l1=min(l1,yy[k].l); 
            l2=max(l2,yy[k].l); 
            r1=max(r1,yy[k].r); 
            r2=min(r2,yy[k].r); 
        }
        if(l2>r2) return true;     // 无并集
        if(tr.query(1,n,1,l2,r2)==r2-l2+1) return true; 
        tr.update(1,n,1,l1,r1);    
    }
    return false; 
}
int main()
{
    // setIO("input");          
    scanf("%d%d",&n,&Q); 
    for(int i=1;i<=Q;++i)
    {
        scanf("%d%d%d",&opt[i].l,&opt[i].r,&opt[i].v); 
    }    
    int l=1,r=Q,ans=0; 
    while(l<=r)
    {
        int mid=(l+r)>>1; 
        if(check(mid)) 
            r = mid - 1, ans=mid; 
        else 
            l=mid+1; 
    }  
    printf("%d\n",ans); 
    return 0; 
}