atcoder 062D - 3N Numbers

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/My_stage/article/details/77651924

http://abc062.contest.atcoder.jp/tasks/arc074_b

题意就是说 给你3n个数，之后删除n个数，使得前n个数 减去 后面 n个数之后的差值最大

思路： 用2个优先队列搞一下就行。一个维护最大值，一个维护最小值，并且我们知道最后这2n个数的分界点一定是在n~2n这N个数之间的。 那么我们可以对N个数求和，对后n个数求和，之后我们在n~2n中找到分界点即可。

#include <bits/stdc++.h>
#define maxs 320022
#define LL long long
#define mme(i,j) memset(i,j,sizeof(i))
using namespace std;
LL a[maxs],pre[maxs],suf[maxs];
priority_queue<LL, vector<LL>, greater<LL> >q;
priority_queue<LL> qq;

int main(){

    int n;
    while(~scanf("%d",&n))
    {
        LL sum1 =0,sum2=0;
        while(!q.empty())
            q.pop();
        while(!qq.empty())
            qq.pop();

        for(int i=1;i<=3*n;i++)
        {
            scanf("%d",&a[i]);
            if(i<=n) sum1+=a[i], q.push(a[i]);
            else if(i>2*n) sum2+=a[i], qq.push(a[i]);
        }
        pre[n]=sum1;
        suf[2*n+1]=sum2;

        for(int i=n+1;i<=2*n;i++){
            if(q.top()<a[i]){
                sum1+= a[i]-q.top();
                q.pop();
                q.push(a[i]);
            }
            pre[i]=sum1;
        }
        suf[2*n]=sum2;
        for(int i=2*n-1;i>=n;i--){
            if(qq.top()>a[i+1]){
                sum2=sum2 - qq.top()+a[i+1];
                qq.pop();
                qq.push(a[i+1]);
            }
            suf[i]=sum2;
        }
        LL ans = pre[n]-suf[n];
        for(int i=n+1;i<=2*n;i++){
            ans = max(ans,pre[i]-suf[i]);
        }

        printf("%lld\n",ans);
    }
    return 0;
}