6 in 874 query out in 24622 in 12194 query
Case #1: 874 24622
| Time Limit: 20000MS | Memory Limit: 65536K | |
| Total Submissions: 41138 | Accepted: 13447 | |
| Case Time Limit: 2000MS | ||
Description
Input
Output
Sample Input
7 3 1 5 2 6 3 7 4 2 5 3 4 4 1 1 7 3
Sample Output
5 6 3
Hint
This problem has huge input,so please use c-style input(scanf,printf),or you may got time limit exceed.
这两题都 要求子区间的第k大数,如果用快排之后,再查,复杂度太大,不能过,就要用到划分树了,划分树,其实,就是线段树的
一个变种了,当然,这样做查询是lg(n)级别,建树是n * lg(n),再加上一次快排,也是n * lg(n),差不多就可以过了!
先来看看划分树,我们在线段树的基础上,如果每个结点都保存了,当前子段向左子树走的个数,如果,查询的k要小于,查询段向左走的个数,自然,我们向左子树就可以查到结果,如果,k>向左走的个数,当然要向右找k-midcount个数,区间的变换,我们可以
推一下向左走就是l + scount, l + ecount - 1,向右走就是mid + 1 + s - l - scount, mid + 1 + e - l - ecount,其中scount就是s之前向左走的个数, ecount,就是e之前向左走的个数。
上核心代码
#define MID(a,b) (((a)+(b))>>1)
#define N 100050
int num[20][N], val[20][N];
//第i层,向左包括自已,向左子树的个数,当前的值
int pri[N], sorted[N];
//原始和排序后的数列
char str[20];
void build(int l, int r, int layer){
if (l >= r){
return;
}
int mid = MID(l, r);
int ql = l, qr = mid + 1, leftCount = mid, eqCount = 0;
for (int i = l; i <= r; i++)
leftCount -= val[layer][i] < sorted[mid];
for (int i = l; i <= r; i++){
if (i == l){
num[layer][i] = 0;
}
else{
num[layer][i] = num[layer][i - 1];
}
if (val[layer][i] < sorted[mid]){
num[layer][i]++;
val[layer + 1][ql++] = val[layer][i];
}
else if (val[layer][i] > sorted[mid]){
val[layer + 1][qr++] = val[layer][i];
}
else {
if (eqCount < leftCount){
eqCount++;
num[layer][i]++;
val[layer + 1][ql++] = val[layer][i];
}
else {
val[layer + 1][qr++] = val[layer][i];
}
}
}
build(l, mid, layer + 1);
build(mid + 1, r, layer + 1);
}
//在layer层l到r间,找s-e之间的第k大数
int query(int l, int r, int layer, int s, int e, int k){
if (l >= r || s >= e){
return val[layer][s];
}
int scount, ecount, mid = MID(l, r);
if (s == l){
scount = 0, ecount = num[layer][e];
}
else {
scount = num[layer][s - 1], ecount = num[layer][e];
}
int midcount = ecount - scount;
if (k <= midcount){
return query(l, mid, layer + 1, l + scount, l + ecount - 1, k);
}
else {
return query(mid + 1, r, layer + 1, mid + 1 + s - l - scount, mid + 1 + e - l - ecount, k - midcount);
}
}
int main()
{
int tcase, tcasenum = 0;
int n, k, q, s, e, begin = 1, nn, qcount, m;
while (S2(n, m) != EOF)
{
FI(n){
S(val[0][i+1]);
}
FI(n){
sorted[i + 1] = val[0][i + 1];
}
sort(sorted + 1, sorted + n + 1);
build(1, n, 0);
FI(m){
S2(s, e); S(q);
Prn(query(1, n, 0, s,e,q));
}
}
return 0;
}
原文:http://blog.csdn.net/mengzhengnan/article/details/46398465