您的位置:首页 > 其它

poj2104K-th Number【划分树入门题】

2016-01-28 15:49 274 查看
卡了半天攻下的知识点,没有想象中的难啊,某大牛不是说过“主席树是当时我弱不会划分树研究出来的替代品,被一小撮别有用心的人取了这么一个名字”(⊙﹏⊙)b 分明是主席树更难一点好么 ……

说说划分树是个什么玩意,归并排序知道吧,不知道就翻一下大话数据结构,两两排序然后合并,再合并大区间,再合并大大区间……划分树建树的思路和他的差不多,我记得归并排序最开始分几波是任意的,而且这个数字也影响效率,但是划分树拿来一个数组就分两组,找到中间值,小于中间值的放左边,大于的放右边,等于的看数量决定分到哪里,反正肯定是左子树分出去一半,右子树是另一半。刚刚说的这个过程是第二层树。原数组是第一层树。拿出第二层的左、右子树进行递归操作直到子树只有一个数。

建树的时候当然是要给赋值的,要不然建树干嘛。每一层树是一个结构体,结构体里面有三个数组,大小就都是原始数组的长度。三个数组意义分别是:val[i]当前下标的值

sum[i]  记录当前区间内比当前元素小的和(这个有没有看情况,要是只问了第k小的元素是什么,这项就不用要了)

num[i]此下标元素所在区间的当前位置之前进入左子树的个数(什么叫做当前区间?这个问题困扰过我,是指那些分成的当前层的分出的子树,而不是1-n)  

/***************
poj2104
2016.1.28
14768K 1219MS G++ 2059B
***************/
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int MAXN = 1000010;
int sorted[MAXN]; //对原集合中元素排序后的值
int val[30][MAXN]; //val记录第k层当前位置的值
int toleft[30][MAXN]; //记录元素所在区间当前位置前的元素进入到左子树的个数
int sum[30][MAXN]; //记录比当前元素小的元素的和
int n;

void build(int l, int r, int d) {
if (l == r) return ;
int mid = (l + r) >> 1;
int same = mid - l + 1;
for (int i=l; i<=r; i++)
if (val[d][i] < sorted[mid])
same--;
int lp = l, rp = mid+1;
for (int i=l; i<=r; i++) {
if (i == l) toleft[d][i] = 0;
else toleft[d][i] = toleft[d][i-1];

if (val[d][i] < sorted[mid]) {
toleft[d][i]++;
val[d+1][lp++] = val[d][i];
} else if (val[d][i] > sorted[mid])
val[d+1][rp++] = val[d][i];
else {
if (same) {
same--;
toleft[d][i]++;
val[d+1][lp++] = val[d][i];
} else val[d+1][rp++] = val[d][i];
}
}
build(l, mid, d+1);
build(mid+1, r, d+1);
}
int query(int a, int b, int k, int l, int r, int d) {
if (a == b) return val[d][a];

int mid = (l + r) >> 1;
int s, ss, sss;
if (a == l) {
s = toleft[d][b];
ss = 0;
} else {
s = toleft[d][b] - toleft[d][a-1];
ss = toleft[d][a-1];
}

if (s >= k) {
a = l + ss;
b = l + ss + s - 1;
return query(a, b, k, l, mid, d+1);
} else {
a = mid+1 + a - l - ss;
b = mid+1 + b - l - toleft[d][b];
return query(a, b, k-s, mid+1, r, d+1);
}
}

int main()
{
//freopen("cin.txt","r",stdin);
int n, m;
scanf("%d %d", &n, &m);
for (int i=1; i<=n; i++) {
scanf("%d", &sorted[i]);
val[0][i] = sorted[i];
}
sort(sorted+1, sorted+1+n);
build(1, n, 0);

int a, b, k;
while (m--) {
scanf("%d%d%d", &a, &b, &k);
printf("%d\n", query(a, b, k, 1, n, 0));
}
return 0;
return 0;
}
对面坐着三个搞数学建模的,实在好奇,刚刚手贱百度了一下,在数学建模吧里有人问ACM和它的取舍,下面的回复除了中立就是推荐前者……加油吧少女~~(捂脸,还有就是今天突发奇想翻了墙,第一次成功登陆CF,但是TC虽然登陆上了,没找到在哪里注册比赛。争取毕业前BC和CF双红吧,加油↖(^ω^)↗
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: