ICPC2018 青岛站 有关题目题解

热身赛D题

题意：

第一行给n个数，第二行给m个数，m<=n，从n个数中取m个数和第二行的m个数匹配，匹配花费abs(ai-bj)，求最小匹配花费。

思路：

建立三个堆，A堆存放第一行的数，B堆存放第二行中未被匹配过的数，C堆存放第二行中被匹配过的数。将n+m个数进行混合排序，从前向后扫描数组。

遇到第一行的数：

1.若B堆非空，则与B堆中花费最小的进行匹配

2.若B堆空的，则看C堆中是否存在匹配使得答案更优，若有取出匹配优化答案，同时将结果取反以及当前节点权值取反放入堆A中，表示若有其它第二行数想匹配这个A所需要花费的代价，类似费用流的反悔操作

3.其它情况，则将该数取反放入堆A

遇到第二行的数：

1.若A堆非空，则匹配A堆中情况最优的，同时将结果取反以及当前节点权值取反放入堆C中，表示若有其它第一行数想匹配这个B所需要花费的代价，类似费用流的反悔操作

2.若A堆空的，则将该数取反放入堆B

代码：

#include<bits/stdc++.h>
using namespace  std;
typedef long long LL;
const int maxn = 200010;
struct node
{
    int val,type;
    friend bool operator<( const node&a , const node&b )
    {
        return a.val<b.val;
    }
}a[maxn];

int main()
{
    int T;scanf( "%d" , &T );
    for( int cas=1 ; cas<=T ; cas++ )
    {
        int n,m; scanf( "%d%d" , &n , &m );
        for( int i=1 ; i<=n ; i++ )
            scanf( "%d" , &a[i+0].val ),a[i+0].type = 0;
        for( int i=1 ; i<=m ; i++ )
            scanf( "%d" , &a[i+n].val ),a[i+n].type = 1;
        sort( a+1 , a+n+m+1 );
        priority_queue< LL,vector<LL>,greater<LL> >A,B,C;
        LL ans = 0;
        for( int i=1 ; i<=n+m ; i++ )
        {
            LL x = a[i].val;
            if( a[i].type==0 )
            {
                if ( !B.empty() )
                {
                    LL t = B.top();
                    B.pop();
                    ans += x+t;
                }
                else if ( !C.empty() )
                {
                    LL t = C.top();
                    if ( x+t<0 )
                    {
                        C.pop();
                        ans += x+t;
                        A.push(-x-t-x);
                    }
                    else A.push(-x);
                }
                else A.push(-x);
            }
            else
            {
                if( !A.empty() )
                {
                    LL t = A.top();
                    A.pop();
                    ans += x+t;
                    C.push(-x-t-x);
                }
                else B.push(-x);
            }
        }
        printf( "%lld\n" , ans );
    }
    ret