题意
有n个城市,你会从前往后依次到达每个城市,你每次可以用a[i]元购买或者卖掉一个物品,你可以同时保存无限个个物品。最开始你身上没有物品,但是有无限的金钱,现在问你最大的收益是多少。
思路
先考虑这样一个问题:如果有a[i]<a[i+1]<a[i+2],我们在a[i]点买入一个,a[i+2]点卖出一个 等价于 我们在a[i]点买入一个,在a[i+1]点卖出一个并且以a[i+1]的价格再买入一个,a[i+2]点卖出一个。我们可以发现:没有中间商赚差价,所以我们在某个位置卖出的时候要把该元素记录下来,留着作为后面更高价的中间商。
所以用优先队列维护到当前位置的最优值即可。
如果当前值 x < 之前最小值,那么就把x多放进去一次,意为作为后面的中转点,交易次数的维护用一个map,如果价格为x的位置有多个,肯定优先用前面的。
每个元素都要入队,意为再此点买入一个物品。
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1e5 + 5;
typedef long long LL;
int main()
{
int T; scanf("%d", &T);
while (T--)
{
int n; scanf("%d", &n);
priority_queue<int, vector<int>, greater<int> > pq;
map<int, int> mp;
LL res = 0, cnt = 0;
for (int i = 0; i < n; i++)
{
int x; scanf("%d", &x);
if (!pq.empty() && pq.top() < x)
{
int u = pq.top(); pq.pop();
res += x-u;
pq.push(x);
cnt++;
if (mp[u]) mp[u]--, cnt--;
mp[x]++;
}
pq.push(x);
}
printf("%lld %lld\n", res, cnt*2);
}
return 0;
}
/*
3
4
1 2 10 9
5
9 5 9 10 5
2
2 1
*/