[HDOJ 4967] Handling the Past [线段树]

作为一个服务器，你会收到一些关于一个栈的一些操作，他们分别是push,pop,top。由于网络原因，你收到操作的时间和操作实际发送的时间可能时不同的，甚至会是乱序的。

现在每个操作后边都会带着一个时间戳，所有的时间戳均不相同，代表这个操作时何时发送的。要求你对于收到的每一个top操作，立刻返回那个时刻的栈顶元素应该是什么。

观察可以发现，假设把目前收到的所有的操作都按照时间排序，那么对于时间t，返回的栈顶元素就是满足[l,t]区间内push操作恰好比pop操作的个数多1的最靠右的l。

所以我们用一棵线段树，对于push操作，在对应的时间位置上记录+1，pop操作记录-1。那么这棵线段树应该支持的操作有，点修改，查询满足[l,rr]的和等于x的最大的x。

又因为时间戳的范围时int的，所以我们要先进行离散化...

#include <cstdio>#include <map>#include <algorithm>#include <utility>#include <iostream>using namespace std;inline int in() {char c=getchar();while (c<'0'||c>'9') c=getchar();int ans=0;while (c>='0'&&c<='9') {ans=ans*10+c-'0';c=getchar();}return ans;}struct Query {int kind,value,time;void read() {char s[9];scanf("%s",s);if (s[1]=='u') {kind=1;value=in();time=in();} else if (s[1]=='o') {kind=2;time=in();} else {kind=3;time=in();}}};struct SeqNode {SeqNode *ls,*rs;int sum,rsum;void repair() {sum=ls->sum+rs->sum;rsum=max(ls->rsum+rs->sum,rs->rsum);}};SeqNode b[100000],*bp,*root;Query a[50000];map<int,int>c;map<int,int>d;int e[50001];int n;SeqNode *maketree(int l,int r) {SeqNode *ans=bp++;if (l==r) {ans->ls=ans->rs=NULL;ans->sum=ans->rsum=0;} else {int t=(l+r)/2;ans->ls=maketree(l,t);ans->rs=maketree(t+1,r);ans->repair();}return ans;}void set(SeqNode *from,int l,int r,int i,int x) {if (l==r) {from->sum=from->rsum=x;} else {int t=(l+r)/2;if (i<=t) set(from->ls,l,t,i,x);else set(from->rs,t+1,r,i,x);from->repair();}}pair<int,int> get(SeqNode *from,int l,int r,int rr,int sum) {//printf("%d %d %d %d\n",l,r,rr,sum);if (l==r) {if (from->sum==sum) return make_pair(l,0);else return make_pair(-1,from->sum);}int t=(l+r)/2;if (rr<=t) return get(from->ls,l,t,rr,sum);if (rr==r) {if (from->rsum<sum) return make_pair(-1,from->sum);if (from->rs->rsum>=sum) return get(from->rs,t+1,r,r,sum);else return get(from->ls,l,t,t,sum-from->rs->sum);} else {pair<int,int> tmp=get(from->rs,t+1,r,rr,sum);if (tmp.first!=-1) return tmp;if (from->ls->rsum<sum-tmp.second) return make_pair(-1,tmp.second+from->ls->sum);else return get(from->ls,l,t,t,sum-tmp.second);}}void print(SeqNode *from,int l,int r) {if (l!=r) {int t=(l+r)/2;print(from->ls,l,t);print(from->rs,t+1,r);} else {printf("%d ",from->sum);}}void print() {print(root,1,n);printf("\n");}int main() {int i,cas=1;while (scanf("%d",&n),n) {c.clear();d.clear();for (i=0;i<n;i++) {a[i].read();c[a[i].time]=1;d[a[i].time]=i;}i=1;for (map<int,int>::iterator it=c.begin();it!=c.end();it++,i++) {it->second=i;e[i]=it->first;}bp=b;root=maketree(1,n);printf("Case #%d:\n",cas++);//cerr<<cas<<' '<<n<<endl;//print();for (i=0;i<n;i++) {if (a[i].kind==1) {set(root,1,n,c[a[i].time],1);} else if (a[i].kind==2) {set(root,1,n,c[a[i].time],-1);} else {int tmp=get(root,1,n,c[a[i].time],1).first;if (tmp==-1) printf("-1\n");else {printf("%d\n",a[d[e[tmp]]].value);}}//print();//if (i%100==0) cerr<<cas<<' '<<i<<endl;}}return 0;}