SPOJ QTREE LCT

单旋splay的LCT  刚好卡过。。 


#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <bitset>
#define MAXN 100000
using namespace std;
struct node;
node *pt;
struct node
{
    node *l,*r,*f;
    int val,size,mx;
    node() {}
    node(int _val) {l=r=f=NULL,val=_val,size=1,mx=_val;}
    void* operator new(size_t) {return pt++;}
}*tree[MAXN],pool[MAXN];
inline bool isroot(node *u) {return (!u->f || (u->f->l!=u&&u->f->r!=u));}
inline void update(node *u)
{
    u->size=1,u->mx=u->val;
    if (u->l) u->size+=u->l->size,u->mx=max(u->mx,u->l->mx);
    if (u->r) u->size+=u->r->size,u->mx=max(u->mx,u->r->mx);
}
void zig(node *u)
{
    node *x=u->f,*y=x->f;
    u->f=y,x->f=u;
    if (y)
    {
        if (y->l==x) y->l=u;
        else if (y->r==x) y->r=u;
    }
    x->l=u->r; if (x->l) x->l->f=x;
    u->r=x;
    update(x);
}
void zag(node *u)
{
    node *x=u->f,*y=x->f;
    u->f=y,x->f=u;
    if (y)
    {
        if (y->l==x) y->l=u;
        else if (y->r==x) y->r=u;
    }
    x->r=u->l; if (x->r) x->r->f=x;
    u->l=x;
    update(x);
}
void splay(node *u)
{
    while (!isroot(u))
        if (u->f->l==u) zig(u); else zag(u);
    update(u);
}
node* expose(node *p)
{
    node *q;
    for (q=NULL;p;p=p->f)
    {
        splay(p);
        p->r=q;
        update(q=p);
    }
    return q;
}
node* lca(node *u,node *v)
{
    expose(u);
    return expose(v);
}
struct edge_node
{
    int next,p,w,num;
}edge[MAXN];
int T,n,head[MAXN],cnt,point[MAXN];
bitset<MAXN> vis;
void insert(int a,int b,int c,int z)
{
    edge[++cnt].p=b;
    edge[cnt].w=c;
    edge[cnt].num=z;
    edge[cnt].next=head[a];
    head[a]=cnt;
}
void dfs(int u,node *fa,int eg)
{
    vis[u]=true;
    point[edge[eg].num]=u;
    tree[u]=new node(edge[eg].w); tree[u]->f=fa;
    for (int k=head[u];k;k=edge[k].next)
    {
        int v=edge[k].p;
        if (!vis[v])
            dfs(v,tree[u],k);
    }
}
char com[10];
int main()
{
    cin>>T;
    while (T--)
    {
        pt=pool;
        cnt=0;
        vis.reset();
        memset(head,0,sizeof head);
        scanf("%d",&n);
        for (int i=1;i<n;++i) {int x,y,z;scanf("%d%d%d",&x,&y,&z); insert(x,y,z,i);insert(y,x,z,i);}
        dfs(1,NULL,0);
        while (1)
        {
            int x,y;
            scanf("%s",com);
            if (com[0]=='D') break;
            if (com[0]=='Q') {
                scanf("%d%d",&x,&y);
                int ans=0;
                node *tmp=lca(tree[x],tree[y]);
                if (tmp!=tree[x]) {expose(tree[x]); expose(tmp); splay(tree[x]); ans=max(ans,tree[x]->mx);}
                if (tmp!=tree[y]) {expose(tree[y]); expose(tmp); splay(tree[y]); ans=max(ans,tree[y]->mx);}
                printf("%d\n",ans);
            }
            if (com[0]=='C') {
                scanf("%d%d",&x,&y);
                x=point[x];
                splay(tree[x]); tree[x]->val=y;
                update(tree[x]);
            }
        }
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值