hdu 4417(划分树)

将h值与划分树的中间值作比较,不小于中间值的话,答案加上查询区间进入左子树的个数并进入右子树,否则直接进入左子树。(注:查询区间可能为空)

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int MAXN = 100010;

int val[20][MAXN], toleft[20][MAXN];
int sorted[MAXN];
int ans;

void BuildT(int tl, int tr, int i) {
	if(tl == tr) {
		val[i][tl] = sorted[tl];
		return ;
	}
	int mid = (tl + tr) >> 1;
	int con = mid - tl + 1;
	int pl = tl;
	int pr = mid + 1;
	for(int j = tl; j <= tr; j++) if(val[i][j] < sorted[mid]) con--;
	for(int j = tl; j <= tr; j++) 
		if(val[i][j] < sorted[mid] || (val[i][j] == sorted[mid] && con)) {
			val[i + 1][pl++] = val[i][j];
			toleft[i][j] = toleft[i][j - 1] + 1;
			if(val[i][j] == sorted[mid]) con--;
		}
		else {
			val[i + 1][pr++] = val[i][j];
			toleft[i][j] = toleft[i][j - 1];
		}
	BuildT(tl, mid, i + 1);
	BuildT(mid + 1, tr, i + 1);
}

void QueryT(int tl, int tr, int l, int r, int h, int i) {
	if(l > r) return ;//  必须有  
	if(tl == tr) {
		if(val[i][tl] <= h) ans++;
		return ;	
	}	
	int mid = (tl + tr) >> 1;
	int ll = toleft[i][l - 1] - toleft[i][tl - 1];
	int mm = toleft[i][r] - toleft[i][l - 1];
	//printf("mid: %d h: %d mm: %d\n", sorted[mid], h, mm);
	if(h >= sorted[mid]) {
		ans += mm;
		QueryT(mid + 1, tr, mid + 1 + (l - tl - ll), mid + (l - tl - ll) + (r - l + 1 - mm), h, i + 1);
	}
	else QueryT(tl, mid, tl + ll, tl + ll + mm - 1, h, i + 1);
}

int main() {
	int t, tt = 1;
	scanf("%d", &t);
	while(t--) {
		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);
		BuildT(1, n, 0);
		printf("Case %d:\n", tt++);
		while(m--) {
			int l, r, h;
			scanf("%d%d%d", &l, &r, &h);
			ans = 0;
			QueryT(1, n, l + 1, r + 1, h, 0);
			printf("%d\n", ans);
		}
	}
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值