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行即命令,为题目描述所示。
1<=N<=30000,1<=M<=100000
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
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
Solution
这两天FFT做的我头疼……先做点水题换换脑子
我的Rotate函数又㕛叒叕打错T了两次真是喜闻乐见
Code
1 #include<iostream> 2 #include<cstring> 3 #include<cstdio> 4 #define N (30000+1000) 5 using namespace std; 6 7 int n,m,x,y,Sum[N],Son[N][2],Father[N],Rev[N],Val[N]; 8 char opt[30]; 9 10 void Update(int x){Sum[x]=Sum[Son[x][0]]+Sum[Son[x][1]]+Val[x];} 11 int Get(int x){return Son[Father[x]][1]==x;} 12 int Is_root(int x){return Son[Father[x]][0]!=x && Son[Father[x]][1]!=x;} 13 14 void Rotate(int x) 15 { 16 int wh=Get(x); 17 int fa=Father[x],fafa=Father[fa]; 18 if (!Is_root(fa)) Son[fafa][Son[fafa][1]==fa]=x; 19 Son[fa][wh]=Son[x][wh^1]; Father[fa]=x; 20 if (Son[fa][wh]) Father[Son[fa][wh]]=fa; 21 Son[x][wh^1]=fa; Father[x]=fafa; 22 Update(fa); Update(x); 23 } 24 25 void Pushdown(int x) 26 { 27 if (Rev[x] && x) 28 { 29 if (Son[x][0]) Rev[Son[x][0]]^=1; 30 if (Son[x][1]) Rev[Son[x][1]]^=1; 31 swap(Son[x][0],Son[x][1]); 32 Rev[x]=0; 33 } 34 } 35 36 void Push(int x){if (!Is_root(x)) Push(Father[x]); Pushdown(x);} 37 void Splay(int x) 38 { 39 Push(x); 40 for (int fa; !Is_root(x); Rotate(x)) 41 if (!Is_root(fa=Father[x])) 42 Rotate(Get(fa)==Get(x)?fa:x); 43 } 44 45 void Access(int x){for (int y=0; x; y=x,x=Father[x]) Splay(x),Son[x][1]=y,Update(x);} 46 void Make_root(int x){Access(x); Splay(x); Rev[x]^=1;} 47 int Find_root(int x){Access(x); Splay(x); while (Son[x][0]) x=Son[x][0]; return x;} 48 void Link(int x,int y){Make_root(x); Father[x]=y;} 49 50 int main() 51 { 52 scanf("%d",&n); 53 for (int i=1; i<=n; ++i) 54 scanf("%d",&Val[i]); 55 scanf("%d",&m); 56 for (int i=1; i<=m; ++i) 57 { 58 scanf("%s%d%d",opt,&x,&y); 59 switch (opt[0]) 60 { 61 case 'b': 62 { 63 if (Find_root(x)==Find_root(y)) 64 printf("no "); 65 else 66 printf("yes "),Link(x,y); 67 break; 68 } 69 case 'p': 70 { 71 Access(x); Splay(x); 72 Val[x]=y; Update(x); 73 break; 74 } 75 case 'e': 76 { 77 if (Find_root(x)!=Find_root(y)) 78 { 79 printf("impossible "); 80 break; 81 } 82 Make_root(x); 83 Access(y); Splay(y); 84 printf("%d ",Sum[y]); 85 break; 86 } 87 } 88 } 89 }