2020牛客寒假算法基础集训营2 F-拿物品 （贪心）

题目传送

题目大意：
有n个物品，但对A和B来说意义不一样，问：两人都想最大化与对方的分差，都按最优策略怎么取。
分析：
可以认为两人刚开始都拥有全部，取一个后，使A的a分受到保护，并使B减少b分，所以尽可能拿a+b大的，排序后轮流取就行了。

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

#define maxn 200000+10

struct node {
	int a, b, flag;
}an[maxn];

int aa[maxn];

int cmp(int a, int b)
{
	return an[a].a + an[a].b > an[b].a + an[b].b;
}

vector<int> na, nb;

int main(void)
{
	int n;
	cin >> n;
	for (int i = 1; i <= n; i++) cin >> an[i].a;
	for (int i = 1; i <= n; i++) {
		cin >> an[i].b;
		an[i].flag = 1;
		aa[i] = i;
	}
	sort(aa + 1, aa + n + 1, cmp);
	int p = 1;
	while (1) {
		while (an[aa[p]].flag == 0 && p < n) p++;
		if (an[aa[p]].flag) {
			na.push_back(aa[p]);
			an[aa[p]].flag = 0;
		}
		while (an[aa[p]].flag == 0 && p < n) p++;
		if (an[aa[p]].flag) {
			nb.push_back(aa[p]);
			an[aa[p]].flag = 0;
		}
		if (na.size() == (n / 2 + n % 2)) break;
	}
	for (int i = 0; i < na.size(); i++) cout << na[i] << " ";
	cout << endl;
	for (int i = 0; i < nb.size(); i++) cout << nb[i] << " ";
	cout << endl;
	return 0;
}