C++解决盛最多水的容器问题

问题

给定 n 个非负整数 a1，a2，…，an，每个数代表坐标中的一个点 (i, ai) 。在坐标内画 n 条垂直线，垂直线 i 的两个端点分别为 (i, ai) 和 (i, 0)。找出其中的两条线，使得它们与 x 轴共同构成的容器可以容纳最多的水。

说明： 你不能倾斜容器，且 n 的值至少为 2。

图中垂直线代表输入数组 [1,8,6,2,5,4,8,3,7]。在此情况下，容器能够容纳水（表示为蓝色部分）的最大值为 49。

示例:

输入: [1,8,6,2,5,4,8,3,7]
输出: 49

解题思路

从两端向中间挪动，保存最大面积。

挪动策略是比较beg 和 end 这两个边，谁小，谁就应该往里面挪动。

#include<iostream>
#include<vector>
#include<sstream>
#include<algorithm>

using namespace std;

int maxArea(vector<int> &height) {
	int ret{ 0 };
	for (auto beg = height.begin(), end = prev(height.end()); beg < end; *beg < *end ? ++beg : --end)
		ret = max(ret, static_cast<int>(end - beg)*min(*beg, *end));
	return ret;
}

int main()
{
	string s;
	vector<int> height;
	getline(cin, s);
	istringstream is(s.substr(1, s.length() -1)); // 除去'[', ']'
	int inter;
	char ch;
	while (is >> inter)
	{
		height.push_back(inter);
		is >> ch;
	}
	//for (int i = 0; i < height.size(); i++)
	//	cout << height[i] << " ";
	//cout << endl;

	int max_Area = maxArea(height);
	cout << "The max Area is " << max_Area << endl;

	return 0;
}