[JZOJ5082].【GDSOI2017第三轮模拟】Informatics Training

题目描述

操作数<=300000

分析

首先这是一道数据结构裸题。

面对这种码农题我们需要有较清晰的思路。

下面是我打题前的思路：

大概是用splay维护h，s，关键字就是编号。然后两个set维护其他东西。

T3

set:

mntr 维护每个splay的根的mn_s ，以及id ，就一个pair嘛

ntr 维护点编号

splay:

vtr维护

mn_h

mn_s

sum_h

siz

tag_s

tag_h

join 新建splay，点值s加入set1

practice ， splay i ， 减x，加y

discuss 给 splay i ，整体减x加y

判断体力是否低于0：

当前节点x低于， 那么左右子树merge，自己扔掉，

若左右子树中有mn低于0，那么继续递归

cooperate，去set2中找最大点和最小点，

test，set1中找到最小值，然后find(mn)，并删除。

query直接输出。

注意每次操作完之前后，splay， set维护。

splay至根之前，删掉 mntr，旋完之后再加入

任何遍历都要update，down

然而还是因为不够清晰打错了不少地方，特别是由于mntr维护的是splay根们的信息，每次splay的时候要先删去，之后再加上。

其实这种题想到做法并不难，重点是要优化代码，像我这种三个数据结构，又臭又长。而王之栋的做法十分优美，只用了2个set。虽然我们都超时了····

而鞋垫的treap跑得飞起。

值得一提的是set的operator重定义，必须要把多个关键字都考虑进去，不然会find错····

