codevs1951，sdoi2013森林，启发式合并好题

来源：互联网发布：黑白网络官网编辑：程序博客网时间：2024/06/05 11:35

一开始我没看出来这能启发式合并，看了题解才知道。
然后就是码码码，写完后死活A不掉，这是我的WA代码

#include<bits/stdc++.h>using namespace std; typedef unsigned int ui;inline int getint(){    int x=0;    char c=getchar();    while(!isdigit(c))c=getchar();    for(;isdigit(c);c=getchar())x=x*10+c-48;    return x;}int buf[100];inline void putint(int x){    if(!x)putchar('0');        else{            int xb=0;            for(;x;x/=10)buf[++xb]=x%10;            for(;xb;--xb)putchar(buf[xb]+48);        }}const int N=80010;int n,m,T,xb,i,x,y,z,f[N],ff[N][20],dep[N],s[N],val[N],u,v,tt,rt[N],b[N],l,la,d[N];pair<int,int> a[N];vector<int> g[N];struct node{    int lc,rc,s;}t[N*100];char c;inline void swap(int&a,int&b){    tt=a;    a=b;    b=tt;}void ins(int x,int&y,int l,int r,int v){    y=++xb;    t[y].s=t[x].s+1;    if(l!=r){        int m=(l+r)>>1;        if(v>m){            t[y].lc=t[x].lc;            ins(t[x].rc,t[y].rc,m+1,r,v);        }else{            t[y].rc=t[x].rc;            ins(t[x].lc,t[y].lc,l,m,v);        }    }}void dfs(int x,int fa){    dep[x]=1+dep[*ff[x]=fa];    int y;    for(y=1;1<<y<=dep[x];++y)ff[x][y]=ff[ff[x][y-1]][y-1];    ins(rt[fa],rt[x],1,l,b[x]);    for(ui i=0;i<g[x].size();++i){        y=g[x][i];        if(y!=fa)dfs(y,x);    }}int gfa(int x){    return f[x]==x?x:f[x]=gfa(f[x]);}int lca(int x,int y){      if (dep[x]<dep[y]) swap(x,y); int i,tmp=dep[x]-dep[y];      for (i=0; i<=16; i++) if (tmp&(1<<i)) x=ff[x][i];      for (i=16; i>=0; i--)          if (ff[x][i]!=ff[y][i]){ x=ff[x][i]; y=ff[y][i]; }      return (x==y)?x:ff[x][0];  }  inline void link(int x,int y){    u=gfa(x);    v=gfa(y);    if(s[u]<s[v]){        swap(u,v);        swap(x,y);    }    s[u]+=s[v];    f[v]=u;    g[x].push_back(y);    g[y].push_back(x);    dfs(y,x);}int query(int w,int x,int y,int z,int l,int r,int k){    //printf("Q%d\n",l);    if(l==r)return d[l];    int m=(l+r)>>1,o=t[t[y].lc].s+t[t[z].lc].s-t[t[x].lc].s-t[t[w].lc].s;    if(k>o)return query(t[w].rc,t[x].rc,t[y].rc,t[z].rc,m+1,r,k-o);        else return query(t[w].lc,t[x].lc,t[y].lc,t[z].lc,l,m,k);}int main(){    scanf("%d%d%d%d",&T,&n,&m,&T);    for(i=1;i<=n;++i)a[i]=make_pair(val[f[i]=i]=getint(),i),s[i]=1;    sort(a+1,a+n+1);    for(i=1;i<=n;++i){        if(a[i].first>a[i-1].first)++l;        d[b[a[i].second]=l]=a[i].first;    }    for(i=1;i<=n;++i)ins(0,rt[i],1,l,b[i]);    for(i=1;i<=m;++i){        x=getint();        y=getint();        link(x,y);    }    while(T--){        c=getchar();        while(!isalpha(c))c=getchar();        if(c=='L'){            x=getint()^la;            y=getint()^la;            //printf("a%d %d\n",x,y);            link(x,y);        }else{            x=getint()^la;            y=getint()^la;            z=getint()^la;            //printf("b%d %d %d\n",x,y,z);            i=lca(x,y);            //printf("i%d\n",i);            putint(la=query(rt[*ff[i]],rt[i],rt[x],rt[y],1,l,z));            putchar('\n');        }        //puts("WTF");        //for(i=1;i<=n;++i)printf("%d\n",*ff[i]);    }    return 0;}

