题目链接:Codeforces-798D-Mike and distribution
题意是给两个长度为n的数组A和B,要从里面选取n/2+1个下标使得选出来的数之和在A,B中比和的一半大。
先将下标根据A从大到小排序,即排序后
Api>Api+1
,p为下标数组。
若n是奇数,选第一个,若n是偶数,选第一个和最后一个。
然后对于中间的每两个数,选
Bpi和Bpi+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;
}