[编程题]纸牌游戏
牛牛和羊羊正在玩一个纸牌游戏。这个游戏一共有n张纸牌, 第i张纸牌上写着数字ai。
牛牛和羊羊轮流抽牌, 牛牛先抽, 每次抽牌他们可以从纸牌堆中任意选择一张抽出, 直到纸牌被抽完。
他们的得分等于他们抽到的纸牌数字总和。
现在假设牛牛和羊羊都采用最优策略, 请你计算出游戏结束后牛牛得分减去羊羊得分等于多少。
输入描述:
输入包括两行。
第一行包括一个正整数n(1 <= n <= 105),表示纸牌的数量。
第二行包括n个正整数ai(1 <= ai <= 109),表示每张纸牌上的数字。
输出描述:
输出一个整数, 表示游戏结束后牛牛得分减去羊羊得分等于多少。
输入例子1:
3 2 7 4
输出例子1:
5
思路
使用multiset
容器,自动排序(从小到大),允许出现重复key
。
详细思如见代码。
代码
#include<iostream>
#include<set>
using namespace std;
int main() {
int n;
cin >> n;
unsigned long ai, sum = 0;
multiset<unsigned long> ms;
for (int i = 0; i < n; i++) {
cin >> ai;
ms.insert(ai);
}
// 得到尾指针
multiset<unsigned long>::reverse_iterator it = ms.rbegin();
for (int i = 1; it != ms.rend(); it++, i++) {
if (i % 2 == 1)//奇数,即牛牛先选
sum += *it;
else
sum -= *it;
}
cout << sum;
return 0;
}