若想要深刻学习反悔贪心,传送门。html
Description:c++
已知接下来 \(n\) 天的股票价格,天天能够买入当天的股票,卖出已有的股票,或者什么都不作,求 \(n\) 天以后最大的利润。学习
Method:spa
咱们能够快速想出一种贪心策略:买入价格最小的股票,在能够赚钱的当天卖出。设计
显然咱们能够发现,上面的贪心策略是错误的,由于咱们买入的股票能够等到能够赚最多的当天在卖出。code
咱们考虑设计一种反悔策略,使全部的贪心状况均可以获得全局最优解。(即设计反悔自动机的反悔策略)htm
定义 \(C_{buy}\) 为全局最优解中买入当天的价格, \(C_{sell}\) 为全局最优解中卖出当天的价格,则:
\[ C_{sell}-C_{buy}=\left(C_{sell}-C_i\right)+\left(C_i-C_{buy}\right) \]blog
\(C_i\) 为任意一天的股票价格。ip
即咱们买价格最小的股票去卖价格最大的股票,以期获得最大的利润。咱们先把当前的价格放入小根堆一次(此次是以上文的贪心策略贪心),判断当前的价格是否比堆顶大,如果比其大,咱们就将差值计入全局最优解,再将当前的价格放入小根堆(此次是反悔操做)。至关于咱们把当前的股票价格若不是最优解,就没有用,最后能够获得全局最优解。element
上面的等式即被称为反悔自动机的反悔策略,由于咱们并无反复更新全局最优解,而是经过差值消去中间项的方法快速获得的全局最优解。
(假如尚未理解这道题,能够看一看代码,有详细的注释)
Code:
#include<bits/stdc++.h> #define int long long using namespace std; inline void read(int &x) { int f=1;x=0;char s=getchar(); while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();} while(s>='0'&&s<='9'){x=x*10+s-'0';s=getchar();} x*=f; } priority_queue<int,vector<int>,greater<int> >qu;//开一个小根堆 int n; int ans=0;//全局最优解 signed main() { read(n); ans=0; for(int i=1,x;i<=n;i++) { read(x);//当前的股票价格 qu.push(x);//贪心策略:买价格最小的股票去买价格最大的股票 if(!qu.empty()&&qu.top()<x)//假如当前的股票价格不是最优解 { ans+=x-qu.top();//将差值计入全局最优解 qu.pop();//将已经统计的最小的股票价格丢出去 qu.push(x);//反悔策略:将当前的股票价格再放入堆中,即记录中间变量(等式中间无用的Ci) } } printf("%lld\n",ans);//输出全局最优解 return 0; }