COCI CONTEST #3 2014 STOGOVI LCA模拟栈堆情况

题目链接:http://acm.zjnu.edu.cn/CLanguage/contests/1154/problems/1003.html

 

题意:

       开始时给你一个空栈编号为0,接下来有n(n<=300000)个操作,第i次操作有三个类型,a表示第一种,输入v表示把之前生成的栈v取过来在栈顶加入i,b是第二种,输入v表示生成新的栈是在栈v弹出顶上的元素后的栈,保证一定存在栈顶,输出的是栈v弹出的元素,c是第三种,输入v w,表示取栈v为新栈,同时比较新的栈和栈w有多少个相同的元素,输出数量。

做法:

       看数据量就知道肯定不是模拟了..时间和容量都不够。

       这道题是用lca做的,反正我是想不到,配合并查集。如果是c操作,得到的新栈和原来的栈就是一个栈,那么用并查集把这u和v并在一起,表示这两个元素是同一个,a操作表示生成了一条新的边,i就是v的儿子,b表示是从v弹出元素之后,那么就是把i和v的父亲进行连接,注意这里所有的v都是在fin(v)之后的,因为要找的是第一次连接的那个点,然后不断更新top。下面代码里有些了注释,不懂可以理解理解,我是理解了很久。。。


#include
#define rep(i,a,b) for(int i=(int)a;i<=(int)b;i++)
using namespace std;
typedef long long ll;

const int maxn=300005;
int q,fa[maxn][20],ff[maxn],dep[maxn],top[maxn];
char op[5];
int fin(int x){
    return ff[x]==x?x:ff[x]=fin(ff[x]);
}
void un(int x,int y){
    int fx=fin(x),fy=fin(y);
    if(fx!=fy){
        ff[fx]=fy;
        fin(fx);
    }
}
void add(int u,int f){
    //printf("from %d to %d \n",u,f);
    dep[u]=dep[f]+1;
    fa[u][0]=f;
        for(int i=1;i=0;i--)
        if(dep[fa[x][i]]>=dep[y])
            x=fa[x][i];
    if(x==y) return x;
    for(int i=19;i>=0;i--)
        if(fa[x][i]!=fa[y][i])
            x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}

int main(){
    memset(fa,-1,sizeof(fa));
    scanf("%d",&q);
    rep(i,0,q) ff[i]=i;
    for(int i=1;i<=q;i++){
        int v,w;
        scanf("%s",op+1);
        if(op[1]=='a'){
            //将点i连接到a所在的集合上
            //表示这个点是在a的基础上多加了一个点
            //向下连接表示栈i是从u加了某一个值而来的
            scanf("%d",&v);
            int fv=fin(v);
            add(i,fv);
            top[i]=i;
        }
        else if(op[1]=='b'){
            //因为要消除v这次加入的点
            //所以要输出top[v]之后把它和v的父亲合并
            //表示是同一个结点 即使加边 但因为会un所以没有影响
            scanf("%d",&v);
            v=fin(v);
            int fv=fin(fa[v][0]);
            top[i]=top[fv];
            add(i,fv);
            un(i,fv);
            printf("%d\n",top[v]);
        }
        else {
            //将栈i顶赋值为v的顶
            //找到v的一个集合 将i和v合并 表示是同一类型
            //虽然v是从i来 但是还是v的父亲加了top[v]的点来的
            //所以还是要将点i连向v的父亲
            scanf("%d%d",&v,&w);
            v=fin(v),w=fin(w);
            top[i]=top[v];
            int fv=fin(fa[v][0]);
            un(i,v);
            add(i,fv);
            int nowfai=fin(i);
            //cout<<"lca = "<

 

你可能感兴趣的:(思维)