-
题目翻译:已知接下来N天的股票价格,每天你可以买进一股股票,卖出一股股票,或者什么也不做.N天之后你拥有的股票应为0,当然,希望这N天内能够赚足够多的钱。
-
一开始想用动态规划+单调队列优化,但是最坏情况下单调队列没有优化效果,时间复杂度还是n方的。
-
买入等于卖出,可以考虑把操作转换成每到一天,从前面找一个价格最小且没有被买入的股票买入,并且允许反悔(撤回某一天的卖出,在转换后的操作的前提下不需要撤回买入)。
-
用优先队列实现找前面价格最小的股票,假设当天股票价格为y,优先队列中最小股票价为x,若y>x,收入+(y-x),然后将两个y加入队列,其中一个表示反悔(不在y的时候卖了),另一个表示买入y;若y<=x,将y加入优先队列。
-
AC代码(队列元素取了负值)
#include<iostream>
#include<queue>
#include<cstdio>
using namespace std;
long long n,p,ans;
priority_queue<int>q;
int main()
{
scanf("%lld%lld",&n,&p);
q.push(-p);
while(--n)
{
scanf("%lld",&p);
long long x=q.top();
if(p>-x){
ans+=p+x;
q.pop();
q.push(-p);
q.push(-p);
}
else
q.push(-p);
}
cout<<ans;
}