下雨积水问题

计算特定数组中元素调整的和

int Adjust(stack<int> & s, int const cur, vector<int> const & L)
{
	int sum = 0;
	while (!s.empty() && L[cur] > L[s.top()])
	{
		int top = L[s.top()];
		s.pop();
		if (!s.empty())
		{
			sum += (min(L[s.top()], L[cur]) - top)*(cur - s.top() - 1);
		}
	}
	s.push(cur);
	return sum;
}
int Sum(vector<int> const& L)
{
	int sum = 0;
	stack<int> s;
	int const size = L.size();
	for (int i = 0; i < size - 1; ++i)
	{
		if (L[i] < L[i + 1] && !s.empty())
		{
			sum += (i - s.top()) * (min(L[s.top()], L[i + 1]) - L[i]);
		}
		else if (L[i] >L[i + 1])
		{
			sum += Adjust(s, i, L);
		}
	}
	sum += Adjust(s, size - 1, L);
	return sum;
}

int main(int argc, char* argv[])
{
	vector<vector<int>> testCase
	{
		{ 2, 5, 1, 2, 3, 4, 7, 7, 6 },
		{6,7,7,4,3,2,1,5,2},
		{1,7,4,1,2,4,2,7,1},
		{1,7,4,11,0,11,2,4,2,7,1},
		{ 11, 0, 7, 0, 5, 0, 3, 0, 1 },
		{5,0,2,4},
		{5, 0, 2, 5}
	};

	std::for_each(testCase.begin(), testCase.end(), [](vector<int> const &L){cout << Sum(L) << endl; });
	return 0;
}







评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值