CodeForces-798D Mike and distribution

按照A的大小给数组的下标排序

排好后，先取第一个下标，也就是A最大的那个

之后按顺序每两个按照下标之间取B较大的那个，最后如果n是偶数还剩最后一个可以取

这样就可以保证满足条件

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int N=1e5+10;
struct node
{
	int v,id,u;
	bool operator < (const node a) const
	{
		return u>a.u;
	}
}a[N];
int n;
int main()
{
	scanf("%d",&n);
	for(int i=0;i<n;i++)
		scanf("%d",&a[i].u);
	for(int i=0;i<n;i++)
		scanf("%d",&a[i].v),a[i].id=i;
	sort(a,a+n);
	int t=0;
	int ans[N];
	ans[t++]=a[0].id;
	for(int i=1;i<n-1;i+=2)
	{
		if(a[i].v>a[i+1].v) ans[t++]=a[i].id;
		else ans[t++]=a[i+1].id;
	}
	if(n%2==0) ans[t++]=a[n-1].id;
	printf("%d\n",t);
	for(int i=0;i<t-1;i++)
		printf("%d ",ans[i]+1);
	printf("%d\n",ans[t-1]+1);
	return 0;
}