atcoder 062D

来源:互联网 发布:手机知乎怎么复制 编辑:程序博客网 时间:2024/06/05 15:13

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;}