Super Mario--线段树或树状数组+离散化+离线处理

最新推荐文章于 2019-08-07 21:29:25 发布

原创最新推荐文章于 2019-08-07 21:29:25 发布 · 213 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

算法专栏收录该内容

145 篇文章

订阅专栏

本文介绍了一种解决区间查询问题的方法，通过使用线段树和树状数组进行区间更新和查询，解决了马里奥跳跃问题中障碍物的计数问题。文章详细解释了线段树和树状数组的实现过程，并提供了完整的代码示例。

题目链接：https://cn.vjudge.net/problem/HDU-4417

题目大意

给定一个长度为n的序列，m个询问，每次询问时，马里奥都有一个跳跃的最大值，问在此区间内马里奥能越过的障碍有多少个（不考虑中间被阻挡）。

分析

其实就是求某个区间内，障碍高度小于等于h的有多少个。先把障碍的高度离散化，同时记录下标，再把区间存下，同时记录下标，将区间按h排序，然后遍历每个区间，将低于该区间h的数放置，最后求区间和。

线段树写法

#include<iostream>
#include<cstdio>
#include<string.h>
#include<algorithm>
using namespace std;
const int N = 100010;
int n,m,ans[N];
struct node{
	int l,r,sum;
}tr[N<<2];
struct Ques{
	int l,r,id,h;
}q[N];
struct A{
	int id,val;
}a[N];
bool cmp1(const Ques x,const Ques y)
{
	return x.h < y.h;
}
bool cmp2(const A x,const A y)
{
	return x.val < y.val;
}
void pushup(int m)
{
	tr[m].sum = tr[m<<1].sum + tr[m<<1|1].sum;
}
void build(int m,int l,int r)
{
	tr[m].l = l;
	tr[m].r = r;
	tr[m].sum = 0;
	if(l == r) return ;
	int mid = (l + r) >> 1;
	build(m<<1,l,mid);
	build(m<<1|1,mid+1,r);
}
void updata(int m,int id)
{
	if(tr[m].l == id && tr[m].r == id)
	{
		tr[m].sum = 1;
		return ;
	}
	int mid = (tr[m].l + tr[m].r) >> 1;
	if(id <= mid)
		updata(m<<1,id);
	else
		updata(m<<1|1,id);
	pushup(m);
}
int query(int m,int l,int r)
{
	if(tr[m].l == l && tr[m].r == r)
		return tr[m].sum;
	int mid = (tr[m].l + tr[m].r) >> 1;
	if(r <= mid)
		return query(m<<1,l,r);
	else if(l > mid)
		return query(m<<1|1,l,r);
	else
		return query(m<<1,l,mid) + query(m<<1|1,mid+1,r);
}
int main()
{
	int t,cas = 1;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&n,&m);
		for(int i=0;i<n;i++)
		{
			scanf("%d",&a[i].val);
			a[i].id = i;
		}
		sort(a,a+n,cmp2);
		build(1,0,n-1);
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d%d",&q[i].l,&q[i].r,&q[i].h);
			q[i].id = i;
		}
		int j = 0;
		sort(q+1,q+m+1,cmp1);
		for(int i=1;i<=m;i++)
		{
			while(a[j].val <= q[i].h && j < n)
			{
				updata(1,a[j].id);
				j++;
			}
			ans[q[i].id] = query(1,q[i].l,q[i].r);
		}
		printf("Case %d:\n",cas++);
		for(int i=1;i<=m;i++)
			printf("%d\n",ans[i]);
	}
	return 0;
}

树状数组写法：要将整个区间右移一个单位

#include <iostream>
#include <cstdio>
#include <string.h>
#include <algorithm>
using namespace std;
const int N = 1e5+5;
int n,m,c[2*N],ans[N];
struct A{
	int id,val;
}a[N];
struct Ques{
	int id,l,r,h;
}q[N];
bool cmp1(const Ques x,const Ques y)
{
	return x.h < y.h;
}
bool cmp2(const A x,const A y)
{
	return x.val < y.val;
}
int lowbit(int x)
{
	return x & (-x);
}
int getsum(int x)
{
	int ans = 0;
	while(x)
	{
		ans += c[x];
		x -= lowbit(x);
	}
	return ans;
}
void add(int x,int val)
{
	while(x < 2*N)
	{
		c[x] += val;
		x += lowbit(x);
	}
}
int main()
{
	int t,cas = 1;
	scanf("%d",&t);
	while(t--)
	{
		memset(c,0,sizeof c);
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i].val);
			a[i].id = i;
		}
		sort(a+1,a+n+1,cmp2);
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d%d",&q[i].l,&q[i].r,&q[i].h);
			q[i].id = i;
		}
		int j = 1;
		sort(q+1,q+m+1,cmp1);
		for(int i=1;i<=m;i++)
		{
			while(a[j].val <= q[i].h && j <= n)
			{
				add(a[j].id,1);
				j++;
			}
			ans[q[i].id] = getsum(q[i].r + 1) - getsum(q[i].l);
		}
		printf("Case %d:\n",cas++);
		for(int i=1;i<=m;i++)
			printf("%d\n",ans[i]);
	}
	return 0;
}