首页 > 代码库 > bzoj2733: [HNOI2012]永无乡(splay)
bzoj2733: [HNOI2012]永无乡(splay)
2733: [HNOI2012]永无乡
Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 3778 Solved: 2020
Description
永无乡包含 n 座岛,编号从 1 到 n,每座岛都有自己的独一无二的重要度,按照重要度可 以将这 n 座岛排名,名次用 1 到 n 来表示。某些岛之间由巨大的桥连接,通过桥可以从一个岛 到达另一个岛。如果从岛 a 出发经过若干座(含 0 座)桥可以到达岛 b,则称岛 a 和岛 b 是连 通的。现在有两种操作:B x y 表示在岛 x 与岛 y 之间修建一座新桥。Q x k 表示询问当前与岛 x连通的所有岛中第 k 重要的是哪座岛,即所有与岛 x 连通的岛中重要度排名第 k 小的岛是哪 座,请你输出那个岛的编号。
Input
输入文件第一行是用空格隔开的两个正整数 n 和 m,分别 表示岛的个数以及一开始存在的桥数。接下来的一行是用空格隔开的 n 个数,依次描述从岛 1 到岛 n 的重要度排名。随后的 m 行每行是用空格隔开的两个正整数 ai 和 bi,表示一开始就存 在一座连接岛 ai 和岛 bi 的桥。后面剩下的部分描述操作,该部分的第一行是一个正整数 q, 表示一共有 q 个操作,接下来的 q 行依次描述每个操作,操作的格式如上所述,以大写字母 Q 或B 开始,后面跟两个不超过 n 的正整数,字母与数字以及两个数字之间用空格隔开。 对于 20%的数据 n≤1000,q≤1000
对于 100%的数据 n≤100000,m≤n,q≤300000
Output
对于每个 Q x k 操作都要依次输出一行,其中包含一个整数,表 示所询问岛屿的编号。如果该岛屿不存在,则输出-1。
Sample Input
4 3 2 5 1
1 2
7
Q 3 2
Q 2 1
B 2 3
B 1 5
Q 2 1
Q 2 4
Q 2 3
Sample Output
2
5
1
2
分析
可以用并查集维护连通性,这个没啥好说的。
然后 Splay 启发式合并,合并时,直接将小的暴力加入另一个。
这个调试了好久qwq。洛谷上还t了最后一个点
code
1 #include<cstdio> 2 #include<algorithm> 3 4 using namespace std; 5 6 const int MAXN = 100100; 7 int fa[MAXN],pa[MAXN],val[MAXN],ch[MAXN][2],siz[MAXN],q[MAXN]; 8 int n,m; 9 char opt[5]; 10 11 int read() 12 { 13 int x = 0, f = 1;char ch = getchar(); 14 while (ch<‘0‘||ch>‘9‘) {if(ch==‘-‘)f=-1; ch=getchar(); } 15 while (ch>=‘0‘&&ch<=‘9‘) {x=x*10+ch-‘0‘; ch=getchar(); } 16 return x*f; 17 } 18 void pushup(int x) 19 { 20 if (!x) return ; 21 siz[x] = siz[ch[x][0]]+siz[ch[x][1]]+1; 22 } 23 int son(int x) 24 { 25 return ch[fa[x]][1]==x;//不是等于1,qwq 26 } 27 void rotate(int x) 28 { 29 int y = fa[x],z = fa[y],b = son(x),c = son(y),a = ch[x][!b]; 30 if (z) ch[z][c] = x;fa[x] = z; 31 if (a) fa[a] = y;ch[y][b] = a; 32 ch[x][!b] = y;fa[y] = x; 33 pushup(y); 34 35 } 36 void splay(int x,int rt) 37 { 38 while (fa[x]!=rt) 39 { 40 int y = fa[x],z = fa[y]; 41 if (z==rt) rotate(x); 42 else 43 { 44 if (son(x)==son(y)) rotate(y),rotate(x); 45 else rotate(x), rotate(x); 46 } 47 } 48 pushup(x); 49 } 50 void insert(int &x,int pre,int id) 51 { 52 if (!x) 53 { 54 x = id;fa[x] = pre;siz[x] = 1; 55 splay(x,0); 56 return ; 57 } 58 if (val[id]<=val[x]) insert(ch[x][0],x,id); 59 else insert(ch[x][1],x,id); 60 pushup(x); 61 } 62 int getkth(int x,int k) 63 { 64 if (k<0 || k>siz[x]) return -1; 65 while (k<=siz[ch[x][0]] || k>siz[ch[x][0]]+1)//x是变的,不能直接l = ch[x][0],用l代替chp[x][0]使用 66 if (k<=siz[ch[x][0]]) x = ch[x][0]; 67 else k -= siz[ch[x][0]]+1, x = ch[x][1];; 68 splay(x,0); 69 return x; 70 } 71 void merge(int x,int y) 72 { 73 splay(x,0);splay(y,0); 74 if (siz[x]>siz[y]) swap(x,y);//启发式合并 75 int head = 0, tail = 1; 76 q[0] = y,q[1] = x; 77 while (head<tail) 78 { 79 int t = q[++head]; 80 if (ch[t][0]) q[++tail] = ch[t][0]; 81 if (ch[t][1]) q[++tail] = ch[t][1]; 82 ch[t][0] = ch[t][1] = 0; 83 insert(q[head-1],0,t); 84 } 85 } 86 int find(int x) 87 { 88 return pa[x]==x?x:pa[x]=find(pa[x]); 89 } 90 int main() 91 { 92 n = read(),m = read(); 93 for (int i=1; i<=n; ++i) 94 { 95 val[i] = read(); 96 pa[i] = i;siz[i] = 1; 97 } 98 for (int x,y,i=1; i<=m; ++i) 99 {100 x = read();y = read();101 if (find(x)!=find(y))102 {103 merge(x,y);104 pa[find(x)] = find(y);//设为find(y),不是y 105 }106 }107 int t = read();108 while (t--)109 {110 scanf("%s",opt);111 int x = read(), y = read();112 if (opt[0]==‘Q‘)113 {114 splay(x,0);115 printf("%d\n",getkth(x,y));116 }117 else 118 {119 if (find(x)!=find(y))120 {121 merge(x,y);122 pa[find(x)] = find(y);123 }124 } 125 }126 return 0;127 }
bzoj2733: [HNOI2012]永无乡(splay)