【HDU 4032】线段树单点更新+区间最值

1.题目链接。题目大意:在一段长为n的线段上,有m个操作。某人起始点在0,现在他想吃馅饼,m个操作种,第一种为在某个点长出来一个馅饼,第二个为这个人要开始吃馅饼。吃馅饼的时候,总是找距离的当前点最近的那个点去吃,如果发现有两个最近点,那么就按照上一次的方向去吃。问最后操作完,这个人走了多长的距离?

2.首先分析,我们需要维护的变量肯定是有curpos,当前点在哪。然后对于当前点,我们需要找到最近的点在哪,那么就是找一下左边有馅饼并且坐标最大的,右边有馅饼并且坐标最小的。OK,模型已经抽象出来了,其实就是维护当前点的两个最值。这个最值不是区间的最大最小值,而是距离这个点出现馅饼坐标的最大最小值,那么对于每一个吃馅饼的操作,我们只需要查询[0,curpos]的最大值,[pos,n]的最小值,看看哪个更近。如果相同,就按照之前的方向走。所以还有维护一个pre的信息,每次吃馅饼操作之后,我们需要维护四个信息:当前点在哪?当前方向在哪?已经走了多远?这个点的数量减一。线段树维护即可。

#include
using namespace std;
#pragma warning(disable:4996)
const int maxn = 100010;
const int INF = 1e9 + 10;
struct node
{
	int l, r;
	int t;
	int minn, maxx;
} seg[maxn * 3];

void build(int i, int l, int r)
{
	seg[i].l = l;
	seg[i].r = r;
	seg[i].t = 0;
	if (l == r)
	{
		seg[i].minn = INF;
		seg[i].maxx = -1;
		return;
	}
	int mid = (l + r) >> 1;
	build(i << 1, l, mid);
	build(i << 1 | 1, mid + 1, r);
	seg[i].maxx = max(seg[i << 1].maxx, seg[i << 1 | 1].maxx);
	seg[i].minn = min(seg[i << 1].minn, seg[i << 1 | 1].minn);
}

void add(int i, int t)
{
	if (seg[i].l == t && seg[i].r == t)
	{
		seg[i].maxx = seg[i].minn = t;
		seg[i].t++;
		return;
	}
	int mid = (seg[i].l + seg[i].r) >> 1;
	if (t <= mid)
		add(i << 1, t);
	else
		add(i << 1 | 1, t);
	seg[i].maxx = max(seg[i << 1].maxx, seg[i << 1 | 1].maxx);
	seg[i].minn = min(seg[i << 1].minn, seg[i << 1 | 1].minn);
}

void del(int i, int t)
{
	if (seg[i].l == t && seg[i].r == t)
	{
		seg[i].t--;
		if (seg[i].t == 0)
		{
			seg[i].minn = INF;
			seg[i].maxx = -1;
		}
		return;
	}
	int mid = (seg[i].l + seg[i].r) >> 1;
	if (t <= mid)
		del(i << 1, t);
	else
		del(i << 1 | 1, t);
	seg[i].maxx = max(seg[i << 1].maxx, seg[i << 1 | 1].maxx);
	seg[i].minn = min(seg[i << 1].minn, seg[i << 1 | 1].minn);
}
int queryMax(int i, int l, int r)
{
	if (seg[i].l == l && seg[i].r == r)
		return seg[i].maxx;
	int mid = (seg[i].l + seg[i].r) >> 1;
	if (r <= mid)
		return queryMax(i << 1, l, r);
	else if (l > mid)
		return queryMax(i << 1 | 1, l, r);
	else
		return max(queryMax(i << 1, l, mid), queryMax(i << 1 | 1, mid + 1, r));
}
int queryMin(int i, int l, int r)
{
	if (seg[i].l == l && seg[i].r == r)
		return seg[i].minn;
	int mid = (seg[i].l + seg[i].r) >> 1;
	if (r <= mid)
		return queryMin(i << 1, l, r);
	else if (l > mid)
		return queryMin(i << 1 | 1, l, r);
	else
		return min(queryMin(i << 1, l, mid), queryMin(i << 1 | 1, mid + 1, r));

}
int main()
{
	int T, Case = 0;
	scanf("%d", &T);
	while (T--)
	{
		int n, m;
		scanf("%d%d", &n, &m);
		build(1, 0, n);
		int pre = 1, curpos = 0, ans = 0;
		while (m--)
		{
			int a, b;
			scanf("%d", &a);
			if (a == 0)
			{
				scanf("%d", &b);
				add(1, b);
			}
			else
			{
				int t1 = queryMax(1, 0, curpos);
				int t2 = queryMin(1, curpos, n);
				if (t1 == -1 && t2 != INF)
				{
					ans += t2 - curpos;
					curpos = t2;
					del(1, t2);
					pre = 1;
				}
				else if (t1 != -1 && t2 == INF)
				{
					ans += curpos - t1;
					curpos = t1;
					del(1, t1);
					pre = -1;
				}
				else if (t1 != -1 && t2 != INF)
				{
					if (curpos - t1 > t2 - curpos)
					{
						ans += t2 - curpos;
						curpos = t2;
						del(1, t2);
						pre = 1;
					}
					else if (curpos - t1 < t2 - curpos)
					{
						ans += curpos - t1;
						curpos = t1;
						del(1, t1);
						pre = -1;
					}
					else
					{
						if (pre == 1)
						{
							ans += t2 - curpos;
							curpos = t2;
							del(1, t2);
							pre = 1;
						}
						else
						{
							ans += curpos - t1;
							curpos = t1;
							del(1, t1);
							pre = -1;
						}
					}
				}
			}
		}
		printf("Case %d: %d\n", ++Case, ans);
	}
	return 0;
}

 

你可能感兴趣的:(数据结构与算法基于c++实现)