事实上，根据这份代码的定义，dep值为3的点第22祖先是不存在的，但事实上此时其第22个祖先就是根节点。
然后无脑从16开始操作，就好了

#include<bits/stdc++.h>using namespace std; typedef unsigned int ui;inline int getint(){    int x=0;    char c=getchar();    while(!isdigit(c))c=getchar();    for(;isdigit(c);c=getchar())x=x*10+c-48;    return x;}int buf[100];inline void putint(int x){    if(!x)putchar('0');        else{            int xb=0;            for(;x;x/=10)buf[++xb]=x%10;            for(;xb;--xb)putchar(buf[xb]+48);        }}const int N=80010;int n,m,T,xb,i,x,y,z,f[N],ff[N][20],dep[N],s[N],u,v,tt,rt[N],b[N],l,la,d[N];pair<int,int> a[N];vector<int> g[N];struct node{    int lc,rc,s;}t[N*400];char c;inline void swap(int&a,int&b){    tt=a;    a=b;    b=tt;}void ins(int x,int&y,int l,int r,int v){    y=++xb;    t[y].s=t[x].s+1;    if(l!=r){        int m=(l+r)>>1;        if(v>m){            t[y].lc=t[x].lc;            ins(t[x].rc,t[y].rc,m+1,r,v);        }else{            t[y].rc=t[x].rc;            ins(t[x].lc,t[y].lc,l,m,v);        }    }}void dfs(int x,int fa){    dep[x]=1+dep[*ff[x]=fa];    int y;    for(y=1;y<=16;++y)ff[x][y]=ff[ff[x][y-1]][y-1];    ins(rt[fa],rt[x],1,l,b[x]);    for(ui i=0;i<g[x].size();++i){        y=g[x][i];        if(y!=fa)dfs(y,x);    }}int gfa(int x){    return f[x]==x?x:f[x]=gfa(f[x]);}int lca(int x,int y){      if (dep[x]<dep[y]) swap(x,y); int i,tmp=dep[x]-dep[y];      for (i=0; i<=16; i++) if (tmp&(1<<i)) x=ff[x][i];      for (i=16; i>=0; i--)          if (ff[x][i]!=ff[y][i]){ x=ff[x][i]; y=ff[y][i]; }      return (x==y)?x:ff[x][0];  }  inline void link(int x,int y){    int u=gfa(x),v=gfa(y);    if(s[u]<s[v]){        swap(u,v);        swap(x,y);    }    s[u]+=s[v];    f[v]=u;    g[x].push_back(y);    g[y].push_back(x);    dfs(y,x);}int query(int w,int x,int y,int z,int l,int r,int k){    int m,o;    while(l<r){        m=(l+r)>>1;        o=t[t[y].lc].s+t[t[z].lc].s-t[t[x].lc].s-t[t[w].lc].s;        if(k>o)w=t[w].rc,x=t[x].rc,y=t[y].rc,z=t[z].rc,l=m+1,k-=o;            else w=t[w].lc,x=t[x].lc,y=t[y].lc,z=t[z].lc,r=m;    }    return d[l];}int main(){    scanf("%d%d%d%d",&T,&n,&m,&T);    for(i=1;i<=n;++i)a[i]=make_pair(getint(),f[i]=i),s[i]=1;    sort(a+1,a+n+1);    for(i=1;i<=n;++i){        if(a[i].first>a[i-1].first)++l;        d[b[a[i].second]=l]=a[i].first;    }    for(i=1;i<=n;++i)ins(0,rt[i],1,l,b[i]);    for(i=1;i<=m;++i){        x=getint();        y=getint();        link(x,y);    }    while(T--){        c=getchar();        while(!isalpha(c))c=getchar();        if(c=='L'){            x=getint()^la;            y=getint()^la;            //printf("a%d %d\n",x,y);            link(x,y);        }else{            x=getint()^la;            y=getint()^la;            z=getint()^la;            //printf("b%d %d %d\n",x,y,z);            i=lca(x,y);            //printf("i%d\n",i);            putint(la=query(rt[*ff[i]],rt[i],rt[x],rt[y],1,l,z));            putchar('\n');        }        //puts("WTF");        //for(i=1;i<=n;++i)printf("%d\n",*ff[i]);    }    return 0;}

这份代码洛谷上是AC不掉的，会MLE

阅读全文

0 0