CodeForces 840 A.Leha and Function（水~）

最新推荐文章于 2020-08-18 07:41:32 发布

原创最新推荐文章于 2020-08-18 07:41:32 发布 · 304 阅读

0 ·

CC 4.0 BY-SA版权

Code Forces 同时被 2 个专栏收录

606 篇文章

订阅专栏

水题

562 篇文章

订阅专栏

本文介绍了一种通过排序实现的最大期望值匹配算法，用于解决给定两个序列如何重排一个序列使得特定函数的总和最大。该算法适用于竞赛编程及算法设计领域。

Description

定义 $F(n,k)$ 为从 $\{1,2,...,n\}$ 中选出一个 $k$ 子集，其最小值的期望，给出两个长度为 $m$ 的序列 $A,B$ ，给 $A$ 重排后得到序列 $A'$ ，问使得 $\sum\limits_{i=1}^mF(A'_i,B_i)$ 最大的 $A'$

Input

第一行一整数 $m$ 表示序列长度，之后输入 $m$ 个整数 $A_i$ ，最后输入 $m$ 个整数 $B_i(1\le m\le 2\cdot 10^5,1\le A_i,B_i\le 10^9)$

Output

输出 $A'$ 序列

Sample Input

5
7 3 5 3 4
2 1 3 2 3

Sample Output

4 7 3 5 3

Solution

显然对于固定的 $k$ ， $n$ 越大 $F(n,k)$ 越大，对于固定的 $n$ ， $k$ 越大则 $F(n,k)$ 越小，故给较小的 $B_i$ 配较大的 $A_i$ 会让答案增大，排个序即可

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=200005;
int m,a[maxn],ans[maxn];
struct node
{
    int val,id;
    bool operator<(const node &a)const
    {
        if(a.val!=val)return val>a.val;
        return id<a.id;
    }
}b[maxn];
int main()
{
    while(~scanf("%d",&m))
    {
        for(int i=1;i<=m;i++)scanf("%d",&a[i]);
        sort(a+1,a+m+1);
        for(int i=1;i<=m;i++)scanf("%d",&b[i].val),b[i].id=i;
        sort(b+1,b+m+1);
        for(int i=1;i<=m;i++)ans[b[i].id]=a[i];
        for(int i=1;i<=m;i++)printf("%d%c",ans[i],i==m?'\n':' ');
    } 
    return 0;
}