XDU暑训2019 Day3 Buy Low Sell High

原创于 2019-07-03 16:52:32 发布 · 146 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#ACM

贪心同时被 2 个专栏收录

13 篇文章

订阅专栏

堆

4 篇文章

订阅专栏

本文探讨了一种在拥有无限资金情况下进行股票交易的策略，旨在通过维护最小堆并适时买卖股票来实现最大收益。策略中详细介绍了如何使用C++实现这一算法，包括堆的运用与收益计算。

题目：假设有无限金钱，给出第i天股票的金额，在第i天可以买入或者卖出股票，卖出时必须持有股票，求最大收益
思路：维护一个最小堆，若堆空或者堆顶元素小于当前放入元素，将元素放入堆中，否则表示可以卖出股票，将堆顶元素出队，收益加上当前元素减去堆顶元素，再将当前元素入堆两次，其中一次表示堆顶元素升值为当前元素
这题很难想得到，看看题解就好

#include<bits/stdc++.h>
using namespace std;
long long n, a[300005];
int main()
{
	while (cin >> n) {
		priority_queue<long long, vector<long long>, greater<long long> > q;
		long long ans = 0;
		for (int i = 0; i < n; i++) {
			cin >> a[i];
			if (q.size() == 0 || q.top() > a[i])
				q.push(a[i]);
			else {
				ans += a[i] - q.top(); q.pop();
				q.push(a[i]);
				q.push(a[i]);
			}
		}
		cout << ans << endl;
	}
	return 0;
}