<html>
题目的意思：给点区间[a, b]。查找第K大的数，和POJ2104题一样，仅仅是HDU上的时间限制5000MS。用我在POJ上的方法。过不了，会超时。
而这一题的代码，改一下main函数的输入，就能够直接AC了POJ上的2104.
这题，用分桶法，WR。纠结了一晚上，最后还是放弃了。实在不知道错在哪里。于是改用了划分树的方法，学习了划分树的建立和查找。
划分树：主要运用于求解序列中区间[a, b]上的第K大的数，也就是区间[a, b]从小到大排序。第K个。
基本的算法思路是：
1。建树：先排序好存放在一个数组中。然后找到中间那个数，在原来数组中将小于中间值的数放在左边。大于的数放在右边。依照这样建立一颗树。
记录各个点进入左子树的个数。在后面查找会用到。
如图：图是盗用的，没时间自己做。
2。查找：查找区间[a, b]上的第K个数。假设进入左孩子的数的个数大于等于K，进入左子树进行查找，假设小于，则进入右子树查找，查找第k - s大的数，s为区间[a,b]进入左子树的数的个数。
更新区间[a,b]。
代码能够当作模版来用。
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int M = 100010;
class node
{
public:
int num[M]; //该位置进入左孩子的个数
int val[M]; //当前位置的值
}t[30];
int sorted[M];
void build(int left, int right, int p)
{
if(left == right)
return;
int i, mid = (left + right) / 2;
int isame = mid - left + 1;
/*
isame用来推断是否存在与sorted[mid]同样的元素
開始如果有mid - left + 1个，去掉比sorted[mid]小的。剩下的就是插入左孩子的
*/
for(i = left; i <= right; i++)//推断
{
if(t[p].val[i] < sorted[mid])
isame--;
}
int l = left, r = mid + 1;
for(i = left; i <= right; i++)
{
if(i == left)
t[p].num[i] = 0;
else
t[p].num[i] = t[p].num[i - 1];
if(t[p].val[i] < sorted[mid]) //小于。进入左孩子
{
t[p].num[i]++;
t[p + 1].val[l++] = t[p].val[i];
}
else if(t[p].val[i] > sorted[mid]) //大于。进入右孩子
t[p + 1].val[r++] = t[p].val[i];
else //等于，推断是否存在相等。没有。进入右孩子。有。进入左孩子
{
if(isame)
{
isame--;
t[p].num[i]++;
t[p + 1].val[l++] = t[p].val[i];
}
else
{
t[p + 1].val[r++] = t[p].val[i];
}
}
}
build(left, mid, p + 1);
build(mid + 1, right, p + 1);
}
int query(int a, int b, int k, int p, int left, int right)
{
if(a == b)
return t[p].val[a];
int s, ss, mid = (left + right) / 2;
/* s为区间[a,b]进入左孩子的个数
ss为区间[left, a - 1]进入左孩子的个数，以下的也能够看得出来
*/
if(a == left)
{
s = t[p].num[b];
ss = 0;
}
else
{
s = t[p].num[b] - t[p].num[a - 1];
ss = t[p].num[a - 1];
}
if(s >= k)
{
a = left + ss; //
b = left + ss + s - 1;
return query(a, b, k, p + 1, left, mid);
}
else //递归右孩子。也就是找第 k-s大的数。由于前面有s个进入了左孩子
{
a = mid + 1 + a - left - ss;
b = mid + 1 + b - left - t[p].num[b];
return query(a, b, k - s, p + 1, mid + 1, right);
}
}
int main()
{
int T, n, m;
scanf("%d", &T);
while(T--)
{
scanf("%d%d", &n, &m);
for(int i = 1; i <= n; i++) //输入，
{
scanf("%d", &sorted[i]);
t[0].val[i] = sorted[i];
}
sort(sorted + 1, sorted + n + 1);
build(1, n, 0); //建树
int a, b, k;
for(int j = 0; j < m; j++) //查找
{
scanf("%d%d%d", &a, &b, &k);
printf("%d\n", query(a, b, k, 0, 1, n));
}
}
return 0;
}
版权声明：本文为博主原创文章，未经博主同意不得转载。
举报
- 本文已收录于下面专栏：
相关文章推荐
-
划分树基础 —— HDU 2665 Kth number
相应 HDU 题目 ：点击打开链接 Kth number Time Limit: 15000/5000 MS (Java/Others) Memory Limit: 32768/32...
- u013351484
- 2015-11-19 22:33
- 296
-
杭电ACM4515 小Q系列故事——世界上最遥远的距离
题目简单介绍： 世界上最遥远的距离 不是生与死 而是我就站在你面前 你却不知道我爱你 世界上最遥远的距离<
- 闫老三
- 2013-08-19 09:04
- 306
-
hdu 2665 Kth number 划分树
#include #include #include #define maxn 100007 using namespace std; int ord[maxn],onleft[18][maxn],v...
- firenet1
- 2014-08-09 08:14
- 489
-
The 36th ACM/ICPC Asia Regional Dalian Site —— Online Contest &&The kth great number
http://acm.hdu.edu.cn/showproblem.php?pid=4006 Proble- feiliboos
- 2011-09-22 11:38
- 211
-
The 36th ACM/ICPC Asia Regional Dalian Site —— Online Contest &&The kth great number
http://acm.hdu.edu.cn/showproblem.php?pid=4006 Proble- dogasshole
- 2011-09-22 11:38
- 254
收藏助手
不良信息举报
0条评论