HDU 6438 Buy and Resell（贪心+优先队列+set）

Description

有$n$个城市，第$i$个城市商品价格为$a_i$，从$1$城市出发依次经过这$n$个城市到达$n$城市，在每个城市可以把手头商品出售也可以至多买一个商品，问最大收益以及在最大收益下的最少交易次数

Input

第一行一整数$T$表示用例组数，每组用例首先输入一整数$n$表示城市数量，之后输出$n$个整数$a_i$表示每个城市商品单价

$(1\le T\le 250,1\le n\le 10^5,\sum n\le 5\cdot 10^5)$

Output

输出最大收益以及在最大收益下的最少交易次数

Sample Input

3
4
1 2 10 9
5
9 5 9 10 5
2
2 1

Sample Output

16 4
5 2
0 0

Solution

贪心，用优先队列维护当前位置之前所有的商品价格，每次选取价格最低的买入，如果该价格的商品之前是卖出的状态，那么就合并这四次交易变成两次交易，需要用一个$mulset$维护卖出状态的商品，时间复杂度$O(nlogn)$

Code

#include<cstdio>
#include<algorithm>
#include<vector>
#include<queue>
#include<set>
using namespace std;
typedef long long ll;
#define maxn 100005
int T,n,a[maxn];
priority_queue<int,vector<int>,greater<int> >que;
multiset<int>s;
multiset<int>::iterator it;
int main()
{
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		for(int i=1;i<=n;i++)scanf("%d",&a[i]);
		s.clear();
		while(!que.empty())que.pop();
		ll ans=0;
		int num=0;
		for(int i=1;i<=n;i++)
		{
			if(!que.empty()&&que.top()<a[i])
			{
				int x=que.top();
				que.pop();
				it=s.find(x);
				if(it==s.end())num+=2;
				else que.push(x),s.erase(it);
				ans+=a[i]-x;
				s.insert(a[i]);
			}
			que.push(a[i]);
		}
		printf("%lld %d\n",ans,num);
	}
	return 0;
}