HDU

来源:互联网 发布:单身贵族 知乎 编辑:程序博客网 时间:2024/06/03 18:06

题目:动态维护一组森林,要求支持一下操作:
link(x,y) : 如果x,y不在同一颗子树中,则通过在x,y之间连边的方式,连接这两颗子树
cut(x,y)  : 如果x,y在同一颗子树中,且x!=y,则将x视为这颗子树的根以后,切断y与其父亲结点的连接
ADD(x,y,w): 如果x,y在同一颗子树中,则将x,y之间路径上所有点的点权增加w
query(x,y): 如果x,y在同一颗子树中,返回x,y之间路径上点权的最大值

所有非法操作输出-1

思路:LCT

代码:

#pragma comment(linker, "/STACK:1024000000,1024000000")#include<iostream>#include<algorithm>#include<cstdio>#include<cmath>#include<cstring>#include<string>#include<vector>#include<map>#include<set>#include<queue>#include<stack>#include<list>#include<numeric>using namespace std;#define LL long long#define ULL unsigned long long#define INF 0x3f3f3f3f#define mm(a,b) memset(a,b,sizeof(a))#define PP puts("*********************");template<class T> T f_abs(T a){ return a > 0 ? a : -a; }template<class T> T gcd(T a, T b){ return b ? gcd(b, a%b) : a; }template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}// 0x3f3f3f3f3f3f3f3f// 0x3f3f3f3fconst int maxn=3e5+50;const int maxe=2*maxn;struct Edge{    int to,next;}edge[maxe];int head[maxn],tot;int ch[maxn][2],key[maxn],pre[maxn],rev[maxn];int add[maxn],Max[maxn];bool rt[maxn];//标记节点是不是splay的根int n;//n个节点,从1开始计数void init(){    tot=0;    for(int i=0;i<=n;i++){        head[i]=-1;        rt[i]=true;        ch[i][0]=ch[i][1]=0;        pre[i]=0;        rev[i]=0;        add[i]=0;    }    Max[0]=-2000000000;}void addedge(int u,int v){    edge[tot].to=v;    edge[tot].next=head[u];    head[u]=tot++;}void dfs(int u,int fa){    pre[u]=fa;    Max[u]=key[u];    for(int i=head[u];i!=-1;i=edge[i].next){        int v=edge[i].to;        if(v!=fa)            dfs(v,u);    }}void update_add(int x,int val){    if(!x) return;    key[x]+=val;    add[x]+=val;    Max[x]+=val;}void update_rev(int x){    if(!x) return;    swap(ch[x][0],ch[x][1]);    rev[x]^=1;}void push_down(int x){    if(add[x]){        update_add(ch[x][0],add[x]);        update_add(ch[x][1],add[x]);        add[x]=0;    }    if(rev[x]){        update_rev(ch[x][0]);        update_rev(ch[x][1]);        rev[x]=0;    }}void push_up(int x){    Max[x]=key[x];    Max[x]=max(Max[x],Max[ch[x][0]]);    Max[x]=max(Max[x],Max[ch[x][1]]);}void rotate(int x){    int y=pre[x],d=ch[y][1]==x;    ch[y][d]=ch[x][!d];    pre[ch[y][d]]=y;    pre[x]=pre[y];    pre[y]=x;    ch[x][!d]=y;    if(rt[y]) rt[y]=false,rt[x]=true;    else ch[pre[x]][ch[pre[x]][1]==y]=x;    push_up(y);}//P函数先将根结点到x的路径上所有的结点的标记逐级下放void P(int x){    if(!rt[x]) P(pre[x]);    push_down(x);}//将x旋转到根void splay(int x){    P(x);    while(!rt[x]){        int f=pre[x],ff=pre[f];        if(rt[f]) rotate(x);        else if((ch[ff][1]==f)==(ch[f][1]==x))            rotate(f),rotate(x);        else            rotate(x),rotate(x);    }    push_up(x);}//将x到根的路径变成首选边int Access(int x){    int y=0;    for(;x;x=pre[y=x]){        splay(x);        rt[ch[x][1]]=true;        rt[ch[x][1]=y]=false;        push_up(x);    }    return y;}//判断是否是同根(真实的树,非splay)bool judge(int u,int v){    while(pre[u]) u=pre[u];    while(pre[v]) v=pre[v];    return u==v;}//让x成为它所在树的根void make_root(int x){    Access(x);    splay(x);    update_rev(x);}//调用后u是原来u和v的lca,v和ch[u][1]分别存着lca的2个儿子//(原来u和v所在的2颗子树)void lca(int &u,int &v){    Access(v),v=0;    while(u){        splay(u);        if(!pre[u]) return;        rt[ch[u][1]]=true;        rt[ch[u][1]=v]=false;        push_up(u);        u=pre[v=u];    }}//如果u,v不在同一棵子树中,则连接这2棵子树void link(int u,int v){    if(judge(u,v)){        printf("-1\n");        return;    }    make_root(u);    pre[u]=v;}//使u成为u所在树的根,并且将v和它父亲的边断开void cut(int u,int v){    if(u==v||!judge(u,v)){        printf("-1\n");        return;    }    make_root(u);    Access(v);    splay(v);    pre[ch[v][0]]=0;    rt[ch[v][0]]=true;    ch[v][0]=0;    push_up(v);}//如果u,v在同一颗子树中,则将u,v之间路径上所有点的点权增加wvoid ADD(int u,int v,int w){    if(!judge(u,v)){        printf("-1\n");        return;    }    lca(u,v);    update_add(ch[u][1],w);    update_add(v,w);    key[u]+=w;    push_up(u);}//如果u,v在同一颗子树中,返回u,v之间路径上点权的最大值void query(int u,int v){    if(!judge(u,v)){        printf("-1\n");        return;    }    lca(u,v);    printf("%d\n",max(max(Max[v],Max[ch[u][1]]),key[u]));}int main(){    int Q,u,v,op,x,y,w;    while(~scanf("%d",&n)){        init();        for(int i=1;i<n;i++){           scanf("%d%d",&u,&v);           addedge(u,v);           addedge(v,u);        }        for(int i=1;i<=n;i++)            scanf("%d",&key[i]);        dfs(1,0);        scanf("%d",&Q);        while(Q--){            scanf("%d",&op);            if(op==1){                scanf("%d%d",&x,&y);                link(x,y);            }            else if(op==2){                scanf("%d%d",&x,&y);                cut(x,y);            }            else if(op==3){                scanf("%d%d%d",&w,&x,&y);                ADD(x,y,w);            }            else{                scanf("%d%d",&x,&y);                query(x,y);            }        }        printf("\n");    }    return 0;}




原创粉丝点击