bzoj 2843 极地旅行社

2843: 极地旅行社

Time Limit: 10 Sec  Memory Limit: 256 MB
Submit: 293  Solved: 200
[ Submit][ Status][ Discuss]

Description

不久之前，Mirko建立了一个旅行社，名叫“极地之梦”。这家旅行社在北极附近购买了N座冰岛，并且提供观光服务。当地最受欢迎的当然是帝企鹅了，这些小家伙经常成群结队的游走在各个冰岛之间。

Mirko的旅行社遭受一次重大打击，以至于观光游轮已经不划算了。旅行社将在冰岛之间建造大桥，并用观光巴士来运载游客。Mirko希望开发一个电脑程序来管理这些大桥的建造过程，以免有不可预料的错误发生。

这些冰岛从1到N标号。一开始时这些岛屿没有大桥连接，并且所有岛上的帝企鹅数量都是知道的。每座岛上的企鹅数量虽然会有所改变，但是始终在[0, 1000]之间。

你的程序需要处理以下三种命令：

1."bridge A B"——在A与B之间建立一座大桥（A与B是不同的岛屿）。由于经费限制，这项命令被接受，当且仅当A与B不联通。若这项命令被接受，你的程序需要输出"yes"，之后会建造这座大桥。否则，你的程序需要输出"no"。

2."penguins A X"——根据可靠消息，岛屿A此时的帝企鹅数量变为X。这项命令只是用来提供信息的，你的程序不需要回应。

3."excursion A B"——一个旅行团希望从A出发到B。若A与B连通，你的程序需要输出这个旅行团一路上所能看到的帝企鹅数量（包括起点A与终点B），若不联通，你的程序需要输出"impossible"。

Input

第一行一个正整数N，表示冰岛的数量。

第二行N个范围[0, 1000]的整数，为每座岛屿初始的帝企鹅数量。

第三行一个正整数M，表示命令的数量。

接下来M行即命令，为题目描述所示。

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

HINT

1<=N<=30000

1<=M<=100000

Source

题解：lct 模板题，注意及时更新点值即可

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 30003
using namespace std;
int n,m;
int size[N],key[N],sum[N],ch[N][3],fa[N],top,st[N],rev[N];
int isroot(int x)
{
 return ch[fa[x]][0]!=x&&ch[fa[x]][1]!=x;
}
int get(int x)
{
  return ch[fa[x]][1]==x;
}
void pushdown(int x)
{
  if (!x) return;
  if (rev[x])
   {
    rev[ch[x][0]]^=1; rev[ch[x][1]]^=1; rev[x]=0;
    swap(ch[x][0],ch[x][1]);
   }
}
void update(int x)
{
 if (!x) return;
 size[x]=size[ch[x][0]]+size[ch[x][1]]+1;
 sum[x]=sum[ch[x][0]]+sum[ch[x][1]]+key[x];
}
void rotate(int x)
{
  int y=fa[x]; int z=fa[y]; int which=get(x);
  if (!isroot(y)) ch[z][ch[z][1]==y]=x;
  ch[y][which]=ch[x][which^1]; fa[ch[y][which]]=y;
  ch[x][which^1]=y; fa[y]=x; fa[x]=z;
  update(y); update(x);
}
void splay(int x)
{
  top=0; st[++top]=x;
  for (int i=x;!isroot(i);i=fa[i])
   st[++top]=fa[i];
  for (int i=top;i>=1;i--) pushdown(st[i]);
  while (!isroot(x))
   {
   	int y=fa[x];
   	if (!isroot(y))
   	 rotate(get(y)==get(x)?y:x);
   	rotate(x);
   }
}
void access(int x)
{
  int t=0;
  while(x)
  {
  	splay(x);
  	ch[x][1]=t; update(x);
  	t=x; x=fa[x];
  }
}
void rever(int x)
{
  access(x); splay(x); rev[x]^=1;
}
void link(int x,int y)
{
  rever(x); fa[x]=y; splay(x); update(x);
}
int find(int x)
{
  access(x); splay(x);
  while (ch[x][0]) x=ch[x][0];
  return x;
}
void change(int x,int y)
{
  key[x]=y; access(x); splay(x);
}
int main()
{
  freopen("a.in","r",stdin);
  scanf("%d",&n);
  for (int i=1;i<=n;i++) scanf("%d",&key[i]);
  scanf("%d",&m);
  for (int i=1;i<=m;i++)
  {
  	char s[20]; int x,y; scanf("%s%d%d",s,&x,&y);
  	if (s[0]=='e')
  	 {
  	  if (find(x)==find(y)) 
  	   {
  	   	 rever(x); access(y); splay(y); 
  	   	 printf("%d\n",sum[y]);
  	   }
  	  else
  	   printf("impossible\n");
  	 }
  	else
  	 if (s[0]=='b')
  	  {
  	   if (find(x)==find(y)) printf("no\n");
  	   else
  	   link(x,y),printf("yes\n");
      }
  	 else
  	  change(x,y);
  }
}