Codeforces-798D-Mike and distribution(贪心)

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

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

#include
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;i2)
    {
        int l=p[i],r=p[i+1];
        if(b[l]if(n%2==0) ans.push_back(p[n]);
    printf("%d\n",ans.size());
    for(auto i:ans) printf("%d ",i);
    return 0;
}

你可能感兴趣的:(贪心)