bzoj 1180: [CROATIAN2009]OTOCI

1180: [CROATIAN2009]OTOCI

Time Limit: 50 Sec  Memory Limit: 162 MB
http://www.lydsy.com/JudgeOnline/problem.php?id=1180

Description

给出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。

Input

第一行包含一个整数n(1<=n<=30000)，表示节点的数目。第二行包含n个整数，第i个整数表示第i个节点初始时对应的权值。第三行包含一个整数q(1<=n<=300000)，表示操作的数目。以下q行，每行包含一个操作，操作的类别见题目描述。任意时刻每个节点对应的权值都是1到1000的整数。

Output

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

Sample Input

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

Sample Output

4
impossible
yes
6
yes
yes
15
yes
15
16

#include<cstdio>
#include<algorithm>
#define N 30001
using namespace std;
int n,key[N],m,sum[N],fa[N],tag[N],st[N],ch[N][2];
inline void up(int x)
{
    sum[x]=key[x]+sum[ch[x][0]]+sum[ch[x][1]];
}
inline void down(int x)
{
    if(tag[x])
    {
        tag[x]^=1;tag[ch[x][0]]^=1;tag[ch[x][1]]^=1;
        swap(ch[x][0],ch[x][1]);
    } 
}
inline bool isroot(int x)
{
    return ch[fa[x]][0]!=x&&ch[fa[x]][1]!=x;
}
inline void rotate(int x)
{
    int y=fa[x],z=fa[y],l,r;
    if(ch[y][0]==x) l=0;else l=1;
    r=l^1;
    if(!isroot(y))
    {
        if(ch[z][0]==y) ch[z][0]=x;
        else ch[z][1]=x;
    }
    fa[x]=z;fa[y]=x;fa[ch[x][r]]=y;
    ch[y][l]=ch[x][r];ch[x][r]=y;
    up(y);
}
inline void splay(int x)
{
    int top=0;st[++top]=x;
    for(int i=x;!isroot(i);i=fa[i]) st[++top]=fa[i];
    for(int i=top;i;i--) down(st[i]);
    while(!isroot(x))
    {
        int y=fa[x],z=fa[y];
        if(!isroot(y))
        {
            if(ch[y][0]==x^ch[z][0]==y) rotate(x);
            else rotate(y);
        }
        rotate(x);
        up(x);
    }
}
inline void access(int x)
{
    int t=0;
    while(x)
    {
        splay(x);
        ch[x][1]=t;
        up(x);
        t=x;x=fa[x];
    }
}
inline void make_root(int x)
{
    access(x);
    splay(x);
    tag[x]^=1;
}
inline int findroot(int x)
{
    access(x);
    splay(x);
    while(ch[x][0]) x=ch[x][0];
    return x;
}
inline void link(int x,int y)
{
    make_root(x);
    fa[x]=y;
    splay(x);
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) 
    {
        scanf("%d",&key[i]);
        up(i);
    }
    scanf("%d",&m);
    char c[10];int x,y;
    for(int i=1;i<=m;i++)
    {
        scanf("%s%d%d",c,&x,&y);
        if(c[0]=='b')
        {
            int a=findroot(x);
            int b=findroot(y);
            if(a==b) printf("no
");
            else {printf("yes
");link(x,y);}
        }
        else if(c[0]=='p')
        {
            make_root(x);
            key[x]=y;
            up(x);
        }
        else
        {
            int a=findroot(x);
            int b=findroot(y);
            if(a!=b) printf("impossible
");
            else 
            {
                make_root(x);
                access(y);
                splay(y);
                printf("%d
",sum[y]);
            }
        }
    }
}

错误：

1、access时没有维护sum

2、输入时没有给sum赋值