P3224 [HNOI2012]永无乡（并查集 + splay）

题目链接

P3224 [HNOI2012]永无乡

思路

我们可以直接建n棵splay，维护每一个节点，然后询问就直接查询第k小get_k。
重点说一下如何建桥的事情。
主要在于只有建桥，没有断桥的事情，所以我们直接用并查集维护连通性，对于要建桥的时候就暴力合并，尽量把节点数少的合并到节点数多的树。

这样合并的时间复杂度是logn的。
证明复杂度的思路是：

考虑任意节点a进行k次合并，由于a所在树节点数是较小的那个棵，那么合并完成之后，a所在的树的节点数至少变为原来的2倍。那么最多也就合并logn次就变成一棵树了。

然后想到可以直接暴力合并两个splay就是一道比较水的题目了。
实际上跑的还挺快，不开O2 500ms左右。

AC代码

#include <bits/stdc++.h>
using namespace std;

#define ll long long
#define endl "\n"
const int max_n = 1e5 + 100;
const int inf = 0x3f3f3f3f;
const int mod = 1e9 + 7;

struct Node
{
   
    int s[2], p, v, size;
    void init(int _v, int _p)
    {
   
        v = _v, p = _p, size = 1;
        s[0] = s[1] = 0;
    }
};
int root[max_n], idx[max_n];
vector<vector<Node>> trs;

int pa[max_n];
void init(int n)
{
   
    for (int i = 0; i <= n; i++)
        pa[i] = i;
}

int find(int x)
{
   
    if (x != pa[x])
        pa[x] = find(pa[x]);
    return pa[x];
}

void pu(vector<Node> &tr, int u)
{
   
    int l = tr[u].s[0], r = tr[u].s[1];
    tr[u].size = tr[l].size + tr[r].size + 1;
}

void rotate(vector<Node> &tr, int x)
{
   
    int y = tr[<