LeetCode 2974. 最小数字游戏

吃着火锅x唱着歌2023-12-26 17:47

你有一个下标从 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

直接模拟即可：

cpp 复制代码

class Solution {
public:
    vector<int> numberGame(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        int sz = nums.size();
        vector<int> arr;
        for (int i = 0; i < sz; i += 2)
        {
            arr.push_back(nums[i + 1]);
            arr.push_back(nums[i]);
        }

        return arr;
    }
};

如果nums中有n个元素，此算法时间复杂度为O(n)，空间复杂度为O(1)。

自己写快排：

cpp 复制代码

class Solution {
public:
    vector<int> numberGame(vector<int>& nums) {
        int sz = nums.size();
        quickSort(nums, 0, sz - 1);
        vector<int> arr;
        for (int i = 0; i < sz; i += 2)
        {
            arr.push_back(nums[i + 1]);
            arr.push_back(nums[i]);
        }

        return arr;
    }

private:
    void quickSort(vector<int> &nums, int start, int end)
    {
        if (start >= end)
        {
            return;
        }

        int mid = (start + end) / 2;
        swap(nums[mid], nums[end]);
        int dividingPos = start - 1;
        for (int i = start; i <= end - 1; ++i)
        {
            if (nums[i] < nums[end])
            {
                swap(nums[i], nums[++dividingPos]);
            }
        }

        swap(nums[dividingPos + 1], nums[end]);
        
        quickSort(nums, start, dividingPos);
        quickSort(nums, dividingPos + 2, end);
    }
};