思路:
由题意得每个地点只能买一次或卖一次,所以不用加while循环,if_else判断即可,
用一个最小堆存未购入商品的价格,遍历输入的商品价格a[i]:
1.当最小堆为空,a[i]直接入堆。
2.当此时最小堆q.top()的价格小于a[i],就可以假设购买了q.top再将其以a[i]的价格卖出,此时get += a[i] - q.top(),因为这时的a[i]代替了q.top(),所以a[i]要入堆两次,其中一次是q.top()的中间品,另一个则是本身,中间品不计入购买数。
3.最后输出获得的总值,和购买数 * 2即可(交易次数 = 购买次数 + 卖出次数),最后最小堆剩余的商品价格就是没有购买的。
代码:
#include <bits/stdc++.h>
#define fastio ios::sync_with_stdio(false), cin.tie(NULL), cout.tie(NULL)
#define debug(a) cout << "debug : " << (#a) << " = " << a << endl
using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const int N = 1e5 + 10;
const int INF = 0x3f3f3f3f;
const double eps = 1e-6;
const int mod = 998244353;
int a[N];
int main()
{
int T;
cin >> T;
while (T--)
{
unordered_map<int, int> Map;
int n;
cin >> n;
for (int i = 1; i <= n; i++)
cin >> a[i];
ll get = 0, num = 0;
priority_queue<int, vector<int>, greater<int>> q; //记录未购买的商品值
for (int i = 1; i <= n; i++)
{
if (q.size() && a[i] > q.top())
{
int tp = q.top();
get += a[i] - tp;
q.pop();
q.push(a[i]); //加入两次是因为一个是未购买的商品a[i],另一个是代替此时出栈元素的q.top(),因为此时get已经获得差值,则q.top()可以用a[i]来代替。
q.push(a[i]);
num++;
if (Map[tp] != 0) //Map[tp]后面可能还会用到,因为可能有重复数字,所以不能直接Map[tp]--这样会使其值变为-1
{
num--;
Map[tp]--;
}
Map[a[i]]++;
}
else
q.push(a[i]);
}
cout << get << ' ' << num * 2 << endl;
}
return 0;
}