牛客寒假多校第二场 F-拿物品(贪心问题）

#include<bits/stdc++.h>
using namespace std;
int a[200005],b[200005];
// 题意 有n个物品 每个物品a,b两个属性 两个人按顺序拿  第一个人拿到物品的值永远是a 第二个人是B
/*问 怎么拿才能在拿完的时候 第一个人的sum 和第二个人的sum 之间的差值最大

讨论问题：

假设物品已经被选完，此时 牛牛选择的物品 A 属性的价值和是 N ， 牛可乐选择的物品 BB 属性价值和是 M 。

如果 牛牛的 (a1,b1)物品和牛可乐的(a2,b2)交换，则N'=N-a1+a2 M'=M+b1-b2,
对于牛牛（第一个人）目标是最大化N-M 来说 会变得更优仅当a1+b1<a2+b2 (N'-M'>N-M化简就能得到)
即 N-a1+a2-M-b1+b2>N-M
 =>a2+b2>a1+b1

对于牛可乐也一样。所以两人都会优先选择ai+bi最大的物品。 

将物品按照两个属性的和从大到小排序，依次分给两人即可。

除排序时间复杂度 O(n)
整体时间 O(logn+n) 



*/ 
struct node{
	int c;
	int id;
}c[200005];
int ans1[200005],ans2[200005];
bool cmp(node a,node b){
	return a.c>b.c;
}
int main(){
	int n;
	cin>>n;
	for(int i=1;i<=n;++i){
		cin>>a[i];
	}
	for(int i=1;i<=n;++i)cin>>b[i];
	for(int i=1;i<=n;++i){
		c[i].c=a[i]+b[i];
		c[i].id=i; 
	}
	sort(c+1,c+n+1,cmp);
	int cnt1=0,cnt2=0;
	for(int i=1;i<=n;++i){
		if(i%2)ans1[++cnt1]=c[i].id;
		else ans2[++cnt2]=c[i].id;
	}
	for(int i=1;i<=cnt1;++i){
		cout<<ans1[i]<<" ";
	}
	cout<<endl;
	for(int i=1;i<=cnt2;++i){
		cout<<ans2[i]<<" ";
	}
	return 0;
}