题目
有 3n 堆数目不一的硬币,你和你的朋友们打算按以下方式分硬币:
每一轮中,你将会选出 任意 3 堆硬币(不一定连续)。
Alice 将会取走硬币数量最多的那一堆。
你将会取走硬币数量第二多的那一堆。
Bob 将会取走最后一堆。
重复这个过程,直到没有更多硬币。
给你一个整数数组 piles ,其中 piles[i] 是第 i 堆中硬币的数目。
返回你可以获得的最大硬币数目。
示例 1:
输入:piles = [2,4,1,2,7,8]
输出:9
解释:选出 (2, 7, 8) ,Alice 取走 8 枚硬币的那堆,你取走 7 枚硬币的那堆,Bob 取走最后一堆。
选出 (1, 2, 4) , Alice 取走 4 枚硬币的那堆,你取走 2 枚硬币的那堆,Bob 取走最后一堆。
你可以获得的最大硬币数目:7 + 2 = 9.
考虑另外一种情况,如果选出的是 (1, 2, 8) 和 (2, 4, 7) ,你就只能得到 2 + 4 = 6 枚硬币,这不是最优解。
示例 2:
输入:piles = [2,4,5]
输出:4
示例 3:
输入:piles = [9,8,7,6,5,1,2,3,4]
输出:18
提示:
3 <= piles.length <= 10^5
piles.length % 3 == 0
1 <= piles[i] <= 10^4
思路
- 因为看到10^5,大概率数组只能遍历一次
- 最优情况就是,Bob把全部最少的拿完,我尽量拿到最多的,那么就想到先把数组排序了
- Bob拿最少的,那就从数组左边开始拿
- 我尽量拿最多的,即尽量拿当前第二多的硬币,所以从数组右边开始拿
代码
class Solution {
public:
int maxCoins(vector<int>& piles) {
int res=0;
sort(piles.begin(), piles.end());
for(int i = 0, j = piles.size()-1; i < piles.size()/3; i++, j-=2){
res+=piles[j-1];
}
return res;
}
};