https://codeforces.com/problemset/problem/867/E
题意:
n天,每一天股票市场的股票价格已知
每天可以买入一股,或者卖出一股,或者什么也不做
初始金钱无限,求最大收益
维护一个小根堆
对于每一天,若前面没有价格更低的,今日价格加入堆。
若前面有价格更低的,即堆顶比今日价格低,就买入堆顶,今日卖出,累加收益。
然后今日价格加入堆2次,堆顶删除
为什么是2次?
每个数只有成为堆顶才有用
当它第一次做堆顶时,之前已经作为卖出价格了,现在又要作为堆顶买入,表示它是一个中间价。
那么既然它是中间价,他就还有一次作为买入的机会。
代码实现的时候可以不用管前面有没有价格更低的
直接累加与堆顶的差,插入堆2次
删除堆顶
因为如果之前没有比它价格更低的,累加差值的时候是0,删除堆顶删掉自己,相当于只插一次
#include<bits/stdc++.h> using namespace std; priority_queue<int,vector<int>,greater<int> >q; int main() { int n,x; long long ans=0; scanf("%d",&n); for(int i=1;i<=n;++i) { scanf("%d",&x); q.push(x); q.push(x); ans+=x-q.top(); q.pop(); } printf("%lld",ans); }