你有一个下标从 0 开始、长度为 偶数 的整数数组 nums
,同时还有一个空数组 arr
。Alice 和 Bob 决定玩一个游戏,游戏中每一轮 Alice 和 Bob 都会各自执行一次操作。游戏规则如下:
- 每一轮,Alice 先从
nums
中移除一个 最小 元素,然后 Bob 执行同样的操作。 - 接着,Bob 会将移除的元素添加到数组
arr
中,然后 Alice 也执行同样的操作。 - 游戏持续进行,直到
nums
变为空。
返回结果数组 arr
。
示例 1:
输入:nums = [5,4,2,3] 输出:[3,2,5,4] 解释:第一轮,Alice 先移除 2 ,然后 Bob 移除 3 。然后 Bob 先将 3 添加到 arr 中,接着 Alice 再将 2 添加到 arr 中。于是 arr = [3,2] 。 第二轮开始时,nums = [5,4] 。Alice 先移除 4 ,然后 Bob 移除 5 。接着他们都将元素添加到 arr 中,arr 变为 [3,2,5,4] 。
示例 2:
输入:nums = [2,5] 输出:[5,2] 解释:第一轮,Alice 先移除 2 ,然后 Bob 移除 5 。然后 Bob 先将 5 添加到 arr 中,接着 Alice 再将 2 添加到 arr 中。于是 arr = [5,2] 。
提示:
1 <= nums.length <= 100
1 <= nums[i] <= 100
nums.length % 2 == 0
问题简要描述:返回结果数组
解法一
Java
class Solution {
public int[] numberGame(int[] nums) {
int[] ans = new int[nums.length];
PriorityQueue<Integer> pq = new PriorityQueue<>();
for (int num : nums) {
pq.offer(num);
}
int i = 0;
while (!pq.isEmpty()) {
int p = pq.poll();
ans[i++] = pq.poll();
ans[i++] = p;
}
return ans;
}
}
Python3
class Solution:
def numberGame(self, nums: List[int]) -> List[int]:
heapify(nums)
ans = []
while nums:
a, b = heappop(nums), heappop(nums)
ans.append(b)
ans.append(a)
return ans
TypeScript
function numberGame(nums: number[]): number[] {
let pq = new MinPriorityQueue();
for (const x of nums) {
pq.enqueue(x);
}
const ans: number[] = [];
while (pq.size()) {
const a = pq.dequeue().element;
const b = pq.dequeue().element;
ans.push(b, a);
}
return ans;
};
C++
class Solution {
public:
vector<int> numberGame(vector<int>& nums) {
priority_queue<int, vector<int>, greater<int>> pq;
for (int x : nums) {
pq.push(x);
}
vector<int> ans;
while (pq.size()) {
int a = pq.top();
pq.pop();
int b = pq.top();
pq.pop();
ans.emplace_back(b);
ans.emplace_back(a);
}
return ans;
}
};
解法二
Java
class Solution {
public int[] numberGame(int[] nums) {
Arrays.sort(nums);
for (int i = 0; i < nums.length; i += 2) {
int t = nums[i];
nums[i] = nums[i + 1];
nums[i + 1] = t;
}
return nums;
}
}
Python3
class Solution:
def numberGame(self, nums: List[int]) -> List[int]:
nums.sort()
for i in range(0, len(nums), 2):
nums[i], nums[i + 1] = nums[i + 1], nums[i]
return nums
TypeScript
function numberGame(nums: number[]): number[] {
nums.sort((a, b) => a - b);
for (let i = 0; i < nums.length; i += 2) {
[nums[i], nums[i + 1]] = [nums[i + 1], nums[i]];
}
return nums;
};
C++
class Solution {
public:
vector<int> numberGame(vector<int>& nums) {
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size(); i+=2){
swap(nums[i], nums[i + 1]);
}
return nums;
}
};
Go
func numberGame(nums []int) []int {
sort.Ints(nums)
for i := 0; i < len(nums); i += 2 {
nums[i], nums[i+1] = nums[i+1], nums[i]
}
return nums
}
标签:arr,pq,return,nums,队列,Lc2974,int,ans,小根堆
From: https://blog.csdn.net/qq_51626500/article/details/140405769