hdu 4417 Super Mario

题意:

有一个长度为  () 的区间,有  次询问 () ,每次询问三个数字L,R,H,询问区间 L - R 内有多少个数字小于等于H。

离线主席树裸题,把离线查询第K大的板子改改就行。但是,这个题目有点意思。

1、询问的 L,R 下标从0开始。所以每次询问先把 L,R 加1。

2、大部分主席树的题目的序列都是的排列,所以不需要离散化,但这题比较妙。

3、首先一定要离散化,然后考虑离散化的数字,长度为  的区间加上  个询问,一共  个数字,然后离散化,这题就做完了。

4、交上去,MLE,嗯,算一算,空间复杂度  ,,结构体开小一点?WA了?

5、所以我们只离散化长度为  的区间,然后针对每一次询问的  ,先找出比  大的数的位置,然后  ,然后就避免了MLE。

6、AC。

Code

#include <bits/stdc++.h>
#define imid int mid=(left+right)/2;
using namespace std;
struct node
{
	int l;
	int r;
	int sum;
	node() { sum = 0; }
}tree[100005 * 20];
int n, m, cnt, a[100005], t[100005], root[100005];
void init()
{
	root[0] = 0;
	tree[0].l = tree[0].r = tree[0].sum = 0;
	cnt = 1;
}
void build(int num, int &rot, int left, int right)
{
	tree[cnt] = tree[rot];
	rot = cnt++;
	tree[rot].sum++;
	if (left == right)
		return;
	imid;
	if (num <= mid)
		build(num, tree[rot].l, left, mid);
	else
		build(num, tree[rot].r, mid + 1, right);
}
int query(int pre, int nex, int num, int left, int right)
{
	int s = tree[tree[nex].l].sum - tree[tree[pre].l].sum;
	if (left == right)
		return s;
	imid;
	if (num < mid)
		return query(tree[pre].l, tree[nex].l, num, left, mid);
	else if (num > mid)
		return s + query(tree[pre].r, tree[nex].r, num, mid + 1, right);
	else
		return s;
}
int main()
{
	int tt, cas = 1;
	scanf("%d", &tt);
	while (tt--)
	{
		printf("Case %d:\n", cas++);
		scanf("%d%d", &n, &m);
		for (int i = 1; i <= n; i++)
		{
			scanf("%d", &a[i]);
			t[i] = a[i];
		}
		int ql, qr, k, cnts = n;
		sort(t + 1, t + 1 + cnts);
		cnts = unique(t + 1, t + 1 + cnts) - t - 1;
		for (int i = 1; i <= n; i++)
			a[i] = lower_bound(t + 1, t + 1 + cnts, a[i]) - t;
		init();
		for (int i = 1; i <= n; i++)
		{
			root[i] = root[i - 1];
			build(a[i], root[i], 1, n);
		}
		for (int i = 0; i < m; i++)
		{
			scanf("%d%d%d", &ql, &qr, &k);
			ql++;
			qr++;
			k = upper_bound(t + 1, t + 1 + cnts, k) - t;
			k--;
			int ans = query(root[ql - 1], root[qr], k, 1, n);
			printf("%d\n", ans);
		}
	}
}

 

全部评论

相关推荐

把球:这个听过,你加了就会发现是字节的hr
点赞 评论 收藏
分享
10-09 00:50
已编辑
长江大学 算法工程师
不期而遇的夏天:1.同学你面试评价不错,概率很大,请耐心等待;2.你的排名比较靠前,不要担心,耐心等待;3.问题不大,正在审批,不要着急签其他公司,等等我们!4.预计9月中下旬,安心过节;5.下周会有结果,请耐心等待下;6.可能国庆节前后,一有结果我马上通知你;7.预计10月中旬,再坚持一下;8.正在走流程,就这两天了;9.同学,结果我也不知道,你如果查到了也告诉我一声;10.同学你出线不明朗,建议签其他公司保底!11.同学你找了哪些公司,我也在找工作。
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务