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()写得眼睛都写花了= =。。

  1 //STATUS:C++_AC_256MS_9458KB

  2 #include <functional>

  3 #include <algorithm>

  4 #include <iostream>

  5 //#include <ext/rope>

  6 #include <fstream>

  7 #include <sstream>

  8 #include <iomanip>

  9 #include <numeric>

 10 #include <cstring>

 11 #include <cassert>

 12 #include <cstdio>

 13 #include <string>

 14 #include <vector>

 15 #include <bitset>

 16 #include <queue>

 17 #include <stack>

 18 #include <cmath>

 19 #include <ctime>

 20 #include <list>

 21 #include <set>

 22 #include <map>

 23 using namespace std;

 24 //using namespace __gnu_cxx;

 25 //define

 26 #define pii pair<int,int>

 27 #define mem(a,b) memset(a,b,sizeof(a))

 28 #define lson l,mid,rt<<1

 29 #define rson mid+1,r,rt<<1|1

 30 #define PI acos(-1.0)

 31 //typedef

 32 typedef __int64 LL;

 33 typedef unsigned __int64 ULL;

 34 //const

 35 const int N= 100010;

 36 const int INF=0x3f3f3f3f;

 37 const int MOD=100000,STA=8000010;

 38 const LL LNF=1LL<<60;

 39 const double EPS=1e-8;

 40 const double OO=1e15;

 41 const int dx[4]={-1,0,1,0};

 42 const int dy[4]={0,1,0,-1};

 43 const int day[13]={0,31,28,31,30,31,30,31,31,30,31,30,31};

 44 //Daily Use ...

 45 inline int sign(double x){return (x>EPS)-(x<-EPS);}

 46 template<class T> T gcd(T a,T b){return b?gcd(b,a%b):a;}

 47 template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}

 48 template<class T> inline T lcm(T a,T b,T d){return a/d*b;}

 49 template<class T> inline T Min(T a,T b){return a<b?a:b;}

 50 template<class T> inline T Max(T a,T b){return a>b?a:b;}

 51 template<class T> inline T Min(T a,T b,T c){return min(min(a, b),c);}

 52 template<class T> inline T Max(T a,T b,T c){return max(max(a, b),c);}

 53 template<class T> inline T Min(T a,T b,T c,T d){return min(min(a, b),min(c,d));}

 54 template<class T> inline T Max(T a,T b,T c,T d){return max(max(a, b),max(c,d));}

 55 //End

 56 

 57 int lsum[N<<2][2],msum[N<<2][2],rsum[N<<2][2],c[N<<2][2];

 58 int T,n,m,a,b,val;

 59 

 60 void pushdown(int rt,int dl,int dr,int p)

 61 {

 62     int ls=rt<<1,rs=rt<<1|1;

 63     if(c[rt][0]!=-1){

 64         c[ls][0]=c[rs][0]=c[rt][0];

 65         lsum[ls][0]=msum[ls][0]=rsum[ls][0]=c[rt][0]?0:dl;

 66         lsum[rs][0]=msum[rs][0]=rsum[rs][0]=c[rt][0]?0:dr;

 67     }

 68     if(c[rt][1]!=-1 && p){

 69         c[ls][1]=c[rs][1]=c[rt][1];

 70         lsum[ls][1]=msum[ls][1]=rsum[ls][1]=c[rt][1]?0:dl;

 71         lsum[rs][1]=msum[rs][1]=rsum[rs][1]=c[rt][1]?0:dr;

 72     }

 73 }

 74 

 75 void pushup(int rt,int dl,int dr,int p)

 76 {

 77     int ls=rt<<1,rs=rt<<1|1;

 78     c[rt][0]=c[ls][0]==c[rs][0]?c[ls][0]:-1;

 79     msum[rt][0]=Max(msum[ls][0],msum[rs][0],rsum[ls][0]+lsum[rs][0]);

 80     lsum[rt][0]=lsum[ls][0]==dl?dl+lsum[rs][0]:lsum[ls][0];

 81     rsum[rt][0]=rsum[rs][0]==dr?dr+rsum[ls][0]:rsum[rs][0];

 82     if(p){

 83         c[rt][1]=c[ls][1]==c[rs][1]?c[ls][1]:-1;

 84         msum[rt][1]=Max(msum[ls][1],msum[rs][1],rsum[ls][1]+lsum[rs][1]);

 85         lsum[rt][1]=lsum[ls][1]==dl?dl+lsum[rs][1]:lsum[ls][1];

 86         rsum[rt][1]=rsum[rs][1]==dr?dr+rsum[ls][1]:rsum[rs][1];

 87     }

 88 }

 89 

 90 void build(int l,int r,int rt)

 91 {

 92     lsum[rt][0]=msum[rt][0]=rsum[rt][0]=r-l+1;

 93     lsum[rt][1]=msum[rt][1]=rsum[rt][1]=r-l+1;

 94     if(l==r)return;

 95     int mid=(l+r)>>1;

 96     build(lson);

 97     build(rson);

 98 }

 99 

100 void update(int l,int r,int rt,int p)

101 {

102     if(a<=l && r<=b){

103         c[rt][0]=val;

104         lsum[rt][0]=msum[rt][0]=rsum[rt][0]=val?0:r-l+1;

105         if(p){

106             c[rt][1]=val;

107             lsum[rt][1]=msum[rt][1]=rsum[rt][1]=val?0:r-l+1;

108         }

109         return;

110     }

111     int mid=(l+r)>>1;

112     pushdown(rt,mid-l+1,r-mid,p);

113     if(a<=mid)update(lson,p);

114     if(b>mid)update(rson,p);

115     pushup(rt,mid-l+1,r-mid,p);

116 }

117 

118 int query(int l,int r,int rt,int p)

119 {

120     if(l==r)return l;

121     int mid=(l+r)>>1;

122     pushdown(rt,mid-l+1,r-mid,p);

123     if(msum[rt<<1][p]>=a)return query(lson,p);

124     else if(rsum[rt<<1][p]+lsum[rt<<1|1][p]>=a)return mid-rsum[rt<<1][p]+1;

125     else return query(rson,p);

126 }

127 

128 int main()

129 {

130  //   freopen("in.txt","r",stdin);

131     int i,j,s,ca=1;

132     char op[10];

133     scanf("%d",&T);

134     while(T--)

135     {

136         scanf("%d%d",&n,&m);

137         mem(c,0);

138         build(1,n,1);

139         printf("Case %d:\n",ca++);

140         while(m--){

141             scanf("%s",op);

142             if(op[0]=='D'){

143                 scanf("%d",&a);

144                 if(msum[1][0]>=a){

145                     s=query(1,n,1,0);

146                     printf("%d,let's fly\n",s);

147                     b=s+a-1;a=s;val=1;

148                     update(1,n,1,0);

149                 }

150                 else puts("fly with yourself");

151             }

152             else if(op[0]=='N'){

153                 scanf("%d",&a);

154                 if(msum[1][0]>=a || msum[1][1]>=a){

155                     s=query(1,n,1,msum[1][0]<a);

156                     printf("%d,don't put my gezi\n",s);

157                     b=s+a-1;a=s;val=1;

158                     update(1,n,1,1);

159                 }

160                 else {

161                     puts("wait for me");

162                 }

163             }

164             else {

165                 scanf("%d%d",&a,&b);

166                 val=0;

167                 update(1,n,1,1);

168                 puts("I am the hope of chinese chengxuyuan!!");

169             }

170         }

171     }

172     return 0;

173 }