HDU4553 约会安排

http://www.mamicode.com/info-detail-422707.html

线段树区间覆盖，开两个线段树，一个记录DS，一个NS

// #pragma comment(linker, "/STACK:1024000000,1024000000")
#include <iostream>
#include <cstdio>
#include <cstring>
#include <sstream>
#include <string>
#include <algorithm>
#include <list>
#include <map>
#include <vector>
#include <queue>
#include <stack>
#include <cmath>
#include <cstdlib>
#include <conio.h>
using namespace std;
#define clc(a,b) memset(a,b,sizeof(a))
#define inf 0x3f3f3f3f
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
const int N = 100010;
const int MOD = 1e9+7;
#define LL long long
#define mi() (l+r)>>1
double const pi = acos(-1);
void fre() {
    freopen("in.txt","r",stdin);
}
// inline int r() {
//     int x=0,f=1;char ch=getchar();
//     while(ch>'9'||ch<'0') {if(ch=='-') f=-1;ch=getchar();}
//     while(ch>='0'&&ch<='9') { x=x*10+ch-'0';ch=getchar();}return x*f;
// }
struct Edge {
    int l,r;
    int lazy;
    int lm,rm,mm;
} e1[N<<2],e2[N<<2];

void pushdown(Edge e[],int rt) {
    if(e[rt].lazy!=-1) {
        if(e[rt].lazy==0) {
            e[rt<<1].lm=e[rt<<1].rm=e[rt<<1].mm=e[rt<<1].r-e[rt<<1].l+1;
            e[rt<<1|1].rm=e[rt<<1|1].lm=e[rt<<1|1].mm=e[rt<<1|1].r-e[rt<<1|1].l+1;
            e[rt<<1].lazy=e[rt<<1|1].lazy=0;
        } else {
            e[rt<<1].lm=e[rt<<1].rm=e[rt<<1].mm=0;
            e[rt<<1|1].rm=e[rt<<1|1].lm=e[rt<<1|1].mm=0;
            e[rt<<1].lazy=e[rt<<1|1].lazy=1;
        }
        e[rt].lazy=-1;
    }
}

void pushup(Edge e[],int rt) {
    e[rt].lm=e[rt<<1].lm;
    e[rt].rm=e[rt<<1|1].rm;
    if(e[rt<<1].lm==e[rt<<1].r-e[rt<<1].l+1) {
        e[rt].lm+=e[rt<<1|1].lm;
    }  
    if(e[rt<<1|1].rm==e[rt<<1|1].r-e[rt<<1|1].l+1) {
        e[rt].rm+=e[rt<<1].rm;
    }
    e[rt].mm=max(max(e[rt<<1].mm,e[rt<<1|1].mm),e[rt<<1].rm+e[rt<<1|1].lm);
}
void build(int l,int r,int rt,Edge a[]) {
    a[rt].l=l;
    a[rt].r=r;
    a[rt].lm=a[rt].rm=a[rt].mm=(r-l+1);
    a[rt].lazy=-1;
    if(l==r)
        return;
    int mid=mi();
    build(lson,a);
    build(rson,a);
}

void update(Edge e[],int l,int r,int rt,int x) {
    if(e[rt].l==l&&e[rt].r==r) {
        if(x==0) {
            e[rt].lm=e[rt].rm=e[rt].mm=r-l+1;
            e[rt].lazy=0;
        } else {
            e[rt].lm=e[rt].rm=e[rt].mm=0;
            e[rt].lazy=1;
        }
        return;
    }
    pushdown(e,rt);
    int mid=(e[rt].l+e[rt].r)>>1;
    if(r<=mid) update(e,l,r,rt<<1,x);
    else if(l>mid) update(e,l,r,rt<<1|1,x);
    else {
        update(e,l,mid,rt<<1,x);
        update(e,mid+1,r,rt<<1|1,x);
    }
    pushup(e,rt);
}

int query(Edge e[],int rt,int c) {
    if(e[rt].l==e[rt].r) {
        return e[rt].l;
    }
    pushdown(e,rt);
    int mid=(e[rt].l+e[rt].r)>>1;
    if(e[rt<<1].mm>=c) {
        return query(e,rt<<1,c);
    } else if(e[rt<<1].rm+e[rt<<1|1].lm>=c) {
        return mid-e[rt<<1].rm+1;
    } else {
        return query(e,rt<<1|1,c);
    }
}
int main() {
    // fre();
    int T;
    int cas=1;
    cin>>T;
    while(T--) {
        printf("Case %d:
",cas++);
        int n,q;
        cin>>n>>q;
        build(1,n,1,e1);//D
        build(1,n,1,e2);//N
        while(q--) {
            char s[20];
            int x,y;
            scanf("%s",s);
            if(s[0]=='S') {
                cin>>x>>y;
                update(e1,x,y,1,0);
                update(e2,x,y,1,0);
                printf("I am the hope of chinese chengxuyuan!!
");
            } else if(s[0]=='D') {
                cin>>x;
                if(e1[1].mm<x) {
                    printf("fly with yourself
");
                } else {
                    int s=query(e1,1,x);
                    printf("%d,let's fly
",s);
                    update(e1,s,s+x-1,1,1);
                }
            } else {
                cin>>x;
                if(e1[1].mm>=x) {
                    int s=query(e1,1,x);
                    printf("%d,don't put my gezi
",s);
                    update(e1,s,s+x-1,1,1);
                    update(e2,s,s+x-1,1,1);
                } else if(e2[1].mm>=x) {
                    int s=query(e2,1,x);
                    printf("%d,don't put my gezi
",s);
                    update(e1,s,s+x-1,1,1);
                    update(e2,s,s+x-1,1,1);
                } else {
                    printf("wait for me
");
                }
            }
        }
    }
    return 0;
}