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
Sample Output
4 impossible yes 6 yes yes 15 yes 15 16
Solution
这两天FFT做的我头疼……先做点水题换换脑子
我的Rotate函数又㕛叒叕打错T了两次真是喜闻乐见
Code
1 #include2 #include 3 #include 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\n");65 else66 printf("yes\n"),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\n");80 break;81 }82 Make_root(x);83 Access(y); Splay(y);84 printf("%d\n",Sum[y]);85 break;86 }87 }88 }89 }