CF 867E. Buy Low Sell High(反悔贪心+堆)

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);
}

 

上一篇:121. Best Time to Buy and Sell Stock [Easy]


下一篇:Leetcode No.121 Best Time to Buy and Sell Stock(c++实现)