题面
小L厌倦了算法竞赛,希望到股市里一展身手。他凭借自己还行的计算机功底和可以的智商,成功建立一个模型预测了一支股票接下来n天的价格。我们把这支股票第i天的价格称为a_i。在接下来n天里,每一天小L可以选择花费a_i买入一股或者卖出一股从而获得a_i元收入。
当然小L卖出股票的时候,自己的账户上必须要有至少一股的剩余。现在小L希望知道,在n天过去之后,采取最优策略的情况下自己最多赚到多少钱。注意小L家产万贯,可以视为初始本金为无穷,另外每一天小L能且只能进行一次买或卖的操作,不能什么都不做。
输入格式
一行一个数n,表示接下来n天
接下来了n行,每行一个正整数,表示股票在这一天的价格。
输出格式
一行一个整数,表示小L的最大获利,如果小L赔钱,则输出相应的负数
样例输入1
4
1
3
2
4
样例输出1
4
样例输入2
4
1
2
3
4
样例输出2
4
数据范围
对于10%的数据, 1 <= n <= 200
对于30%的数据, 1 <= n <= 3000
对于100%的数据, 1 <= n, a_i <= 200000
评测环境:Win7+Lemon
做法
数据范围很大,dp根本无法进行,因此算法应该是贪心,问题就在于怎么贪
此题的难点就在于手中的股票数量必须始终非负,这导致了许多方法无法实现
但我们可以利用这个条件,发现前i天必须至少有(i/2)(向上取整)天卖出股票,否则将入不敷出.
因而可以先假设将股票全部卖光,记录一个小根堆,每过两天就pop此时的最小值,表示在这一天卖掉股票,这样就能得出答案了.
代码
#include<iostream>
#include<cstdio>
#include<queue>
#define ll long long
#define N 200100
using namespace std;
ll n,num[N],ans;
priority_queue<ll,vector<ll>,greater<ll> >pq;
int main()
{
ll i,j,p;
cin>>n;
for(i=1;i<=n;i++)
{
scanf("%lld",&num[i]);
ans+=num[i];
}
for(i=1;i<=n;i++)
{
pq.push(num[i]);
if((i+1)/2>i-pq.size())
{
ans-=2*pq.top();
pq.pop();
}
}
cout<<ans;
}