Codeforces-798D-Mike and distribution（贪心）

最新推荐文章于 2019-08-10 09:43:00 发布

原创最新推荐文章于 2019-08-10 09:43:00 发布 · 1k 阅读

1 ·

CC 4.0 BY-SA版权

贪心专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一个解决Codeforces-798D问题的算法思路，该问题要求从两个长度相同的数组中选择部分元素使它们的总和超过各自数组一半的总和。通过特定的排序和选择策略来实现这一目标。

题目链接：Codeforces-798D-Mike and distribution

题意是给两个长度为n的数组A和B，要从里面选取n/2+1个下标使得选出来的数之和在A，B中比和的一半大。
先将下标根据A从大到小排序，即排序后 $A_{p_i}>A_{p_{i+1}}$ ，p为下标数组。
若n是奇数，选第一个，若n是偶数，选第一个和最后一个。
然后对于中间的每两个数，选 $B_{p_i}和B_{p_{i+1}}$ 中较大的那一个，这样就能满足条件了。。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+7;
int a[maxn],b[maxn],p[maxn];
bool cmp(const int &a, const int &b)
{
    return ::a[a]>::a[b];
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=n;i++) scanf("%d",&b[i]);
    for(int i=1;i<=n;i++) p[i]=i;
    sort(p+1,p+1+n,cmp);
    vector<int> ans;
    ans.push_back(p[1]);
    for(int i=2;i<n;i+=2)
    {
        int l=p[i],r=p[i+1];
        if(b[l]<b[r]) swap(l,r);
        ans.push_back(l);
    }
    if(n%2==0) ans.push_back(p[n]);
    printf("%d\n",ans.size());
    for(auto i:ans) printf("%d ",i);
    return 0;
}