[线段树 合并] BZOJ 4399 魔法少女LJJ

1505人阅读 评论(0)

c<=7

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;

inline char nc()
{
static char buf[100000],*p1=buf,*p2=buf;
if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }
return *p1++;
}

{
char c=nc(),b=1;
for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}

int n;
int size,root[400005],fat[400005];
int sum[6000005],ls[6000005],rs[6000005];
double loog[6000005];

namespace Memory{
int Stack[6000005],top;
inline void init(){
for (int i=6000000;i;i--) Stack[++top]=i;
}
inline int New(){
return Stack[top--];
}
inline void Del(int u){
Stack[++top]=u;
sum[u]=loog[u]=0; ls[u]=rs[u]=0;
}
inline int getfat(int u){
return fat[u]==u?u:fat[u]=getfat(fat[u]);
}
}
using namespace Memory;

int sx[400005];
int icnt;

inline int Bin(int x){
return lower_bound(sx+1,sx+icnt+1,x)-sx;
}

inline int update(int u){
loog[u]=loog[ls[u]]+loog[rs[u]];
sum[u]=sum[ls[u]]+sum[rs[u]];
}

inline void insert(int &u,int l,int r,int x,int num)
{
if (!u) u=New();
if (l==r){
loog[u]+=num*log(sx[x]);
sum[u]+=num;
return;
}
int mid=(l+r)>>1;
if (x<=mid)
insert(ls[u],l,mid,x,num);
else
insert(rs[u],mid+1,r,x,num);
update(u);
}

inline void clear(int &u,int l,int r,int a,int b){
if (a==l && b==r)
{
Del(u),u=0;
return;
}
int mid=(l+r)>>1;
if (b<=mid)
clear(ls[u],l,mid,a,b);
else if (mid+1<=a)
clear(rs[u],mid+1,r,a,b);
else
clear(ls[u],l,mid,a,mid),clear(rs[u],mid+1,r,mid+1,b);
update(u);
}

inline int query(int u,int l,int r,int a,int b)
{
if (a==l && b==r) return sum[u];
int mid=(l+r)>>1;
if (b<=mid)
return query(ls[u],l,mid,a,b);
else if (mid+1<=a)
return query(rs[u],mid+1,r,a,b);
else
return query(ls[u],l,mid,a,mid)+query(rs[u],mid+1,r,mid+1,b);
}

inline int rank(int &u,int l,int r,int k)
{
if (l==r) return l;
int mid=(l+r)>>1;
if (k<=sum[ls[u]])
return rank(ls[u],l,mid,k);
else
return rank(rs[u],mid+1,r,k-sum[ls[u]]);
}

inline int merge(int x,int y)
{
if (!x) return y;
if (!y) return x;
ls[x]=merge(ls[x],ls[y]);
rs[x]=merge(rs[x],rs[y]);
update(x);
return x;
}

struct event{
int c,a,b;
}eve[400005];

int main()
{
init();
int Q,order,itmp,ans,u,v,k,x,fx,fy;
freopen("t.in","r",stdin);
freopen("t.out","w",stdout);
for (int i=1;i<=Q;i++)
{
if (eve[i].c==1)
else if (eve[i].c==3 || eve[i].c==4)
else if (eve[i].c==7)
else
}
sort(sx+1,sx+icnt+1);
icnt=unique(sx+1,sx+icnt+1)-sx-1;
for (int i=1;i<=Q;i++)
{
if (eve[i].c==1)
{
++n; fat[n]=n;
x=Bin(eve[i].b);
insert(root[fat[n]],1,icnt,x,1);
}
else if (eve[i].c==2)
{
u=eve[i].a; v=eve[i].b;
fx=getfat(u); fy=getfat(v);
if (fx!=fy)
{
fat[fx]=fy;
root[fy]=merge(root[fx],root[fy]);
}
}
else if (eve[i].c==3)
{
u=getfat(eve[i].a); x=Bin(eve[i].b);
if (x==1) continue;
itmp=query(root[u],1,icnt,1,x-1);
insert(root[u],1,icnt,x,itmp);
clear(root[u],1,icnt,1,x-1);
}
else if (eve[i].c==4)
{
u=getfat(eve[i].a); x=Bin(eve[i].b);
if (x==icnt) continue;
itmp=query(root[u],1,icnt,x+1,icnt);
insert(root[u],1,icnt,x,itmp);
clear(root[u],1,icnt,x+1,icnt);
}
else if (eve[i].c==5)
{
u=getfat(eve[i].a); k=eve[i].b;
ans=sx[rank(root[u],1,icnt,k)];
printf("%d\n",ans);
}
else if (eve[i].c==6)
{
u=getfat(eve[i].a); v=getfat(eve[i].b);
ans=loog[root[u]]>loog[root[v]]?1:0;
printf("%d\n",ans);
}
else if (eve[i].c==7)
{
u=getfat(eve[i].a);
ans=sum[root[u]];
printf("%d\n",ans);
}
}
return 0;
}


个人资料
等级：
访问量： 38万+
积分： 1万+
排名： 1225
文章分类
最新评论