51nod 2206 低买高卖&codeforces867E Buy Low Sell High

题目

Solution

首先想到的是这类题可以用堆来做
假设当前最小值是$min$，遇到$x$，比$min$小，肯定不会选择卖，直接放堆里
如果比$min$大，默认卖掉+反悔操作
卖掉很简单，$ans+=x-min$，然后$pop$掉这个$min$
还可以不卖也不买，那么肯定留到后面一个$y$上面卖，到时候能赚$y-min$，但我们这波已经赚了$x-min$了，所以还要补上$y-x$，这与放进一个$x$元素等价
最后一种选择是买，那么之后肯定会有$y,z$，让$min,x$卖出，顺序无所谓，最终赚了$(y-x)+(z-x)+(x-min)$，$y-x$就是放进一个$x$，$z-x$也是放进一个$x$，所以在上一步基础上再放一个$x$即可

Code

#include<bits/stdc++.h>
using namespace std;
int i,ans,x,n;
priority_queue<int,vector<int>,greater<int> >q;
inline char gc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int rd(){
    int x=0,fl=1;char ch=gc();
    for (;ch<48||ch>57;ch=gc())if(ch=='-')fl=-1;
    for (;48<=ch&&ch<=57;ch=gc())x=(x<<3)+(x<<1)+(ch^48);
    return x*fl;
}
inline void wri(int a){if(a<0)a=-a,putchar('-');if(a>=10)wri(a/10);putchar(a%10|48);}
inline void wln(int a){wri(a);puts("");}
int main(){
	n=rd();
	for (i=0;i<n;i++){
		x=rd();
		if (q.empty() || x<=q.top()) q.push(x);
		else ans+=x-q.top(),q.pop(),q.push(x),q.push(x);
	}
	printf("%d",ans);
}