代码

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<set>
using namespace std;
typedef long long ll;
typedef double db;
#define fo(i,j,k) for(i=j;i<=k;i++)
#define fd(i,j,k) for(i=j;i>=k;i--)
const int N=300005;
struct pi
{
int val,id;
pi(int i_=0,int v_=0) {val=v_,id=i_;}
}tmp;
bool operator <(pi a,pi b)
{
return a.val<b.val||a.val==b.val&&a.id<b.id;
}
struct rec
{
int mn_h,mn_s,siz,h,s,tag_h;
ll sum,tag_s;
}vtr
;
int tr
[2],fa
,st,sta
;
int n,x,y,z,t,l,m,kan,tp,k1;
char ch;
multiset<pi> mntr;
set<int> :: iterator it;
set<pi> :: iterator kk;
set<int> ntr;
void down(int x)
{
if (!x||(!vtr[x].tag_h&&!vtr[x].tag_s)) return;
vtr[x].mn_s+=vtr[x].tag_s;
vtr[x].s+=vtr[x].tag_s;
vtr[x].mn_h+=vtr[x].tag_h;
vtr[x].h+=vtr[x].tag_h;
vtr[x].sum+=1ll*vtr[x].tag_s*vtr[x].siz;
int i;
fo(i,0,1)
if (tr[x][i])
{
vtr[tr[x][i]].tag_s+=vtr[x].tag_s;
vtr[tr[x][i]].tag_h+=vtr[x].tag_h;
}
vtr[x].tag_h=vtr[x].tag_s=0;
}
void update(int x)
{
if (!x)
return ;
down(x);
down(tr[x][0]);
down(tr[x][1]);
vtr[x].mn_h=vtr[x].h;
vtr[x].mn_s=vtr[x].s;
vtr[x].siz=vtr[tr[x][0]].siz+vtr[tr[x][1]].siz+1;
vtr[x].sum=vtr[tr[x][0]].sum+vtr[tr[x][1]].sum+vtr[x].s;
int i;
fo(i,0,1)
if (tr[x][i])
{
vtr[x].mn_h=min(vtr[x].mn_h,vtr[tr[x][i]].mn_h);
vtr[x].mn_s=min(vtr[x].mn_s,vtr[tr[x][i]].mn_s);
}
}
int pd(int x)
{
return tr[fa[x]][1]==x;
}
void rotate(int x)
{
int y=fa[x],z=pd(x);
down(y);
down(x);
fa[x]=fa[y];
if (fa[y]) tr[fa[y]][pd(y)]=x;
tr[y][z]=tr[x][1-z];
if (tr[x][1-z]) fa[tr[x][1-z]]=y;
fa[y]=x;
tr[x][1-z]=y;
update(y);
update(x);
}
void downdate(int x,int y)
{
st=0;
while (x!=y)
{
sta[++st]=x;
x=fa[x];
}
if (!y)
mntr.erase(mntr.find(pi(sta[st],vtr[sta[st]].mn_s)));
while (st)
{
down(sta[st]);
st--;
}
}
void splay(int x,int y)
{
if (!x)  return;
downdate(x,y);
while (fa[x]!=y)
{
if (fa[fa[x]]!=y)
{
if (pd(x)==pd(fa[x])) rotate(fa[x]);
else rotate(x);
}
rotate(x);
}
if (!y) mntr.insert(pi(x,vtr[x].mn_s));
}
int go(int x,int sig)
{
while (tr[x][sig])
{
update(x);
x=tr[x][sig];
}
update(x);
return x;
}
int merge(int x,int y,int z,int t,int sig)// middle lson,rson, father
{
if (sig&&x)
mntr.erase(mntr.find(pi(x,vtr[x].mn_s)));
y=go(y,1);
z=go(z,0);
if (y||z)
{
if (y) splay(y,x);
if (z) splay(z,x);
if (y)
{
if (z)
{
// erase z
if (sig&&!x)
mntr.erase(mntr.find(pi(z,vtr[z].mn_s)));
fa[z]=y;
}
tr[y][1]=z;
}else y=z;
// erase y
if (sig&&!x)
mntr.erase(mntr.find(pi(y,vtr[y].mn_s)));
update(y);
// insert y
if (sig) mntr.insert(pi(y,vtr[y].mn_s));
}
if (t) tr[t][pd(x)]=y;
fa[y]=t;
if (t) update(t);
return y;
}
int del(int x,int sig)// remember to update fa[x], actually merge tr[x][0\1]
{
it=ntr.find(x);
kan=*it;
ntr.erase(ntr.find(x));
int y=merge(x,tr[x][0],tr[x][1],fa[x],sig);
return y;
}
void Merge(int x,int y)
{
splay(x,0);
splay(y,0);
int z=x;
while (fa[z]!=0) z=fa[z];
if (z==y) return;
z=merge(0,x,y,0,1);
splay(z,0);
}
void join()
{
scanf("%d %d",&x,&y);
n++;
ntr.insert(n);
vtr
.siz=1;
vtr
.mn_h=vtr
.h=x;
vtr
.sum=y;
vtr
.mn_s=y;
vtr
.s=y;
mntr.insert(pi(n,y));
}
void prac()
{
scanf("%d %d %d",&x,&y,&z);
splay(x,0);
mntr.erase(mntr.find(pi(x,vtr[x].mn_s)));
vtr[x].h-=y;
vtr[x].s+=z;
update(x);
mntr.insert(pi(x,vtr[x].mn_s));
if (vtr[x].h<=0) del(x,1);
}
int dfs(int x)// 删除不合法的蒟蒻
{
down(x);
int y;
if (vtr[x].h<=0)
{
y=del(x,0);
if (y) return dfs(y);
}else
{
int i;
fo(i,0,1)
if (tr[x][i])
{
down(tr[x][i]);
if (vtr[tr[x][i]].mn_h<=0) dfs(tr[x][i]);
}
update(x);
return x;
}
}
void diss()
{
scanf("%d %d %d",&x,&y,&z);
splay(x,0);
mntr.erase(mntr.find(pi(x,vtr[x].mn_s)));
vtr[x].tag_h-=y;
vtr[x].tag_s+=z;
update(x);
t=dfs(x);
if (t)
{
mntr.insert(pi(t,vtr[t].mn_s));
splay(t,0);
}
}
void elim()
{
scanf("%d",&x);
splay(x,0);
del(x,1);
}
int find(int x,int y)
{
while (x)
{
update(x);
if (vtr[x].s==y) return x;
if (vtr[tr[x][0]].mn_s==y) x=tr[x][0];else
x=tr[x][1];
}
}
int main()
{
freopen("training.in","r",stdin);
freopen("training.out","w",stdout);
scanf("%d\n",&m);
fo(l,1,m)
{
scanf("%c",&ch);
if (ch=='J') join();else
if (ch=='P') prac();else
if (ch=='D') diss();else
if (ch=='E') elim();else
if (ch=='M')
{
scanf("%d %d",&x,&y);
Merge(x,y);
}else if (ch=='C')
{
kan=ntr.size();
if (!kan) continue;
x=*ntr.begin();
it=ntr.end();
it--;
y=*it;
Merge(x,y);
}else if (ch=='T')
{
kan=ntr.size();
if (!kan) continue;
tmp=(*mntr.begin());
tp=tmp.val;
while (tmp.val==tp)
{
k1=find(tmp.id,tp);
splay(k1,0);
del(k1,1);
tmp=(*mntr.begin());
}
}else
{
scanf("%d",&x);
splay(x,0);
printf("%d %lld\n",vtr[x].siz,vtr[x].sum);
}
scanf("\n");
}
}