BZOJ2843极地旅行社&BZOJ1180[CROATIAN2009]OTOCI——LCT

题目描述

给出n个结点以及每个点初始时对应的权值wi。起始时点与点之间没有连边。有3类操作： 

1、bridge A B：询问结点A与结点B是否连通。

如果是则输出“no”。否则输出“yes”，并且在结点A和结点B之间连一条无向边。 

2、penguins A X：将结点A对应的权值wA修改为X。 

3、excursion A B：如果结点A和结点B不连通，则输出“impossible”。

否则输出结点A到结点B的路径上的点对应的权值的和。

给出q个操作，要求在线处理所有操作。

数据范围：1<=n<=30000, 1<=q<=300000, 0<=wi<=1000。

输入

第一行包含一个整数n(1<=n<=30000)，表示节点的数目。

第二行包含n个整数，第i个整数表示第i个节点初始时对应的权值。

第三行包含一个整数q(1<=n<=300000)，表示操作的数目。

以下q行，每行包含一个操作，操作的类别见题目描述。

任意时刻每个节点对应的权值都是1到1000的整数。

输出

输出所有bridge操作和excursion操作对应的输出，每个一行。

样例输入

5
4 2 4 5 6
10
excursion 1 1
excursion 1 2
bridge 1 2
excursion 1 2
bridge 3 4
bridge 3 5
excursion 4 5
bridge 1 3
excursion 2 4
excursion 2 5

样例输出

4
impossible
yes
6
yes
yes
15
yes
15
16

 

LCT模板题，splay维护子树和，注意操作1连通输出no，不连通输出yes。

#include<set>
#include<map>
#include<cmath>
#include<queue>
#include<stack>
#include<vector>
#include<cstdio>
#include<bitset>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int n,m;
int x,y;
char ch[10];
int s[30010][2];
int f[30010];
int r[30010];
int st[30010];
int v[30010];
int sum[30010];
int get(int rt)
{
    return rt==s[f[rt]][1];
}
void pushup(int rt)
{
    sum[rt]=sum[s[rt][0]]+sum[s[rt][1]]+v[rt];
}
void pushdown(int rt)
{
    if(r[rt])
    {
        r[s[rt][0]]^=1;
        r[s[rt][1]]^=1;
        r[rt]^=1;
        swap(s[rt][0],s[rt][1]);
    }
}
int is_root(int rt)
{
    return rt!=s[f[rt]][0]&&rt!=s[f[rt]][1];
}
void rotate(int rt)
{
    int fa=f[rt];
    int anc=f[fa];
    int k=get(rt);
    if(!is_root(fa))
    {
        s[anc][get(fa)]=rt;
    }
    s[fa][k]=s[rt][k^1];
    f[s[fa][k]]=fa;
    s[rt][k^1]=fa;
    f[fa]=rt;
    f[rt]=anc;
    pushup(fa);
    pushup(rt);
}
void splay(int rt)
{
    int top=0;
    st[++top]=rt;
    for(int i=rt;!is_root(i);i=f[i])
    {
        st[++top]=f[i];
    }
    for(int i=top;i>=1;i--)
    {
        pushdown(st[i]);
    }
    for(int fa;!is_root(rt);rotate(rt))
    {
        if(!is_root(fa=f[rt]))
        {
            rotate(get(fa)==get(rt)?fa:rt);
        }
    }
}
void access(int rt)
{
    for(int x=0;rt;x=rt,rt=f[rt])
    {
        splay(rt);
        s[rt][1]=x;
        pushup(rt);
    }
}
void reverse(int rt)
{
    access(rt);
    splay(rt);
    r[rt]^=1;
}
int find(int rt)
{
    access(rt);
    splay(rt);
    while(s[rt][0])
    {
        rt=s[rt][0];
    }
    return rt;
}
void link(int x,int y)
{
    reverse(x);
    f[x]=y;
}
void change(int rt,int x)
{
    v[rt]=x;
    access(rt);
    splay(rt);
}
int query(int x,int y)
{
    reverse(x);
    access(y);
    splay(y);
    return sum[y];
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&v[i]);
    }
    scanf("%d",&m);
    while(m--)
    {
        scanf("%s",ch);
        scanf("%d%d",&x,&y);
        if(ch[0]=='e')
        {
            if(find(x)!=find(y))
            {
                printf("impossible\n");
            }
            else
            {
                printf("%d\n",query(x,y));
            }
        }
        else if(ch[0]=='b')
        {
            if(find(x)!=find(y))
            {
                printf("yes\n");
                link(x,y);
            }
            else
            {
                printf("no\n");
            }
        }
        else
        {
            change(x,y);
        }
    }
}