hdu 4417 Super Mario

来源：互联网发布：ubuntu下安装到U盘编辑：程序博客网时间：2024/05/04 22:14
划分树，在区间内二分查找
#include<stdio.h>#include<iostream>#include<string.h>#include<algorithm>using namespace std;// 划分树，主要解决区间内第几大的数// 本题用 大于中位数的和减去小于中位数的和const int MAXN = 100005;int node[20][MAXN];// 表示每层上每个位置的值int num[MAXN]; // 排序的数int frm_left[20][MAXN]; // 每层1-i有多少个数到左孩子void mBuild(int l, int r, int dep){if (l == r) {return;}int mid = (l+r)>>1;int same = mid-l+1; // 表示等于中间值且分到左孩子的数的个数int i;for ( i = l; i<= r; ++i){if (node[dep][i] < num[mid])--same;}int ll = l;int rr = mid+1;for (i = l; i<= r; ++i){if (node[dep][i] < num[mid]) // 分配给左孩子{node[dep+1][ll++] = node[dep][i];}else if (node[dep][i] == num[mid] && same > 0) // 去左边{node[dep+1][ll++] = node[dep][i];--same;}else// 右孩子node[dep+1][rr++] = node[dep][i];frm_left[dep][i] = frm_left[dep][l-1] + ll - l;}mBuild(l, mid, dep+1);mBuild(mid+1, r, dep+1);}int res;int mFind( int a, int b, int l, int r, int dep, int k){if (a == b) return node[dep][a];int mid = (l+r)>>1;int cnt = frm_left[dep][b] - frm_left[dep][a-1]; // 从a到b进入左孩子的数的个数if (cnt >= k) // 所求的数全在左孩子{// L+查询区间去左孩子的数的个数int newL =  l + frm_left[dep][a-1] - frm_left[dep][l-1];// 左端点+查询区间去左孩子的个数int newR = newL + cnt -1;return mFind(newL, newR, l, mid, dep+1, k);}else{// b + b到R中进入左孩子的个数int newR = b + frm_left[dep][r] - frm_left[dep][b];// 右端点 - 区间内进入右孩子的个数int newL = newR - (b-a-cnt);return mFind(newL, newR, mid+1, r, dep+1, k-cnt);}}int main()   {#ifndef ONLINE_JUDGE    freopen("in.txt", "r", stdin);#endifint cs= 0, t;int n, m, A, B, C;int i;scanf("%d", &t);while (t--){printf("Case %d:\n", ++cs);scanf("%d%d", &n, &m);for (i = 1; i<= n; i++){scanf("%d", &node[0][i]);num[i] = node[0][i];}sort(num+1, num+1+n);// 构建划分树mBuild(1,n,0);while (m--){scanf("%d%d%d", &A, &B, &C);A++; B++;int f = 0, t = B-A;int mid;while (f <= t){mid = (f+t)>>1;if (mFind(A,B,1,n,0,1+mid)<= C)f = mid+1;elset = mid-1;}printf("%d\n", f);}}return 0;}