原题链接
CGY 每到一座城市(编号为 ii)可以有以下三种操作:
1、花a_i元买入一块宝石
2、以a_i元卖出一块宝石
3、Do nothing
由于城市对宝石的管控十分严格,所以每次最多买入或卖出一块宝石。CGY 想知道自己去 M 国的路途中最多可以赚多少钱,当然,CGY 并不想花多余的力气与商人交易,所以交易的次数也要尽可能的少。你可以假设 CGY 一开始带着无限的金钱。
代码如下:
#include <bits/stdc++.h>
#define lowbit(x) x&-x
using namespace std;
/*
优先队列+贪心。假设三座城市的交易价格一次为A1,A2,A3,其中A3>A2>A1。
我们可以发现A3-A1=(A3-A2) + (A2-A1)。
也就是我们用A1元买入一块宝石,以A2元买出,赚了a元,
然后发现A3元大于A2元,可以再用A3元-A2元得到b元,其中A3-A1=a+b,
也就是说,A2元购入并不影响最终结果,没有中间商赚差价。
所以我们可以用一个优先队列来维护前面的交易价格的最小值,然后当前交易价格Ai大于队列最小值时,
购入,最小值出列,将Ai入队两次(因为Ai有可能是中间商),并标记Ai。如果队列最小值被标记过了,
则表明之前的Aj的交易为中间商,总次数-2即可。时间复杂度O(nlgn)。
本题目将每次的交易价格入队两次的思维比较巧妙。
*/
const int maxv=101000;
const int maxNum=0x7fffffff;
const double PI=3.1415927;
const double eps=1e-6;
const int mod=1e9+7;
typedef long long LL;
int a[maxv];
struct Node{
int id;
int value;
};
bool operator <(const Node &a,const Node &b){
if(a.value==b.value) return a.id>b.id;
else return a.value>b.value;
}
int main()
{
int t,n,val;
scanf("%d",&t);
while(t--){
scanf("%d",&n);
priority_queue<Node > q;
map<int,int> vis;
LL ans=0,cnt=0;
for(int i=1;i<=n;i++){
scanf("%d",&val);
if(q.size()&&val>q.top().value){
int top=q.top().value;q.pop();
ans+=val-top,cnt+=2,q.push(Node{i,val});
if(vis[top]) cnt-=2,vis[top]--;
vis[val]++;
}
q.push(Node{i,val});
}
printf("%lld %lld\n",ans,cnt);
}
return 0;
}
参考于链接