积木盛水问题

题目描述如下

解题思路：

1. 计算每一位置左边积木高度最高值，右边高度最高值。
2. 此位置可以装水的单位是min（左边最高，右边最高）-此位置的高度

举个例子

5号位置高度为0，左边最高是2，右边最高是3，所以这个5号位置可以蓄水为min（2,3）-0=2

代码如下

#include<iostream>
using namespace std;
int leftindex(int *a, int k,int n)
{
	int index_l = k;
	for (int i = 0; i < k; i++)
	{
		if (a[i] >= a[index_l])
		{
			index_l = i;
		}
	}
	if (index_l == k)
		return -1;
	else
		return index_l;
}
int rightindex(int *a, int k, int n)
{
	int index_r = a[k];
	for (int i = k; i < n; i++)
	{
		if (a[i] >= a[index_r])
		{
			index_r = i;
		}
	}
	if (index_r == k)
		return -1;
	else
		return index_r;
}
int min(int *a,int l,int r)
{
	int m = a[l] > a[r] ? a[r] : a[l];
	return m;
}
int main()
{
	int n;
	int cnt=0;
	cin >> n;
	int *high = new int[n];
	for (int i = 0; i < n; i++)
	{
		cin >> high[i];
	}
	for (int i = 1; i < n-1; i++)
	{
		int left = leftindex(high, i, 12);
		int right = rightindex(high, i, 12);
		if (left != -1 && right != -1)
		{
			cnt = cnt + min(high, left, right) - high[i];
		}
	}
	cout << cnt;
	delete[]high;
	return 0;
}

结果如下：

给定积木排列图：

很明显可以装6单位的水

这个解法 时间复杂度O(n^2)
空间复杂度O(1)