pta题目链接
邪恶学长
单纯的模板题目,求区间第k大
思路
离散化+去重
离散化的目的是防止数据过大,建树过程中造成内存过大。大致过程就是讲整组数据排序,然后按照顺序赋予一个序号,使得序号和数据对应,建树插入的就是对应序号。去重就是删除重复的部分采用一个函数unique将重复的部分放到最后,并返回重复部分的第一个地址,要求数据有序
for (int i = 1; i <= n; i++)
{
cin >> arr[i];
v.push_back(arr[i]);
}
sort(v.begin(), v.end());
v.erase(unique(v.begin(), v.end()), v.end());
然后这是一个查找数据对应序号的函数
inline int getid(int x)
{
return lower_bound(v.begin(), v.end(), x) - v.begin() + 1;
}
lower_bound返回大于等于val的第一个元素位置
插入建树
主席树是可持久化线段树的一种,用于查找历史某个状态,然后怎么快速查找某个历史版本,采用空间换时间的思路,第一反应就是每个版本都存一份出来,但是这样需求空间太大,怎么优化空间。
观察每次数据的更新或者修改,我们发现变化的只有这次修改的叶子结点一直向上到根结点这一条链,其他都和上一个版本一样,那么这些相同部分就是可以空间优化的部分,我们每次都更新一条新的链(连接有数据修改的部分),那些重复的部分我们则直接链接到这条新的链上,举个例子T[x]是一个结点,他的左边被我更新了,我递归修改T[x].l(线段树的修改套路),T[x].r=T[y].r,直接等于上一次这个位置结点的右边部分,假设y是上一同一个位置的结点的序号。
inline void update(int l, int r, int& x, int y, int pos)//y是同位置的上一个版本序号
{
T[++cnt] = T[y];//先复制一下上个版本的
x = cnt;//把复制的版本序号传递回去
T[x].sum++;
if (l == r)return;
int mid = (l + r) >> 1;
if (pos > mid)update(mid + 1, r, T[x].r, T[x].r, pos);//引用,直接就可以开始修改这个版本的左子树
else update(l, mid, T[x].l,T[x].l,pos);
}
区间查询
区间查询类似权值线段树的样式,本题求第k大,然后我是从小到大排序,就是求排完序后从左数第k个,第a个到第b个之间就是第a-1个版本和第b个版本之间的差距,先算两个版本右子树的sum差,为什么都是右边的:因为通过分治的思路,一直判断第k的在当前结点的左边还是右边,然后左边是大于本身的,知道的是第k大,就需要看大的部分,所以是r。如果差大于k就证明第k大包含在右边,查右边。但是在查左边的过程中,已经去掉了 差 个大的,传递的k变成k-sum
inline int query(int l, int r, int x, int y, int k)
{
if (l == r)return l;
int sum = T[T[y].r].sum - T[T[x].r].sum;
int mid = (l + r) >> 1;
if (sum >= k)query(mid + 1, r, T[x].r, T[y].r, k);
else query(l, mid, T[x].l, T[y].l, k - sum);
}
x在主函数里的输入是root[a-1],y是root[b],root是版本
代码
#include <bits/stdc++.h>
using namespace std;
const int maxn = 2e5 + 10;
struct qwq
{
int l, r, sum;
}T[maxn*40];
int root[maxn], arr[maxn], n, m, cnt = 0;
vector<int>v;
inline int getid(int x)
{
return lower_bound(v.begin(), v.end(), x) - v.begin() + 1;
}
inline void update(int l, int r, int& x, int y, int pos)
{
T[++cnt] = T[y];
x = cnt;
T[x].sum++;
if (l == r)return;
int mid = (l + r) >> 1;
if (pos > mid)update(mid + 1, r, T[x].r, T[x].r, pos);
else update(l, mid, T[x].l,T[x].l,pos);
}
inline int query(int l, int r, int x, int y, int k)
{
if (l == r)return l;
int sum = T[T[y].r].sum - T[T[x].r].sum;
int mid = (l + r) >> 1;
if (sum >= k)query(mid + 1, r, T[x].r, T[y].r, k);
else query(l, mid, T[x].l, T[y].l, k - sum);
}
int main()
{
ios::sync_with_stdio(false);
cin >> n >> m;
for (int i = 1; i <= n; i++)
{
cin >> arr[i];
v.push_back(arr[i]);
}
sort(v.begin(), v.end());
v.erase(unique(v.begin(), v.end()), v.end());
for (int i = 1; i <= n; i++)
{
update(1, n, root[i], root[i - 1], getid(arr[i]));
}
for (int i = 0; i < m; i++)
{
int a, b, k;
cin >> a >> b >> k;
cout << v[query(1, n, root[a - 1], root[b], k) - 1] << endl;
}
return 0;
}