[牛客网]纸牌游戏

[编程题]纸牌游戏

牛牛和羊羊正在玩一个纸牌游戏。这个游戏一共有n张纸牌, 第i张纸牌上写着数字a_i。
牛牛和羊羊轮流抽牌, 牛牛先抽, 每次抽牌他们可以从纸牌堆中任意选择一张抽出, 直到纸牌被抽完。
他们的得分等于他们抽到的纸牌数字总和。
现在假设牛牛和羊羊都采用最优策略, 请你计算出游戏结束后牛牛得分减去羊羊得分等于多少。

输入描述:

输入包括两行。
第一行包括一个正整数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;
}