HDU-4553 约会安排 线段树区间更新

　　题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4553

　　POJ3667的加强版。

　　建立两颗线段树，一个是DS区间，另一个是NS区间。那么根据题意，如果是DS的请求，那么首先查找DS的区间，如果有满足的区间就更新DS区间，NS的区间不需要更新。如果是NS的请求，首先看DS区间是否有满足的区间，否则查找NS区间，如果有就同时更新DS区间和NS区间。那么可以归纳为，只要是NS的请求，就同时更新两颗线段树，否则只更新DS的线段树。提交后，发现时间效率还是不错的256ms。 PS：写pushdown() 和 pushup()写得眼睛都写花了= =。。

//STATUS:C++_AC_256MS_9458KB
#include <functional>
#include <algorithm>
#include <iostream>
//#include <ext/rope>
#include <fstream>
#include <sstream>
#include <iomanip>
#include <numeric>
#include <cstring>
#include <cassert>
#include <cstdio>
#include <string>
#include <vector>
#include <bitset>
#include <queue>
#include <stack>
#include <cmath>
#include <ctime>
#include <list>
#include <set>
#include <map>
using namespace std;
//using namespace __gnu_cxx;
//define
#define pii pair<int,int>
#define mem(a,b) memset(a,b,sizeof(a))
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define PI acos(-1.0)
//typedef
typedef __int64 LL;
typedef unsigned __int64 ULL;
//const
const int N= 100010;
const int INF=0x3f3f3f3f;
const int MOD=100000,STA=8000010;
const LL LNF=1LL<<60;
const double EPS=1e-8;
const double OO=1e15;
const int dx[4]={-1,0,1,0};
const int dy[4]={0,1,0,-1};
const int day[13]={0,31,28,31,30,31,30,31,31,30,31,30,31};
//Daily Use ...
inline int sign(double x){return (x>EPS)-(x<-EPS);}
template<class T> T gcd(T a,T b){return b?gcd(b,a%b):a;}
template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}
template<class T> inline T lcm(T a,T b,T d){return a/d*b;}
template<class T> inline T Min(T a,T b){return a<b?a:b;}
template<class T> inline T Max(T a,T b){return a>b?a:b;}
template<class T> inline T Min(T a,T b,T c){return min(min(a, b),c);}
template<class T> inline T Max(T a,T b,T c){return max(max(a, b),c);}
template<class T> inline T Min(T a,T b,T c,T d){return min(min(a, b),min(c,d));}
template<class T> inline T Max(T a,T b,T c,T d){return max(max(a, b),max(c,d));}
//End

int lsum[N<<2][2],msum[N<<2][2],rsum[N<<2][2],c[N<<2][2];
int T,n,m,a,b,val;

void pushdown(int rt,int dl,int dr,int p)
{
int ls=rt<<1,rs=rt<<1|1;
if(c[rt][0]!=-1){
c[ls][0]=c[rs][0]=c[rt][0];
lsum[ls][0]=msum[ls][0]=rsum[ls][0]=c[rt][0]?0:dl;
lsum[rs][0]=msum[rs][0]=rsum[rs][0]=c[rt][0]?0:dr;
}
if(c[rt][1]!=-1 && p){
c[ls][1]=c[rs][1]=c[rt][1];
lsum[ls][1]=msum[ls][1]=rsum[ls][1]=c[rt][1]?0:dl;
lsum[rs][1]=msum[rs][1]=rsum[rs][1]=c[rt][1]?0:dr;
}
}

void pushup(int rt,int dl,int dr,int p)
{
int ls=rt<<1,rs=rt<<1|1;
c[rt][0]=c[ls][0]==c[rs][0]?c[ls][0]:-1;
msum[rt][0]=Max(msum[ls][0],msum[rs][0],rsum[ls][0]+lsum[rs][0]);
lsum[rt][0]=lsum[ls][0]==dl?dl+lsum[rs][0]:lsum[ls][0];
rsum[rt][0]=rsum[rs][0]==dr?dr+rsum[ls][0]:rsum[rs][0];
if(p){
c[rt][1]=c[ls][1]==c[rs][1]?c[ls][1]:-1;
msum[rt][1]=Max(msum[ls][1],msum[rs][1],rsum[ls][1]+lsum[rs][1]);
lsum[rt][1]=lsum[ls][1]==dl?dl+lsum[rs][1]:lsum[ls][1];
rsum[rt][1]=rsum[rs][1]==dr?dr+rsum[ls][1]:rsum[rs][1];
}
}

void build(int l,int r,int rt)
{
lsum[rt][0]=msum[rt][0]=rsum[rt][0]=r-l+1;
lsum[rt][1]=msum[rt][1]=rsum[rt][1]=r-l+1;
if(l==r)return;
int mid=(l+r)>>1;
build(lson);
build(rson);
}

void update(int l,int r,int rt,int p)
{
if(a<=l && r<=b){
c[rt][0]=val;
lsum[rt][0]=msum[rt][0]=rsum[rt][0]=val?0:r-l+1;
if(p){
c[rt][1]=val;
lsum[rt][1]=msum[rt][1]=rsum[rt][1]=val?0:r-l+1;
}
return;
}
int mid=(l+r)>>1;
pushdown(rt,mid-l+1,r-mid,p);
if(a<=mid)update(lson,p);
if(b>mid)update(rson,p);
pushup(rt,mid-l+1,r-mid,p);
}

int query(int l,int r,int rt,int p)
{
if(l==r)return l;
int mid=(l+r)>>1;
pushdown(rt,mid-l+1,r-mid,p);
if(msum[rt<<1][p]>=a)return query(lson,p);
else if(rsum[rt<<1][p]+lsum[rt<<1|1][p]>=a)return mid-rsum[rt<<1][p]+1;
else return query(rson,p);
}

int main()
{
//   freopen("in.txt","r",stdin);
int i,j,s,ca=1;
char op[10];
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&m);
mem(c,0);
build(1,n,1);
printf("Case %d:\n",ca++);
while(m--){
scanf("%s",op);
if(op[0]=='D'){
scanf("%d",&a);
if(msum[1][0]>=a){
s=query(1,n,1,0);
printf("%d,let's fly\n",s);
b=s+a-1;a=s;val=1;
update(1,n,1,0);
}
else puts("fly with yourself");
}
else if(op[0]=='N'){
scanf("%d",&a);
if(msum[1][0]>=a || msum[1][1]>=a){
s=query(1,n,1,msum[1][0]<a);
printf("%d,don't put my gezi\n",s);
b=s+a-1;a=s;val=1;
update(1,n,1,1);
}
else {
puts("wait for me");
}
}
else {
scanf("%d%d",&a,&b);
val=0;
update(1,n,1,1);
puts("I am the hope of chinese chengxuyuan!!");
}
}
}
return 0;
}