您的位置:首页 > 其它

bzoj 3123 可持久化线段树启发式合并

2014-04-21 12:56 393 查看
  首先没有连边的操作的时候,我们可以用可持久化线段树来维护这棵树的信息,建立权值可持久化线段树,那么每个点继承父节点的线段树,当询问为x,y的时候我们可以询问rot[x]+rot[y]-rot[lca(x,y)]-rot[lca(x,y)->father]这棵树来得知这个链的信息。

  那么对于连边操作,相当于合并两棵树,我们可以将树的节点数小的树全部拆掉连到节点大的树中,这样每个点最多会被操作logn次,每次操作的时间复杂度为logn,所以是mlog^2n的。

  反思:对于树的连通性我是用并查集维护的,对于合并操作还需要dfs一次小的树来维护各种信息,但是忘记对x,y点连边了,导致一直RE.(RE是因为某次值不正确,导致下一次^的点超过n)。

     各种慢= =。

/**************************************************************
Problem: 3123
User: BLADEVIL
Language: C++
Result: Accepted
Time:16612 ms
Memory:81512 kb
****************************************************************/

//By BLADEVIL
#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 100010
#define maxm 200010

using namespace std;

struct rec {
int key,num;
rec() {
key=num=0;
}
}a[maxn];

struct segment {
int left,right,cnt;
int son[2];
segment() {
left=right=cnt=0;
memset(son,0,sizeof son);
}
}t[maxn<<5];

int n,m,l,N,tot;
int pre[maxm<<1],last[maxm<<1],other[maxm<<1];
int ans[maxn],rot[maxn];
int dep[maxn],jump[maxn][20],father[maxn],size[maxn],que[maxn];

void connect(int x,int y) {
pre[++l]=last[x];
last[x]=l;
other[l]=y;
}

int getfather(int x) {
if (father[x]==x) return x;
return father[x]=getfather(father[x]);
}

void dfs(int x,int fa) {
jump[x][0]=fa; dep[x]=dep[fa]+1;
for (int p=last[x];p;p=pre[p]) {
if (other[p]==fa) continue;
dfs(other[p],x);
}
}

bool cmp1(rec x,rec y) {
return x.key<y.key;
}

bool cmp2(rec x,rec y) {
return x.num<y.num;
}

int lca(int x,int y) {
if (dep[x]>dep[y]) swap(x,y);
int d=dep[y]-dep[x];
for (int i=0;i<=16;i++) if (d&(1<<i)) y=jump[y][i];
if (x==y) return x;
for (int i=16;i>=0;i--) if (jump[x][i]!=jump[y][i]) x=jump[x][i],y=jump[y][i];
return jump[x][0];
}

void build(int &x,int l,int r) {
if (!x) x=++tot;
t[x].left=l; t[x].right=r;
if (t[x].left==t[x].right) return ;
int mid=t[x].left+t[x].right>>1;
build(t[x].son[0],l,mid); build(t[x].son[1],mid+1,r);
}

void insert(int &x,int rot,int key) {
if (!x) x=++tot;
t[x].left=t[rot].left; t[x].right=t[rot].right;
if (t[x].left==t[x].right) {
t[x].cnt=t[rot].cnt+1;
return ;
}
int mid=t[x].left+t[x].right>>1;
if (key>mid) {
t[x].son[0]=t[rot].son[0];
insert(t[x].son[1],t[rot].son[1],key);
} else {
t[x].son[1]=t[rot].son[1];
insert(t[x].son[0],t[rot].son[0],key);
}
t[x].cnt=t[rot].cnt+1;
}

int query(int x,int y,int l1,int l2,int k) {
//printf("%d %d %d ",x,t[x].left,t[x].right);
if (t[x].left==t[x].right) return t[x].left;
int cur=t[t[x].son[0]].cnt+t[t[y].son[0]].cnt-t[t[l1].son[0]].cnt-t[t[l2].son[0]].cnt;
//printf("%d\n",cur);
if (k>cur) return query(t[x].son[1],t[y].son[1],t[l1].son[1],t[l2].son[1],k-cur); else
return query(t[x].son[0],t[y].son[0],t[l1].son[0],t[l2].son[0],k);
}

void make(int x,int fa) {
insert(rot[x],rot[fa],a[x].key);
for (int p=last[x];p;p=pre[p]) {
if (other[p]==fa) continue;
make(other[p],x);
}
}

void update(int x,int fa,int j) {
jump[x][j]=jump[jump[x][j-1]][j-1];
for (int p=last[x];p;p=pre[p]) {
if (other[p]==fa) continue;
update(other[p],x,j);
}
}

void merge(int x,int y) {
int fx=getfather(x),fy=getfather(y);
//if (x==1) printf("%d %d\n",fx,fy);
if (size[fx]<size[fy]) swap(x,y);
fx=getfather(x),fy=getfather(y);
father[fy]=fx; size[fx]+=size[fy];
dfs(y,x); make(y,x);
for (int i=1;i<=16;i++) update(y,x,i);
//for (int i=1;i<=n;i++) printf("%d %d\n",i,jump[i][0]);
}

int main() {
//freopen("3123.in","r",stdin); freopen("3123.out","w",stdout);
int task; scanf("%d",&task);
scanf("%d%d%d",&n,&m,&task);
for (int i=1;i<=n;i++) scanf("%d",&a[a[i].num=i].key);

sort(a+1,a+1+n,cmp1);
int cur; ans[1]=cur=a[1].key;
for (int i=1,j=1;i<=n;i++)
if (a[i].key==cur) a[i].key=j; else ans[++j]=cur=a[i].key,a[i].key=j,N=j;
sort(a+1,a+1+n,cmp2);
//for (int i=1;i<=n;i++) printf("%d ",ans[i]); printf("\n");
//for (int i=1;i<=n;i++) printf("%d ",a[i].key); printf("\n");

for (int i=1;i<=n;i++) father[i]=i,size[i]=1;
for (int i=1;i<=m;i++) {
int x,y; scanf("%d%d",&x,&y);
connect(x,y); connect(y,x);
int fx=getfather(x),fy=getfather(y);
father[fx]=fy; size[fy]+=size[fx];
}
for (int i=1;i<=n;i++) if (!jump[i][0]) dfs(i,0);
//for (int i=1;i<=n;i++) printf("%d %d\n",i,father[i]);
//for (int i=1;i<=n;i++) if (i==getfather(i)) printf("%d %d\n",i,size[i]);

for (int j=1;j<=16;j++)
for (int i=1;i<=n;i++) jump[i][j]=jump[jump[i][j-1]][j-1];

build(rot[0],1,N);
for (int i=1;i<=n;i++) if (!jump[i][0]) make(i,0);

int ANS=0; char Q[3];
while (task--) {
scanf("%s",Q);
if (Q[0]=='Q') {
int x,y,k; scanf("%d%d%d",&x,&y,&k);
x^=ANS; y^=ANS; k^=ANS;
//printf("|%d %d %d\n",x,y,k);
int rr=lca(x,y); //printf("%d\n",rr);
//for (int i=1;i<=n;i++) printf("%d %d\n",i,jump[i][0]);
printf("%d\n",ANS=ans[query(rot[x],rot[y],rot[rr],rot[jump[rr][0]],k)]);
//for (int i=1;i<=n;i++) if (i==getfather(i)) printf("%d %d\n",i,size[i]);
} else {
int x,y; scanf("%d%d",&x,&y);
x^=ANS; y^=ANS; connect(x,y); connect(y,x);
//printf("|%d %d\n",x,y);
merge(x,y);
}
}
return 0;
}


 
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: