题目大意:
有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;
}