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 }
