[CODEVS1743]反转卡片（splay）

题目描述

题解

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;

const int max_n=3e5+5;
const int INF=1e9;

int n,root,sz,cnt;
int a[max_n],f[max_n],ch[max_n][2],size[max_n],key[max_n],delta[max_n];
bool flag;

inline int get(int x)
{
return ch[f[x]][1]==x;
}
inline void update(int x)
{
if (x)
{
size[x]=1;
if (ch[x][0]) size[x]+=size[ch[x][0]];
if (ch[x][1]) size[x]+=size[ch[x][1]];
}
}
inline void pushdown(int x)
{
if (x&&delta[x])
{
swap(ch[x][0],ch[x][1]);
if (ch[x][0]) delta[ch[x][0]]^=1;
if (ch[x][1]) delta[ch[x][1]]^=1;
delta[x]=0;
}
}
inline int build(int l,int r,int fa)
{
int mid=(l+r)>>1;
if (l>r) return 0;
int now=++sz;
key[now]=a[mid]; f[now]=fa; delta[now]=0;
int lch=build(l,mid-1,now);
int rch=build(mid+1,r,now);
ch[now][0]=lch; ch[now][1]=rch;
update(now);
return now;
}
inline void rotate(int x)
{
pushdown(f[x]);
pushdown(x);
int old=f[x],oldf=f[old],which=get(x);
ch[old][which]=ch[x][which^1];
f[ch[old][which]]=old;
ch[x][which^1]=old;
f[old]=x;
if (oldf)
ch[oldf][ch[oldf][1]==old]=x;
f[x]=oldf;
update(old);
update(x);
}
inline void splay(int x,int tar)
{
for (int fa;(fa=f[x])!=tar;rotate(x))
if (f[fa]!=tar)
rotate((get(x)==get(fa))?fa:x);
if (!tar) root=x;
}
inline int find(int x)
{
int now=root;
while (1)
{
pushdown(now);
if (ch[now][0]&&size[ch[now][0]]>=x) now=ch[now][0];
else
{
int temp=1;
if (ch[now][0]) temp+=size[ch[now][0]];
if (x<=temp) return now;
x-=temp;
now=ch[now][1];
}
}
}
int main()
{
scanf("%d",&n);
a[1]=-INF; a[n+2]=INF;
for (int i=1;i<=n;++i) scanf("%d",&a[i+1]);
root=build(1,n+2,0);
flag=false;
while (cnt<100000)
{
int aa=find(2);
int t;
if ((t=key[aa])!=1)
{
cnt++;
int aa=find(1);
int bb=find(t+2);
splay(aa,0);
splay(bb,aa);
delta[ch[ch[root][1]][0]]^=1;
}
else
{
flag=true;
break;
}
}
if (flag) printf("%d\n",cnt);
else printf("-1\n");
}

总结