POJ2796 区间最大参考值

给定一个数组，定义某个区间的参考值为：区间所有元素的和*区间最小元素。求该数组中的最大参考值以及对应的区间。

设某个区间所有元素的和为height，区间最小元素为width，

则对于单个元素的区间，height = width = 元素的值。

建立一个单调递增的栈。从第一个元素开始入栈，每个元素入栈之前必须先从栈顶开始删除大于或等于它的元素，把删除的所有元素的height累加到当前元素的height，然后把当前元素的值保存在width值中，这表示把当前元素前面比它大或相等的连续元素的值加起来，乘以它自己，也就是这段区间的参考值。每一次删除元素都需要计算一个参考值，取参考值的最大值就是答案了。不过题目还要求给出对应区间的起点和终点，因此在栈的操作过程中还得记录当前元素保存的区间的起点和大小，在更新参考值的过程中顺便更新区间的起点和终点就可以了。

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

const int N = 100005;

struct Elem
{
	long long height;
	long long width;
	int begin;
	int count;
};

Elem stack[N];
int top;

int main()
{
	int num, n;
	long long ans, tmp, tot;
	int ansbeg, ansend, count;
	scanf("%d", &n);
	top = 0;
	ans = 0;
	ansbeg = ansend = 1;
	for (int i = 0; i < n; ++i)
	{
		scanf("%d", &num);
		tmp = 0;
		count = 0;
		while (top > 0 && stack[top - 1].width >= num)
		{
			stack[top - 1].count += count;
			tot = (stack[top - 1].height + tmp) * stack[top - 1].width;
			if (tot > ans) 
			{
				ans = tot;
				ansbeg = stack[top - 1].begin;
				ansend = ansbeg + stack[top - 1].count - 1;
			}
			tmp += stack[top - 1].height;
			count = stack[top - 1].count;
			--top;
		}
		stack[top].height = num + tmp;
		stack[top].width = num;
		stack[top].begin = i + 1 - count;
		stack[top].count = 1 + count;
		++top;
	}
	tmp = 0;
	count = 0;
	while (top > 0)
	{
		stack[top - 1].count += count;
		tot = (stack[top - 1].height + tmp) * stack[top - 1].width;
		if (tot > ans) 
		{
			ans = tot;
			ansbeg = stack[top - 1].begin;
			ansend = ansbeg + stack[top - 1].count - 1;
		}
		tmp += stack[top - 1].height;
		count = stack[top - 1].count;
		--top;
	}
	printf("%lld\n%d %d\n", ans, ansbeg, ansend);
	return 0;
}